51 lines
2.1 KiB
Python
51 lines
2.1 KiB
Python
import telegram
|
|
import yaml
|
|
from telegram import Message, User, Chat
|
|
from telegram.error import Unauthorized
|
|
|
|
from config import telegram_token
|
|
|
|
try:
|
|
with open("db.yaml") as f:
|
|
db = yaml.safe_load(f)
|
|
except FileNotFoundError:
|
|
db = {"subscribed": {}, "last_update": 0}
|
|
|
|
bot = telegram.Bot(token=telegram_token)
|
|
for update in bot.get_updates(offset=db["last_update"] + 1):
|
|
message: Message = update.message
|
|
text = message.text
|
|
user: User = update.effective_user
|
|
chat: Chat = update.effective_chat
|
|
if "/subscribe" in text:
|
|
if chat.id in db["subscribed"]:
|
|
bot.sendMessage(chat_id=chat.id,
|
|
text="Du bekommst bereits die Nachrichten. Verwende /unsubscribe zum Abmelden.")
|
|
else:
|
|
subscriber = {
|
|
"first_name": chat.first_name,
|
|
"last_name": chat.last_name,
|
|
"username": chat.username,
|
|
}
|
|
db["subscribed"][chat.id] = subscriber
|
|
bot.send_message(chat_id=chat.id,
|
|
text="Du bekommst nun regelmäßige Nachrichten. Verwende /unsubscribe zum Abmelden.")
|
|
elif "/unsubscribe" in text:
|
|
if chat.id in db["subscribed"]:
|
|
del db["subscribed"][chat.id]
|
|
bot.sendMessage(chat_id=chat.id,
|
|
text="Du bekommst keine Nachrichten mehr. Verwende /subscribe um sie wieder zu bekommen.")
|
|
else:
|
|
bot.sendMessage(chat_id=chat.id,
|
|
text="Du bekommst bereits keine Nachrichten. Verwende /subscribe um sie wieder zu bekommen.")
|
|
elif "/start" in text or "/help" in text:
|
|
try:
|
|
bot.sendMessage(chat_id=chat.id,
|
|
text="Verwende /subscribe um regelmäßige Nachrichten zu bekommen. Mit /unsubscribe kannst du dich wieder abmelden. "
|
|
"Es kann bis zu 5 Minuten dauern, bis eine Bestätigung kommt.")
|
|
except Unauthorized:
|
|
pass
|
|
db["last_update"] = update.update_id
|
|
|
|
with open("db.yaml", "w") as f:
|
|
yaml.safe_dump(db, f)
|