Cache management

python_appimage/commands/cache/clean.py

@@ -0,0 +1,64 @@
+import glob
+import json
+import os
+from pathlib import Path
+import subprocess
+
+from ...utils.deps import CACHE_DIR
+from ...utils.fs import remove_file, remove_tree
+from ...utils.log import log
+
+
+__all__ = ['execute']
+
+
+def _unpack_args(args):
+    '''Unpack command line arguments
+    '''
+    return (args.tags, args.all)
+
+
+def execute(images, all_):
+    '''Clean cached image(s)
+    '''
+
+    cache = Path(CACHE_DIR)
+
+    if not images:
+        images = [image[9:] for image in sorted(os.listdir(cache /
+                                                         'share/images'))]
+
+    for image in images:
+        try:
+            image, tag = image.rsplit(':', 1)
+        except ValueError:
+            tag = None
+
+        if not image.replace('_', '').isalnum():
+            raise ValueError(f'bad image tag ({image})')
+
+        path = cache / f'share/images/manylinux{image}'
+        if not path.exists():
+            raise ValueError(f'no such image ({image})')
+
+        if tag is None:
+            if not all_:
+                path = path / 'extracted'
+            remove_tree(str(path))
+        else:
+            tag_file = path / f'tags/{tag}.json'
+            if not tag_file.exists():
+                raise ValueError(f'no such image ({image}:{tag})')
+
+            if all_:
+                with tag_file.open() as f:
+                    layers = json.load(f)["layers"]
+                for layer in layers:
+                    layer = path / f'layers/{layer}.tar.gz'
+                    if layer.exists():
+                        remove_file(str(layer))
+                remove_file(str(tag_file))
+            else:
+                path = cache / f'share/images/{image}/extracted/{tag}'
+                if path.exists():
+                    remove_tree(str(path))

python_appimage/commands/cache/list.py

@@ -0,0 +1,41 @@
+import glob
+import os
+from pathlib import Path
+import subprocess
+
+from ...utils.deps import CACHE_DIR
+from ...utils.log import log
+
+
+__all__ = ['execute']
+
+
+def _unpack_args(args):
+    '''Unpack command line arguments
+    '''
+    return tuple()
+
+
+def execute():
+    '''List cached image(s)
+    '''
+
+    cache = Path(CACHE_DIR)
+
+    images = sorted(os.listdir(cache / 'share/images'))
+    for image in images:
+        tags = ', '.join((
+            tag[:-5] for tag in \
+                sorted(os.listdir(cache / f'share/images/{image}/tags'))
+        ))
+        if not tags:
+            continue
+        path = cache / f'share/images/{image}'
+        memory = _getsize(path)
+        log('LIST', f'{image} ({tags}) [{memory}]')
+
+
+def _getsize(path: Path):
+    r = subprocess.run(f'du -sh {path}', capture_output=True, check=True,
+                       shell=True)
+    return r.stdout.decode().split(None, 1)[0]