Split the actions (#108)

Author: sahilavaran

mlc/CacheAction.py

@@ -0,0 +1,76 @@
+from .action import Action
+import os
+import json
+from . import utils
+from .logger import logger
+
+class CacheAction(Action):
+
+    def __init__(self, parent=None):
+        #super().__init__(parent)
+        self.parent = parent
+        self.__dict__.update(vars(parent))
+    
+    def search(self, i):
+        i['target_name'] = "cache"
+        #logger.debug(f"Searching for cache with input: {i}")
+        return self.parent.search(i)
+
+    find = search
+
+    def rm(self, i):
+        i['target_name'] = "cache"
+        #logger.debug(f"Removing cache with input: {i}")
+        return self.parent.rm(i)
+
+    def show(self, run_args):
+        self.action_type = "cache"
+        res = self.search(run_args)
+        logger.info(f"Showing cache with tags: {run_args.get('tags')}")
+        cached_meta_keys_to_show = ["uid", "tags", "dependent_cached_path", "associated_script_item"]
+        cached_state_keys_to_show = ["new_env", "new_state", "version"]
+        for item in res['list']:
+            print(f"""Location: {item.path}:
+Cache Meta:""")
+            for key in cached_meta_keys_to_show:
+                if key in item.meta:
+                    print(f"""    {key}: {item.meta[key]}""")
+            print("""Cached State:""")
+            cached_state_meta_file = os.path.join(item.path, "mlc-cached-state.json")
+            if not os.path.exists(cached_state_meta_file):
+                continue
+            try:
+                # Load and parse the JSON file containing the cached state
+                with open(cached_state_meta_file, 'r') as file:
+                    meta = json.load(file)
+                    for key in cached_state_keys_to_show:
+                        if key in meta:
+                            print(f"""    {key}:""", end="")
+                            if meta[key] and isinstance(meta[key], dict):
+                                print("")
+                                utils.printd(meta[key], yaml=False, sort_keys=True, begin_spaces=8)
+                            else:
+                                print(f""" {meta[key]}""")
+            except json.JSONDecodeError as e:
+                logger.error(f"Error decoding JSON: {e}")
+            print("......................................................")
+            print("")
+            
+        return {'return': 0}
+
+    def list(self, args):
+        self.action_type = "cache"
+        run_args = {"fetch_all": True}  # to fetch the details of all the caches generated
+        
+        res = self.search(run_args)
+        if res['return'] > 0:
+            return res
+        
+        logger.info(f"Listing all the caches and their paths")
+        print("......................................................")
+        for item in res['list']:
+            print(f"tags: {item.meta['tags'] if item.meta.get('tags') else 'None'}")
+            print(f"Location: {item.path}")
+            print("......................................................")
+
+        return {'return': 0}

mlc/CfgAction.py

@@ -0,0 +1,42 @@
+import yaml
+import os
+from . import utils
+from .action import Action, default_parent
+from .logger import logger
+
+class CfgAction(Action):
+    def __init__(self, parent=None):
+        if parent is None:
+            parent = default_parent
+        #super().__init__(parent)
+        self.parent = parent
+        self.__dict__.update(vars(parent))
+
+    def load(self, args):
+        """
+        Load the configuration.
+        
+        Args:
+            args (dict): Contains the configuration details such as file path, etc.
+        """
+        #logger.info("In cfg load")
+        default_config_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'config.yaml')
+        config_file = args.get('config_file', default_config_path)
+        logger.info(f"In cfg load, config file = {config_file}")
+        if not config_file or not os.path.exists(config_file):
+            logger.error(f"Error: Configuration file '{config_file}' not found.")
+            return {'return': 1, 'error': f"Error: Configuration file '{config_file}' not found."}
+        
+        #logger.info(f"Loading configuration from {config_file}")
+        
+        # Example loading YAML configuration (can be modified based on your needs)
+        try:
+            with open(config_file, 'r') as file:
+                config_data = yaml.safe_load(file)
+                logger.info(f"Loaded configuration: {config_data}")
+                # Store configuration in memory or perform other operations
+                self.cfg = config_data
+        except yaml.YAMLError as e:
+            logger.error(f"Error loading YAML configuration: {e}")
+        
+        return {'return': 0, 'config': self.cfg}

mlc/ExperimentAction.py

@@ -0,0 +1,21 @@
+from .action import Action, default_parent
+from .logger import logger
+import os
+from . import utils
+
+class ExperimentAction(Action):
+    def __init__(self, parent=None):
+        if parent is None:
+            parent = default_parent
+        #super().__init__(parent)
+        self.parent = parent
+        self.__dict__.update(vars(parent))
+
+    def show(self, args):
+        logger.info(f"Showing experiment with identifier: {args.details}")
+        return {'return': 0}
+
+    def list(self, args):
+        logger.info("Listing all experiments.")
+        return {'return': 0}
+