add data

Gabrielolobo · Jun 1, 2023 · eb0280d · eb0280d
1 parent 68c2b39
commit eb0280d
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/crawley/cli.py b/crawley/cli.py
@@ -6,14 +6,12 @@
 
 # Calls crawler results
 def main(args):
-    output_callable = get_output(args.output)
+    output_callable = get_output(args.output, args.data)
 
-    all_results = []
-    for crawler in AVAILABLE_CRAWLERS.values():
-        results = crawler.run()
-        all_results.extend(results)
+    crawler = AVAILABLE_CRAWLERS[args.crawler]
+    results = crawler.run()
 
-    output_callable(all_results)
+    output_callable(results)
 
 
 if __name__ == "__main__":
@@ -24,7 +22,7 @@ def main(args):
     )
 
     cli_parser.add_argument(
-        'output', choices=['print', 'save json', 'save csv'])
+        'output', choices=['print', 'save_json', 'save_csv'])
 
     args = cli_parser.parse_args()
     main(args)