Pass table in - it's actually called only_tables

MrCurtis · Feb 1, 2024 · 5438edd · 5438edd
1 parent 34420ea
commit 5438edd
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/src/fk_graph/cli.py b/src/fk_graph/cli.py
@@ -13,7 +13,11 @@ def main():
         setup_data(engine)
     elif args.connection_string:
         engine = create_engine(args.connection_string)
-    graph = get_graph(engine, args.table, args.primary_key)
+    graph = get_graph(
+        engine,
+        args.table,
+        args.primary_key,
+        only_tables=args.only_tables)
     run_app(graph)
 
 def _parse_args():
@@ -30,7 +34,7 @@ def _parse_args():
     parser.add_argument("--connection-string")
     parser.add_argument("--table", required=True)
     parser.add_argument("--primary-key", required=True)
-    parser.add_argument("--include-only", type=_to_list)
+    parser.add_argument("--only-tables", type=_to_list)
     args = parser.parse_args()
     if (
         (not args.demo and args.connection_string is None)

diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -70,7 +70,7 @@ def test_can_pass_in_list_of_tables_to_include(self):
             " --demo"
             " --table=table_a"
             " --primary-key=1"
-            " --include-only=\"['table_a']\""
+            " --only-tables='[\"table_a\"]'"
         )
         # PIPE just stops stdout from printing to screen.
         process = Popen(shlex_split(command), stdout=PIPE)