[CLI] generate entity file for specified tables only (full text match…

…ing, not substring matching) (#1245)
SeaQL · Nov 24, 2022 · 4e6e23b · 4e6e23b
1 parent 0eccbc9
commit 4e6e23b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 13 deletions.
diff --git a/sea-orm-cli/src/cli.rs b/sea-orm-cli/src/cli.rs
@@ -141,7 +141,7 @@ pub enum GenerateSubcommands {
             takes_value = true,
             help = "Generate entity file for specified tables only (comma separated)"
         )]
-        tables: Option<String>,
+        tables: Vec<String>,
 
         #[clap(
             value_parser,

diff --git a/sea-orm-cli/src/commands/generate.rs b/sea-orm-cli/src/commands/generate.rs
@@ -56,19 +56,8 @@ pub async fn run_generate_command(
             // above
             let is_sqlite = url.scheme() == "sqlite";
 
-            let tables = match tables {
-                Some(t) => t,
-                _ => "".to_string(),
-            };
-
             // Closures for filtering tables
-            let filter_tables = |table: &str| -> bool {
-                if !tables.is_empty() {
-                    return tables.contains(table);
-                }
-
-                true
-            };
+            let filter_tables = |table: &String| -> bool { tables.contains(table) };
 
             let filter_hidden_tables = |table: &str| -> bool {
                 if include_hidden_tables {