evanferrao
/
dump-mirrorbot
mirror of https://github.com/Box-boi/dump-mirrorbot.git
1
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

Fix disk_usage size unit conversion logic 

Signed-off-by: lzzy12 <jhashivam2020@gmail.com>
This commit is contained in:
lzzy12 2019-10-07 22:05:36 +05:30
parent 80388249c5
commit 1761d16f70
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
bot/__main__.py
View File

@ -1,6 +1,7 @@
from telegram.ext import CommandHandler, run_async from telegram.ext import CommandHandler, run_async
from bot import dispatcher, LOGGER, updater from bot import dispatcher, LOGGER, updater
from bot.helper.ext_utils import fs_utils from bot.helper.ext_utils import fs_utils
from .helper.ext_utils.bot_utils import get_readable_file_size
import signal import signal
import time import time
from bot.helper.telegram_helper.message_utils import * from bot.helper.telegram_helper.message_utils import *
@ -11,14 +12,13 @@ from .modules import authorize, list, cancel_mirror, mirror_status, mirror
@run_async @run_async
def disk_usage(update, context): def disk_usage(update, context):
total, used, free = shutil.disk_usage('/') total, used, free = shutil.disk_usage('.')
divider = 1024*1024*1024 total = get_readable_file_size(total)
total //= divider used = get_readable_file_size(used)
used //= divider free = get_readable_file_size(free)
free //= divider disk_usage_string = f'Total disk space: {total}\n' \
disk_usage_string = f'Total disk space: {total} GBs\n' \ f'Used: {used}\n' \
f'Used: {used} GBs\n' \ f'Free: {free}'
f'Free: {free} GBs'
sendMessage(disk_usage_string, context, update) sendMessage(disk_usage_string, context, update)