Clear up the cleared flag on packages

Author: bitterpanda63

aikido_zen/background_process/cloud_connection_manager/send_heartbeat.py

@@ -17,7 +17,7 @@ def send_heartbeat(connection_manager):
     routes = list(connection_manager.routes)
     outgoing_domains = connection_manager.hostnames.as_array()
     ai_stats = connection_manager.ai_stats.get_stats()
-    packages = PackagesStore.get_packages()
+    packages = PackagesStore.export()
 
     connection_manager.statistics.clear()
     connection_manager.users.clear()

aikido_zen/background_process/packages.py

@@ -59,7 +59,7 @@ def is_version_supported(version, required_version):
 
 class PackagesStore:
     @staticmethod
-    def get_packages():
+    def export():
         global packages
         result = []
         for package in packages.values():
@@ -88,6 +88,9 @@ def get_package(package_name):
 
     @staticmethod
     def clear():
+        # To clear we set the `cleared` attribute to True
+        # This is to ensure you can still get the packages
+        # But that they will not show up during an export
         global packages
         for package in packages:
             packages[package]["cleared"] = True

aikido_zen/thread/thread_cache.py

@@ -65,7 +65,7 @@ def renew(self):
                 "users": self.users.as_array(),
                 "stats": self.stats.get_record(),
                 "ai_stats": self.ai_stats.get_stats(),
-                "packages": PackagesStore.get_packages(),
+                "packages": PackagesStore.export(),
             },
             receive=True,
         )