21 lines
657 B
Python
21 lines
657 B
Python
from typing import Union
|
|
from urllib.parse import quote
|
|
|
|
from telethon.tl.custom import Message
|
|
|
|
def get_file_name(message: Message, quote_name: bool = True) -> str:
|
|
if message.file.name:
|
|
name = message.file.name
|
|
else:
|
|
ext = message.file.ext or ""
|
|
name = f"{message.date.strftime('%Y-%m-%d_%H:%M:%S')}{ext}"
|
|
return quote(name) if quote_name else name
|
|
|
|
def get_human_size(num: Union[int, float]) -> str:
|
|
base = 1024.0
|
|
sufix_list = ["B", "KiB", "MiB", "GiB", "TiB", "PiB", "EiB", "ZiB", "YiB"]
|
|
for unit in sufix_list:
|
|
if abs(num) < base:
|
|
return f"{round(num, 2)} {unit}"
|
|
num /= base
|