changed colours a bit

1 files changed, 6 insertions, 4 deletions

diff --git a/src/util.py b/src/util.py
index de05b08..3badca0 100644
--- a/src/util.py
+++ b/src/util.py
@@ -4,8 +4,11 @@ import colors
 import time
 import os
 
+DEFAULT_BAR_COLOR = colors.BLACK + colors.BG_CYAN
+DEFAULT_BAR_COLOR_RESET = colors.BG_BLACK + colors.CYAN
+
 def loading_bar(completed, total, text, 
-        unit=""):
+        unit="", color=DEFAULT_BAR_COLOR, reset=DEFAULT_BAR_COLOR_RESET):
 
     columns, rows = shutil.get_terminal_size((80, 20))
     
@@ -14,11 +17,10 @@ def loading_bar(completed, total, text,
     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))]) 
+    info = "".join([info[i] + (reset if i == reset_at else "") for i in range(len(info))]) 
 
-    print(colors.BLACK + colors.BG_GREEN + info, end="\r")
+    print(color + info, end="\r")