util.py

from datetime import datetime
import traceback
import time
import os

media_types =  [
    'audio',
    'document',
    'photo',
    'sticker',
    'animation',
    'video',
    'voice',
    'video_note'
]

def time_us():
    return int(time.time() * 1000000)

def time_ms():
    return int(time_us() / 1000)

def format_duration_us(t_us):
    t_us = int(t_us)

    t_ms = t_us / 1000
    t_s = t_ms / 1000
    t_m = t_s / 60
    t_h = t_m / 60
    t_d = t_h / 24

    if t_d >= 1:
        rem_h = t_h % 24
        rem_m = t_m % 60
        rem_s = t_s % (24 * 60 * 60) % 60
        return '%dd%dh%dm%ds' % (t_d, rem_h, rem_m, rem_s)
    elif t_h >= 1:
        rem_m = t_m % 60
        rem_s = t_s % (60 * 60) % 60
        return '%dh%dm%ds' % (t_h, rem_m, rem_s)
    elif t_m >= 1:
        rem_s = t_s % 60
        return '%dm%ds' % (t_m, rem_s)
    elif t_s >= 10:
        return '%ds' % t_s
    elif t_ms >= 10:
        return '%d ms' % t_ms
    else:
        return '%d μs' % t_us

def find_prefixed_funcs(obj, prefix):
    results = []

    for sym in dir(obj):
        if sym.startswith(prefix):
            name = sym[len(prefix):]
            func = getattr(obj, sym)
            if not callable(func):
                continue

            results.append((name, func))

    return results

def filter_input_block(inp):
    if inp.startswith('```') and inp.endswith('```'):
        inp = inp[3:][:-3]
    elif inp.startswith('`') and inp.endswith('`'):
        inp = inp[1:][:-1]

    return inp

def format_exception(exp):
    tb = traceback.extract_tb(exp.__traceback__)

    # Replace absolute paths with relative paths
    cwd = os.getcwd()
    for frame in tb:
        if cwd in frame.filename:
            frame.filename = os.path.relpath(frame.filename)

    stack = ''.join(traceback.format_list(tb))
    msg = str(exp)
    if msg:
        msg = ': ' + msg

    return f'Traceback (most recent call last):\n{stack}{type(exp).__name__}{msg}'