init

tools/helper.py

@@ -0,0 +1,74 @@
+import os
+import platform
+import subprocess
+import requests
+from tqdm import tqdm
+import hashlib
+
+def get_download_dir():
+    download_loc = ""
+    if os.environ.get("XDG_CACHE_HOME", None) is None:
+        download_loc = os.path.join('/', "home", os.environ.get("SUDO_USER", os.environ["USER"]), ".cache", "redroid", "downloads")
+    else:
+        download_loc = os.path.join(os.environ["XDG_CACHE_HOME"], "redroid", "downloads")
+    if not os.path.exists(download_loc):
+        os.makedirs(download_loc)
+    return download_loc
+
+def run(args):
+    result = subprocess.run(args=args, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    if result.stderr:
+        print(result.stderr.decode("utf-8"))
+        raise subprocess.CalledProcessError(
+                    returncode = result.returncode,
+                    cmd = result.args,
+                    stderr = result.stderr
+                )
+    return result
+
+def download_file(url, f_name):
+    md5 = ""
+    response = requests.get(url, stream=True, proxies={"https":"127.0.0.1:7890"})
+    total_size_in_bytes = int(response.headers.get('content-length', 0))
+    block_size = 1024  # 1 Kibibyte
+    progress_bar = tqdm(total=total_size_in_bytes, unit='iB', unit_scale=True)
+    with open(f_name, 'wb') as file:
+        for data in response.iter_content(block_size):
+            progress_bar.update(len(data))
+            file.write(data)
+    progress_bar.close()
+    with open(f_name, "rb") as f:
+        bytes = f.read()
+        md5 = hashlib.md5(bytes).hexdigest()
+    if total_size_in_bytes != 0 and progress_bar.n != total_size_in_bytes:
+        raise ValueError("Something went wrong while downloading")
+    return md5
+
+def host():
+    machine = platform.machine()
+
+    mapping = {
+        "i686": ("x86", 32),
+        "x86_64": ("x86_64", 64),
+        "aarch64": ("arm64-v8a", 64),
+        "armv7l": ("armeabi-v7a", 32),
+        "armv8l": ("armeabi-v7a", 32)
+    }
+    if machine in mapping:
+        # if mapping[machine] == "x86_64":
+        #     with open("/proc/cpuinfo") as f:
+        #         if "sse4_2" not in f.read():
+        #             print("x86_64 CPU does not support SSE4.2, falling back to x86...")
+        #             return ("x86", 32)
+        return mapping[machine]
+    raise ValueError("platform.machine '" + machine + "'"
+                     " architecture is not supported")
+
+class bcolors:
+    RED = '\033[31m'
+    YELLOW = '\033[33m'
+    GREEN = '\033[32m'
+    ENDC = '\033[0m'
+
+def print_color(str, color):
+    print(color+str+bcolors.ENDC)