sea-orm-cli: allow skipping impl ActiveModelBehavior for generated en…

…tities
SeaQL · Jul 10, 2024 · c512541 · c512541
1 parent 9fd5523
commit c512541
Show file tree

Hide file tree

Showing 4 changed files with 131 additions and 8 deletions.
diff --git a/sea-orm-cli/src/cli.rs b/sea-orm-cli/src/cli.rs
@@ -304,6 +304,13 @@ pub enum GenerateSubcommands {
             long_help = "Generate helper Enumerations that are used by Seaography."
         )]
         seaography: bool,
+
+        #[arg(
+            long,
+            default_value = "true",
+            long_help = "Generate empty ActiveModelBehavior impls."
+        )]
+        gen_impl_active_model_behavior: bool,
     },
 }
 

diff --git a/sea-orm-cli/src/commands/generate.rs b/sea-orm-cli/src/commands/generate.rs
@@ -34,6 +34,7 @@ pub async fn run_generate_command(
             enum_extra_derives,
             enum_extra_attributes,
             seaography,
+            gen_impl_active_model_behavior,
         } => {
             if verbose {
                 let _ = tracing_subscriber::fmt()
@@ -191,6 +192,7 @@ pub async fn run_generate_command(
                 enum_extra_derives,
                 enum_extra_attributes,
                 seaography,
+                gen_impl_active_model_behavior,
             );
             let output = EntityTransformer::transform(table_stmts)?.generate(&writer_context);
 

diff --git a/sea-orm-codegen/src/entity/transformer.rs b/sea-orm-codegen/src/entity/transformer.rs
@@ -398,6 +398,7 @@ mod tests {
                     &Default::default(),
                     &Default::default(),
                     false,
+                    true,
                 )
                 .into_iter()
                 .skip(1)