astra
/
tgsupportbot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

.

master
sfan5 2024-07-20 14:30:49 +02:00
parent 5d86bf140c
commit c95b089e47
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
supportbot/bot.py
View File

@ -21,9 +21,10 @@ bot_self_id: int = None
target_group: Optional[int] = None target_group: Optional[int] = None
welcome_text: str = None welcome_text: str = None
reply_text: str = None reply_text: str = None
integration_fmt: Optional[str] = None
def init(config: dict, _db): def init(config: dict, _db):
global bot, db, bot_self_id, target_group, welcome_text, reply_text global bot, db, bot_self_id, target_group, welcome_text, reply_text, integration_fmt
if not config.get("bot_token"): if not config.get("bot_token"):
logging.error("No telegram token specified.") logging.error("No telegram token specified.")
exit(1) exit(1)
@ -36,6 +37,7 @@ def init(config: dict, _db):
target_group = int(config["target_group"]) target_group = int(config["target_group"])
welcome_text = config["welcome_text"] welcome_text = config["welcome_text"]
reply_text = config["reply_text"] reply_text = config["reply_text"]
integration_fmt = config.get("integration_fmt")
set_handler(handle_msg, content_types=ALL_CONTENT_TYPES) set_handler(handle_msg, content_types=ALL_CONTENT_TYPES)
bot_self_id = bot.get_me().id bot_self_id = bot.get_me().id
@ -264,7 +266,7 @@ def handle_private(ev: TMessage):
with db_modify_user(user.id) as user: with db_modify_user(user.id) as user:
user.last_messaged = now user.last_messaged = now
def handle_private_command(ev: TMessage, user, c): def handle_private_command(ev: TMessage, user: User, c):
if c == "start": if c == "start":
callwrapper(lambda: bot.send_message(ev.chat.id, welcome_text, parse_mode="HTML")) callwrapper(lambda: bot.send_message(ev.chat.id, welcome_text, parse_mode="HTML"))
return True return True
@ -297,7 +299,7 @@ def escape_html(s):
def format_datetime(dt): def format_datetime(dt):
return dt.strftime("%Y-%m-%d %H:%M:%S") return dt.strftime("%Y-%m-%d %H:%M:%S")
def format_user_info(user): def format_user_info(user: User):
realname = user.realname realname = user.realname
if not str_is_printable(realname): if not str_is_printable(realname):
realname = "<empty name>" realname = "<empty name>"
@ -305,5 +307,7 @@ def format_user_info(user):
user.id, escape_html(realname)) user.id, escape_html(realname))
if user.username is not None: if user.username is not None:
s += " (@%s)" % escape_html(user.username) s += " (@%s)" % escape_html(user.username)
if integration_fmt:
s += "\n\u27a4 " + (integration_fmt % user.id)
s += "\nID: <code>%d</code>" % user.id s += "\nID: <code>%d</code>" % user.id
return s return s