Generate arbitrary named entity (#69)

SeaQL · Aug 10, 2021 · 9be094b · 9be094b
1 parent b1d28db
commit 9be094b
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/sea-orm-cli/Cargo.toml b/sea-orm-cli/Cargo.toml
@@ -19,7 +19,7 @@ clap = { version = "^2.33.3" }
 dotenv = { version = "^0.15" }
 async-std = { version = "^1.9", features = [ "attributes" ] }
 sea-orm = { version = "^0.1", features = [ "sqlx-all" ] }
-sea-orm-codegen = { version = "^0.1" }
+sea-orm-codegen = { path = "../sea-orm-codegen", version = "^0.1" }
 sea-schema = { version = "^0.2.4", default-features = false, features = [
     "sqlx-mysql",
     "sqlx-postgres",

diff --git a/sea-orm-cli/README.md b/sea-orm-cli/README.md
@@ -9,5 +9,5 @@ cargo run -- -h
 Running Entity Generator:
 
 ```sh
-cargo run -- entity generate -url mysql://sea:sea@localhost/bakery -schema bakery -o out
+cargo run -- generate entity -u mysql://sea:sea@localhost/bakery -s bakery -o out
 ```
diff --git a/sea-orm-codegen/src/entity/writer.rs b/sea-orm-codegen/src/entity/writer.rs
@@ -34,7 +34,7 @@ impl EntityWriter {
                 let code_blocks = Self::gen_code_blocks(entity);
                 Self::write(&mut lines, code_blocks);
                 OutputFile {
-                    name: format!("{}.rs", entity.table_name),
+                    name: format!("{}.rs", entity.get_table_name_snake_case()),
                     content: lines.join("\n\n"),
                 }
             })
@@ -123,11 +123,11 @@ impl EntityWriter {
     }
 
     pub fn gen_impl_entity_name(entity: &Entity) -> TokenStream {
-        let table_name_snake_case = entity.get_table_name_snake_case();
+        let table_name = entity.table_name.as_str();
         quote! {
             impl EntityName for Entity {
                 fn table_name(&self) -> &str {
-                    #table_name_snake_case
+                    #table_name
                 }
             }
         }
@@ -341,7 +341,7 @@ mod tests {
                 }],
             },
             Entity {
-                table_name: "cake_filling".to_owned(),
+                table_name: "_cake_filling_".to_owned(),
                 columns: vec![
                     Column {
                         name: "cake_id".to_owned(),

diff --git a/sea-orm-codegen/tests/entity/cake_filling.rs b/sea-orm-codegen/tests/entity/cake_filling.rs
@@ -7,7 +7,7 @@ pub struct Entity;
 
 impl EntityName for Entity {
     fn table_name(&self) -> &str {
-        "cake_filling"
+        "_cake_filling_"
     }
 }