summaryrefslogtreecommitdiff
path: root/src/home/telegram/_botutil.py
blob: 6d1ee8fb6935fa983a44d23b6fc41b2ffb090df4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
import logging
import traceback

from html import escape
from telegram import User
from home.api import WebAPIClient as APIClient
from home.api.types import BotType
from home.api.errors import ApiResponseError

_logger = logging.getLogger(__name__)


def user_any_name(user: User) -> str:
    name = [user.first_name, user.last_name]
    name = list(filter(lambda s: s is not None, name))
    name = ' '.join(name).strip()

    if not name:
        name = user.username

    if not name:
        name = str(user.id)

    return name


class ReportingHelper:
    def __init__(self, client: APIClient, bot_type: BotType):
        self.client = client
        self.bot_type = bot_type

    def report(self, message, text: str = None) -> None:
        if text is None:
            text = message.text
        try:
            self.client.log_bot_request(self.bot_type, message.chat_id, text)
        except ApiResponseError as error:
            _logger.exception(error)


def exc2text(e: Exception) -> str:
    tb = ''.join(traceback.format_tb(e.__traceback__))
    return f'{e.__class__.__name__}: ' + escape(str(e)) + "\n\n" + escape(tb)


class IgnoreMarkup:
    pass