Séparation méthodes en modules, début docstring

lib/system.py

@@ -0,0 +1,28 @@
+import tarfile, os
+
+def extract_tar(file_path, dest_dir = "."):
+    print('Extracting "{:s}" to "{:s}"'.format(file_path, dest_dir))
+    with tarfile.open(file_path) as tar:
+        tar.extractall(dest_dir)
+        tar.close()
+
+def which(program, additional_paths = None):
+
+    def is_exe(fpath):
+        return os.path.isfile(fpath) and os.access(fpath, os.X_OK)
+
+    fpath, fname = os.path.split(program)
+    if fpath:
+        if is_exe(program):
+            return program
+    else:
+        paths = os.environ["PATH"].split(os.pathsep);
+        if additional_paths != None:
+            paths.append(additional_paths)
+        for path in paths:
+            path = path.strip('"')
+            exe_file = os.path.join(path, program)
+            if is_exe(exe_file):
+                return exe_file
+
+    return None

lib/tamarin.py

@@ -0,0 +1,54 @@
+import os, glob, subprocess
+import web, system
+
+def get_workspace_dir():
+    """Return the absolute path to the tamarin workspace ($HOME/.tamarin)"""
+    home = os.environ["HOME"]
+    return os.path.join(os.sep, home, '.tamarin')
+
+def get_workspace_subdir(subdir):
+    """Return the absolute path to a subdirectory in tamarin workspace"""
+    dir_path = os.path.join(os.sep, get_workspace_dir(), subdir)
+    os.makedirs(dir_path, exist_ok=True)
+    return dir_path
+
+def get_acbuild_achive_dest_dir():
+    """Return the first path matching the acbuild archive extraction destination in tamarin workspace"""
+    workspace_tmp = get_workspace_subdir('tmp')
+    return glob.glob(os.path.join(os.sep, workspace_tmp, 'acbuild-v*'))[0]
+
+def get_rkt_achive_dest_dir():
+    """Return the first path matching the rkt archive extraction destination in tamarin workspace"""
+    workspace_tmp = get_workspace_subdir('tmp')
+    return glob.glob(os.path.join(os.sep, workspace_tmp, 'rkt-v*'))[0]
+
+def get_acbuild_workspace_dir():
+    """Return the current path (linked to the process pid) to the acbuild workspace"""
+    return get_workspace_subdir("tmp/build_{:d}".format(os.getpid()))
+
+def download_rkt():
+    """Download a local copy of rkt in the tamarin workspace and return the absolute path to the archive"""
+    url = "https://github.com/coreos/rkt/releases/download/v1.22.0/rkt-v1.22.0.tar.gz"
+    file_path=os.path.join(os.sep, get_workspace_subdir('tmp'), "rkt.tar.gz")
+    web.download_file(file_url=url, dest_path=file_path)
+    return file_path
+
+def download_acbuild():
+    """Download a local copy of acbuild in the tamarin workspace and return the absolute path to the archive"""
+    url = "https://github.com/containers/build/releases/download/v0.4.0/acbuild-v0.4.0.tar.gz"
+    file_path=os.path.join(os.sep, get_workspace_subdir('tmp'), "acbuild.tar.gz")
+    web.download_file(file_url=url, dest_path=file_path)
+    return file_path
+
+def run_rkt(args):
+    """Run rkt with the specified args (use the local copy if rkt is not found in the $PATH)"""
+    rkt_bin = system.which('rkt', get_workspace_subdir('rkt'))
+    cmd = ( ["sudo", "-E", rkt_bin] if os.geteuid() != 0 else [rkt_bin] ) + args
+    print(" ".join(cmd))
+    return subprocess.call(cmd, stdin=subprocess.PIPE)
+
+def run_acbuild(args):
+    """Run acbuild with the specified args (use the local copy if acbuild is not found in the $PATH)"""
+    acbuild_bin = system.which('acbuild', get_workspace_subdir('acbuild'))
+    print(" ".join([acbuild_bin] + args))
+    return subprocess.call([acbuild_bin] + args, stdin=subprocess.PIPE)

lib/web.py

@@ -0,0 +1,33 @@
+from urllib import request
+import math, sys
+
+def print_progress_bar(percent_progress=0, char_size=50, clear_line=True):
+
+    bar_progress = math.floor(char_size*(percent_progress/100))
+    bar = "=" * bar_progress + " " * (char_size - bar_progress)
+
+    if clear_line:
+        sys.stdout.write(u"\u001b[1000D")
+    sys.stdout.write("[{:s}] {:d}%".format(bar, int(percent_progress)))
+    sys.stdout.flush()
+
+def download_file(file_url, dest_path, bulk_size = 8192):
+
+    req = request.urlopen(file_url)
+    meta = req.info()
+    file_size = int(meta.get('Content-Length'))
+
+    with open(dest_path, 'wb') as dest_file:
+        print('Downloading "{:s}". Size: {:d}b'.format(file_url, file_size))
+        downloaded_size = 0
+        while True:
+            buff = req.read(bulk_size)
+            if not buff:
+                break
+            downloaded_size += len(buff)
+            dest_file.write(buff)
+            progress = downloaded_size/file_size*100
+            print_progress_bar(progress)
+        dest_file.close()
+    # Add linebreak
+    print("")