Archiving Context refactor complete

src/auto_archiver/core/context.py

@@ -1,3 +1,5 @@
+from loguru import logger
+
 
 class ArchivingContext:
     """
@@ -7,11 +9,15 @@ class ArchivingContext:
     ArchivingContext.set(key, value)
     and 
     ArchivingContext.get(key, default)
+
+    When reset is called, all values are cleared EXCEPT if they were .set(keep_on_reset=True)
+        reset(full_reset=True) will recreate everything including the keep_on_reset status
     """
     _instance = None
 
     def __init__(self):
         self.configs = {}
+        self.keep_on_reset = set()
 
     @staticmethod
     def get_instance():
@@ -20,13 +26,22 @@ class ArchivingContext:
         return ArchivingContext._instance
 
     @staticmethod
-    def set(key, value):
-        ArchivingContext.get_instance().configs[key] = value
+    def set(key, value, keep_on_reset: bool = False):
+        logger.error(f"SET [{key}]={value}")
+        ac = ArchivingContext.get_instance()
+        ac.configs[key] = value
+        if keep_on_reset: ac.keep_on_reset.add(key)
 
     @staticmethod
     def get(key: str, default=None):
         return ArchivingContext.get_instance().configs.get(key, default)
 
+    @staticmethod
+    def reset(full_reset: bool = False):
+        ac = ArchivingContext.get_instance()
+        if full_reset: ac.keep_on_reset = set()
+        ac.configs = {k: v for k, v in ac.configs.items() if k in ac.keep_on_reset}
+
     # ---- custom getters/setters for widely used context values
 
     @staticmethod