TelegramIndex-Fork/app/util.py

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