tools.py

from __future__ import annotations
from datetime import time, datetime
from typing import TYPE_CHECKING, Sequence

if TYPE_CHECKING:
    from telegram import Update

from config import TG_CHAT_ID, TIMEZONE

TIMESET_HELP_MSG = "You should provide your command \
with hours minutes and seconds in this fashon:\n \
/your_command HH:MM:SS \
"

def validate_user(func):
    async def wrapper(update: Update, *args, **kwargs):
        if update.effective_chat.id == TG_CHAT_ID:
            return await func(update, *args, **kwargs)

    return wrapper

def protect_for_html(text_data):
    return text_data.replace('&', '&')\
                    .replace('<', '&lt;')\
                    .replace('>', '&gt;')

def singleton(cls, *args, **kw):
    instances = {}
    def _singleton(*args, **kw):
        if cls not in instances:
            instances[cls] = cls(*args, **kw)
        return instances[cls]
    return _singleton

def time_from_args(args: Sequence) -> time | str:
    try:
        digits = map(int, args[0].split(':'))
        return time(*digits, tzinfo=TIMEZONE)
    except IndexError:
        return TIMESET_HELP_MSG
    except ValueError as ve:
        if "invalid literal for int()" in ve.args[0]:
            return TIMESET_HELP_MSG
        return str(ve.args[0])

def dt_from_time(time: time):
    return datetime.combine(
        date=datetime.now(),
        time=time,
        tzinfo=TIMEZONE)