- add new command to purge the cache items

Author: RootLUG

aura/cache.py

@@ -52,7 +52,6 @@ def iter_items(cls, tags=None) -> Generator[CacheItem, None, None]:
             else:
                 yield obj
 
-
     @property
     def mtime(self) -> int:
         return self.item_stat.st_mtime

aura/cli.py

@@ -255,6 +255,12 @@ def check_requirement():
     commands.check_requirement(payload)
 
 
+@cli.command()
+@click.option("--tag", "-t", multiple=True)
+def cleanup(tag=None):
+    commands.cleanup(cache_tags=tag)
+
+
 def main():
     if "AURA_USER" in os.environ:
         username = os.environ["AURA_USER"]

aura/commands.py

@@ -21,6 +21,7 @@
 from . import utils
 from . import mirror
 from . import typos
+from . import cache
 from .analyzers.detections import Detection
 from .output.base import ScanOutputBase, DiffOutputBase, InfoOutputBase, TyposquattingOutputBase
 
@@ -228,3 +229,8 @@ def generate_typosquatting(distance=2, limit=None, pkgs=None, format_uri="text",
     formatter = TyposquattingOutputBase.from_uri(format_uri)
     formatter.output_typosquatting(islice(typos.enumerator(combinations, f, extended=extended), 0, limit))
 
+
+def cleanup(cache_tags):
+    cache_items = tuple(cache.CacheItem.iter_items(tags=cache_tags))
+    click.confirm(f"You are about to delete {len(cache_items)} cache items. Proceed?", abort=True)
+    cache.CacheItem.cleanup(items=cache_items)