issue #3: give option to change bar size

azamaulanaaa · Sep 30, 2020 · 4b76bd4 · 4b76bd4
1 parent eac195d
commit 4b76bd4
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/bot/__init__.py b/bot/__init__.py
@@ -16,7 +16,8 @@
     'CHAT_ID' : [],
     'EDIT_SLEEP' : 3,
     'UPLOAD_MAX_SIZE' : 2000 * 1024 * 1024,
-    'ARIA2_DIR' : 'downloads'
+    'ARIA2_DIR' : 'downloads',
+    'BAR_SIZE' : 10
 })
 
 # GOAL:

diff --git a/bot/handlers/leech_handler.py b/bot/handlers/leech_handler.py
@@ -92,8 +92,8 @@ async def progress_dl(message : Message, aria2_api : aria2.aria2, gid : int, pre
         if not download.is_complete:
             if not download.error_message:
                 block = ""
-                for i in range(1, 11):
-                    if i <= floor(download.progress/10):
+                for i in range(1, CONFIG.BAR_SIZE + 1):
+                    if i <= floor(download.progress * CONFIG.BAR_SIZE/100):
                         block += LOCAL.BLOCK_FILLED
                     else:
                         block += LOCAL.BLOCK_EMPTY

diff --git a/bot/handlers/upload_to_tg_handler.py b/bot/handlers/upload_to_tg_handler.py
@@ -117,8 +117,8 @@ async def upload_fn(chat_id, file, **kwargs):
 async def progress_upload_tg(current, total, message, info):
     percentage = round(current * 10000 / total) / 100
     block = ""
-    for i in range(1, 11):
-        if i <= floor(percentage/10):
+    for i in range(1, CONFIG.BAR_SIZE + 1):
+        if i <= floor(percentage * CONFIG.BAR_SIZE/100):
             block += LOCAL.BLOCK_FILLED
         else:
             block += LOCAL.BLOCK_EMPTY