[Fix] Fix manager handle error

2025-07-11 14:49:11 +08:00
parent 8e487863cf
commit ab5c5e2262
4 changed files with 98 additions and 2 deletions
--- a/main.py
+++ b/main.py
@ -0,0 +1,93 @@
+#!/usr/bin/env python3
+
+from src.manager import *
+import logging
+import colorlog
+import sys
+
+def initialize_logger():
+    """Initialize and return a color logger."""
+    handler = colorlog.StreamHandler()
+    handler.setFormatter(colorlog.ColoredFormatter(
+        '%(log_color)s%(asctime)s - %(levelname)s - %(message)s',
+        log_colors={
+            'DEBUG':    'cyan',
+            'INFO':     'green',
+            'WARNING':  'yellow',
+            'ERROR':    'red',
+            'CRITICAL': 'bold_red',
+        }
+    ))
+
+    logger = colorlog.getLogger(__name__)
+    logger.setLevel(logging.INFO)
+    logger.addHandler(handler)
+    return logger
+
+def initialize_service_manager():
+    """Initialize and return the service manager."""
+    return Manager()
+
+logger = initialize_logger()
+service_manager = initialize_service_manager()
+
+def start_menu():
+    logger.info(f"=======================================")
+    logger.info(f"========== Services Manager ===========")
+    logger.info(f"=======================================")
+    logger.info(f"Please choose a operation to carry out:")
+    logger.info(f"=====> 1. Register a new service <=====")
+    logger.info(f"=====> 2. Manage exsist services <=====")
+    logger.info(f"=====> c. Count the number of services")
+    logger.info(f"=====> q. Quit ")
+
+def main_menu():
+    """Display the main menu and handle user input."""
+
+    while True:
+        start_menu()
+
+        start_choice = input("Enter your choice: ").strip()
+        if start_choice == '1':
+            service_tag = input("Please input the deploy way of service you want to register('docker' | 'sys'): ")
+            service_name = input("Please input the name of the service: ")
+            if service_tag == "docker":
+                service_path = input("Please input the config path of the service where 'docker-compose.yml' located: ")
+                service_manager.register_service(service_tag, service_name, service_path)
+            else:
+                service_manager.register_service(service_tag, service_name)
+        if start_choice == '2':
+            if not service_manager.services_list:
+                logger.error("No services registered.")
+                continue
+            while True:
+                logger.info(f"=====> Exsist Services Management <=====")
+                service_code = 1
+                for service in service_manager.services_list:
+                    logger.info(f"=====> {service_code}: {service._name}")
+                    service_code += 1
+                # logger.info("=====> a. Select All")
+                logger.info("=====> q. Quit")
+
+                manage_choice = input("Please select a choice to carry out: ").strip()
+                if manage_choice == 'q':
+                    logger.info("Exit main manager")
+                    break
+                # if manage_choice == 'a':
+                #     pass
+                try:
+                    code = int(manage_choice)
+                    if 0 <= code < len(service_manager.services_list):
+                        service_manager.services_list[code].service_operation()
+                    else:
+                        logger.error("Invalid service code.")
+                except ValueError:
+                    logger.error("Please enter a valid number.")
+        if start_choice == 'c':
+            service_manager.list_services()
+        if start_choice == 'q':
+            logger.info(f"Exit the manager process")
+            break
+
+if __name__ == '__main__':
+    sys.exit(main_menu())