summaryrefslogtreecommitdiff
path: root/src/util.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/util.py')
-rw-r--r--src/util.py37
1 files changed, 37 insertions, 0 deletions
diff --git a/src/util.py b/src/util.py
new file mode 100644
index 0000000..de05b08
--- /dev/null
+++ b/src/util.py
@@ -0,0 +1,37 @@
+import shutil
+import requests
+import colors
+import time
+import os
+
+def loading_bar(completed, total, text,
+ unit=""):
+
+ 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] + (colors.RESET if i == reset_at else "") for i in range(len(info))])
+
+ print(colors.BLACK + colors.BG_GREEN + info, end="\r")
+
+
+
+
+def curl(url):
+ r = requests.get(url)
+ return r.status_code, r.text
+
+def mkdir(path):
+ if not os.path.exists(path):
+ os.makedirs(path)
+
+if __name__ == "__main__":
+ for i in range(1000):
+ loading_bar(i, 1000, "it is loading...")
+ time.sleep(0.01)