129 lines
3.3 KiB
Python
129 lines
3.3 KiB
Python
from bot import download_dict, download_dict_lock
|
|
import logging
|
|
import re
|
|
|
|
LOGGER = logging.getLogger(__name__)
|
|
|
|
MAGNET_REGEX = r"magnet:\?xt=urn:btih:[a-zA-Z0-9]*"
|
|
|
|
URL_REGEX = r"(?:(?:https?|ftp):\/\/)?[\w/\-?=%.]+\.[\w/\-?=%.]+"
|
|
|
|
class MirrorStatus:
|
|
STATUS_UPLOADING = "Uploading"
|
|
STATUS_DOWNLOADING = "Downloading"
|
|
STATUS_WAITING = "Queued"
|
|
STATUS_FAILED = "Failed. Cleaning download"
|
|
STATUS_CANCELLED = "Cancelled"
|
|
STATUS_ARCHIVING = "Archiving"
|
|
|
|
|
|
PROGRESS_MAX_SIZE = 100 // 8
|
|
PROGRESS_INCOMPLETE = ['▏', '▎', '▍', '▌', '▋', '▊', '▉']
|
|
|
|
SIZE_UNITS = ['B', 'KB', 'MB', 'GB', 'TB', 'PB']
|
|
|
|
|
|
def get_readable_file_size(size_in_bytes) -> str:
|
|
index = 0
|
|
while size_in_bytes >= 1024:
|
|
size_in_bytes /= 1024
|
|
index += 1
|
|
try:
|
|
return f'{round(size_in_bytes, 2)} {SIZE_UNITS[index]}'
|
|
except IndexError:
|
|
return 'File too large'
|
|
|
|
|
|
def get_download(message_id):
|
|
with download_dict_lock:
|
|
return download_dict[message_id].download()
|
|
|
|
|
|
def get_download_status_list():
|
|
with download_dict_lock:
|
|
return list(download_dict.values())
|
|
|
|
|
|
def get_progress_bar_string(status):
|
|
if status.status() == MirrorStatus.STATUS_UPLOADING:
|
|
completed = status.upload_helper.uploaded_bytes / 8
|
|
else:
|
|
completed = status.download().completed_length / 8
|
|
total = status.download().total_length / 8
|
|
if total == 0:
|
|
p = 0
|
|
else:
|
|
p = round(completed * 100 / total)
|
|
p = min(max(p, 0), 100)
|
|
cFull = p // 8
|
|
cPart = p % 8 - 1
|
|
p_str = '█' * cFull
|
|
if cPart >= 0:
|
|
p_str += PROGRESS_INCOMPLETE[cPart]
|
|
p_str += ' ' * (PROGRESS_MAX_SIZE - cFull)
|
|
p_str = f"[{p_str}]"
|
|
return p_str
|
|
|
|
|
|
def get_download_index(_list, gid):
|
|
index = 0
|
|
for i in _list:
|
|
if i.download().gid == gid:
|
|
return index
|
|
index += 1
|
|
|
|
|
|
def get_download_str():
|
|
result = ""
|
|
with download_dict_lock:
|
|
for status in list(download_dict.values()):
|
|
result += (status.progress() + status.speed() + status.status())
|
|
return result
|
|
|
|
|
|
def get_readable_message():
|
|
with download_dict_lock:
|
|
progress_list = list(download_dict.values())
|
|
msg = ''
|
|
for download in progress_list:
|
|
msg += f'<b>Name:</b> {download.name()}\n' \
|
|
f'<b>Status:</b> {download.status()}\n' \
|
|
f'<code>{get_progress_bar_string(download)}</code> {download.progress()} of {download.size()}\n' \
|
|
f'<b>Speed:</b> {download.speed()}\n' \
|
|
f'<b>ETA:</b> {download.eta()}\n\n'
|
|
# LOGGER.info(msg)
|
|
return msg
|
|
|
|
|
|
def get_readable_time(seconds: int) -> str:
|
|
result = ''
|
|
(days, remainder) = divmod(seconds, 86400)
|
|
days = int(days)
|
|
if days != 0:
|
|
result += f'{days}d'
|
|
(hours, remainder) = divmod(remainder, 3600)
|
|
hours = int(hours)
|
|
if hours != 0:
|
|
result += f'{hours}h'
|
|
(minutes, seconds) = divmod(remainder, 60)
|
|
minutes = int(minutes)
|
|
if minutes != 0:
|
|
result += f'{minutes}m'
|
|
seconds = int(seconds)
|
|
result += f'{seconds}s'
|
|
return result
|
|
|
|
|
|
def is_url(url: str):
|
|
url = re.findall(URL_REGEX,url)
|
|
if url:
|
|
return True
|
|
return False
|
|
|
|
|
|
def is_magnet(url: str):
|
|
magnet = re.findall(MAGNET_REGEX,url)
|
|
if magnet:
|
|
return True
|
|
return False
|