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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
import shutil
import requests
import colors
import time
import os
import hashlib
import tarfile
DEFAULT_BAR_COLOR = colors.BLACK + colors.BG_CYAN
DEFAULT_BAR_COLOR_RESET = colors.BG_BLACK + colors.CYAN
def extract_tar(package_path, destination):
cmd = f"tar -h --no-overwrite-dir -xvf {package_path} -C {destination}"
os.popen(cmd).read()
with tarfile.open(package_path) as tar:
return "\n".join(["".join(m.name[1:]) for m in tar.getmembers() if not m.isdir()])
def add_path(*argv):
a = argv[0]
for b in argv[1:]:
a = a + ("" if a[-1] == "/" else "/") + (b[1:] if b[0] == "/" else b)
return a
def is_root():
return os.environ.get("SUDO_UID") or os.geteuid() == 0
def loading_bar(completed, total, text,
unit="", color=DEFAULT_BAR_COLOR, reset=DEFAULT_BAR_COLOR_RESET):
columns, rows = shutil.get_terminal_size((80, 20))
count = f"[{completed}{unit}/{total}{unit}]"
spaces = columns - (len(count) + len(text))
info = text + "".join([" " for i in range(spaces)]) + count
reset_at = int((completed/total)*len(info)) if total > 0 else 0
info = "".join([info[i] + (reset if i == reset_at else "") for i in range(len(info))])
print(color + info, end="\r")
def fill_line(text, color, end="\n"):
columns, rows = shutil.get_terminal_size((80, 20))
spaces = columns - (len(text))
print(color + text + "".join([" " for i in range(spaces)]), end=end)
def print_reset(text):
print(colors.RESET + text)
def curl(url, raw=False):
try:
r = requests.get(url)
except:
return 500, ""
return r.status_code, r.content if raw else r.text
def get_unit(n):
base = 1000
if n > base**4: return base**4, "TB"
elif n > base**3: return base**3, "GB"
elif n > base**2: return base**2, "MB"
elif n > base**1: return base**1, "KB"
else: return 1, "B"
def curl_to_file(url, path, text=""):
with requests.get(url, stream=True) as r:
r.raise_for_status()
length = int(r.headers['content-length']) if "content-length" in r.headers else 1000
with open(path, "wb") as f:
c_size = 4096
ic = r.iter_content(chunk_size=c_size)
done = 0
for chunk in ic:
if text:
divisor, unit = get_unit(length)
loading_bar(round(done/divisor, 2), round(length/divisor, 2), "Downloading " + text, unit=unit)
f.write(chunk)
done += c_size
if text:
divisor, unit = get_unit(length)
loading_bar(int(done/divisor), int(length/divisor), "Downloaded " + text, unit=unit)
return r.status_code, path
def mkdir(path):
if not os.path.exists(path):
os.makedirs(path)
def md5sum(filename):
md5_hash = hashlib.md5()
with open(filename,"rb") as f:
for byte_block in iter(lambda: f.read(4096),b""):
md5_hash.update(byte_block)
return md5_hash.hexdigest()
def ask_confirmation(text, default=True, no_confirm=False):
yes = "Y" if default else "y"
no = "n" if default else "N"
if no_confirm:
reponse = "y" if default else "n"
print(f"{text} [{yes},{no}] {colors.RESET} {reponse}")
else:
reponse = input(f"{text} [{yes},{no}] " + colors.RESET)
return reponse.lower() == "y" or len(reponse) == 0
if __name__ == "__main__":
for i in range(1000):
loading_bar(i, 1000, "it is loading...")
time.sleep(0.01)
|
