Orchestrator: convert to class implementation

cisagov · Jul 12, 2024 · 22ea201 · 22ea201
1 parent 9401ead
commit 22ea201
Show file tree

Hide file tree

Showing 2 changed files with 375 additions and 353 deletions.
diff --git a/scubagoggles/main.py b/scubagoggles/main.py
@@ -6,15 +6,15 @@
 """
 
 import argparse
-from scubagoggles.orchestrator import gws_products, start_automation
+from scubagoggles.orchestrator import Orchestrator
 
 def get_gws_args(parser):
     """
     Adds the arguments for the GWS parser
 
     :param parser: argparse object
     """
-    gws = gws_products()
+    gws = Orchestrator.gws_products()
     gws_baselines = gws["gws_baselines"]
 
     default_file_output_names = {
@@ -140,6 +140,6 @@ def dive():
     args = parser.parse_args()
 
     if args.scuba_cmd == 'gws':
-        start_automation(args)
+        Orchestrator(args).start_automation()
     else:
         raise Exception("Invalid subparser. Run scubagoggles -h to see a list of valid subparsers")