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

.

master
sfan5 2021-02-17 16:06:02 +01:00
parent 4a09534362
commit cc6a04ffac
1 changed files with 13 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
src/core.py
View File

@ -6,7 +6,7 @@ from datetime import datetime, timedelta
LONG_LONG_TIME = datetime(2100, 1, 1) LONG_LONG_TIME = datetime(2100, 1, 1)
ID_REMIND_DURATION = timedelta(days=2) ID_REMIND_DURATION = timedelta(days=2)
ALL_CONTENT_TYPES = ["text", "location", "venue", "contact", ALL_CONTENT_TYPES = ["text", "location", "venue", "contact", "animation",
"audio", "document", "photo", "sticker", "video", "video_note", "voice"] "audio", "document", "photo", "sticker", "video", "video_note", "voice"]
bot = None bot = None
@ -155,13 +155,7 @@ def handle_group(ev):
# handle commands # handle commands
if ev.content_type == "text" and ev.text.startswith("/"): if ev.content_type == "text" and ev.text.startswith("/"):
arg = "" c, _, arg = ev.text[1:].partition(" ")
if " " in ev.text:
pos = ev.text.find(" ")
c = ev.text[1:pos].lower()
arg = ev.text[pos+1:]
else:
c = ev.text[1:].lower()
return handle_group_command(ev, user_id, c, arg) return handle_group_command(ev, user_id, c, arg)
# deliver message # deliver message
@ -225,8 +219,7 @@ def handle_private(ev):
# handle commands # handle commands
if ev.content_type == "text" and ev.text.startswith("/"): if ev.content_type == "text" and ev.text.startswith("/"):
pos = ev.text.find(" ") if " " in ev.text else len(ev.text) c = ev.text[1:].split(" ", 2)[0]
c = ev.text[1:pos].lower()
if handle_private_command(ev, user, c): if handle_private_command(ev, user, c):
return return
@ -305,10 +298,14 @@ def resend_message(chat_id, ev):
photo = sorted(ev.photo, key=lambda e: e.width*e.height, reverse=True)[0] photo = sorted(ev.photo, key=lambda e: e.width*e.height, reverse=True)[0]
return bot.send_photo(chat_id, photo.file_id, caption=ev.caption) return bot.send_photo(chat_id, photo.file_id, caption=ev.caption)
elif ev.content_type == "audio": elif ev.content_type == "audio":
kwargs = {} kwargs = {
for prop in ["performer", "title"]: "caption": ev.caption,
kwargs[prop] = getattr(ev.audio, prop) "performer": ev.audio.performer,
return bot.send_audio(chat_id, ev.audio.file_id, caption=ev.caption) "title": ev.audio.performer,
}
return bot.send_audio(chat_id, ev.audio.file_id, **kwargs)
elif ev.content_type == "animation":
return bot.send_animation(chat_id, ev.animation.file_id, caption=ev.caption)
elif ev.content_type == "document": elif ev.content_type == "document":
return bot.send_document(chat_id, ev.document.file_id, caption=ev.caption) return bot.send_document(chat_id, ev.document.file_id, caption=ev.caption)
elif ev.content_type == "video": elif ev.content_type == "video":
@ -328,12 +325,12 @@ def resend_message(chat_id, ev):
"latitude": ev.venue.location.latitude, "latitude": ev.venue.location.latitude,
"longitude": ev.venue.location.longitude, "longitude": ev.venue.location.longitude,
} }
for prop in ["title", "address", "foursquare_id"]: for prop in ("title", "address", "foursquare_id", "foursquare_type", "google_place_id", "google_place_type"):
kwargs[prop] = getattr(ev.venue, prop) kwargs[prop] = getattr(ev.venue, prop)
return bot.send_venue(chat_id, **kwargs) return bot.send_venue(chat_id, **kwargs)
elif ev.content_type == "contact": elif ev.content_type == "contact":
kwargs = {} kwargs = {}
for prop in ["phone_number", "first_name", "last_name"]: for prop in ("phone_number", "first_name", "last_name"):
kwargs[prop] = getattr(ev.contact, prop) kwargs[prop] = getattr(ev.contact, prop)
return bot.send_contact(chat_id, **kwargs) return bot.send_contact(chat_id, **kwargs)
elif ev.content_type == "sticker": elif ev.content_type == "sticker":