-
Notifications
You must be signed in to change notification settings - Fork 32
Expand file tree
/
Copy pathclean.py
More file actions
64 lines (50 loc) · 1.73 KB
/
clean.py
File metadata and controls
64 lines (50 loc) · 1.73 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
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))