Update main.py

aperture-sci · Apr 12, 2023 · e48acb0 · e48acb0
1 parent 94e849c
commit e48acb0
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/src/main.py b/src/main.py
@@ -1,13 +1,15 @@
+import sys
+
+from constants import DEVBOT_TASKS
+from helpers.logging_functions import handle_error
 from services.dev_bot_svc import DevBotService
 
 if __name__ == '__main__':
     try:
-        # if len(sys.argv) < 2 or sys.argv[1] not in DEVBOT_TASKS:
-        #     print("Error: Please provide an argument ('dev' or 'merge').")
-        #     exit(-1)
+        if len(sys.argv) < 2 or sys.argv[1] not in DEVBOT_TASKS:
+            handle_error("Please provide an argument ('dev' or 'merge').")
         devbot = DevBotService()
-        # devbot.run(sys.argv[1])
-        devbot.run("dev")
+        devbot.run(sys.argv[1])
     except Exception as e:
         print("Error: ", e)
         exit(-1)