mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-27 13:34:28 +00:00
48 lines
1.9 KiB
Python
48 lines
1.9 KiB
Python
|
from telegram import Bot, Update
|
||
|
|
from telegram.ext import Updater, CommandHandler
|
||
|
|
import logging
|
||
|
|
from db import session, Royal, Telegram
|
||
|
|
|
||
|
|
# Init the config reader
|
||
|
|
import configparser
|
||
|
|
config = configparser.ConfigParser()
|
||
|
|
config.read("config.ini")
|
||
|
|
|
||
|
|
# Init the logger
|
||
|
|
logging.basicConfig(level=logging.DEBUG, format="[%(levelname)s] %(asctime)s | %(message)s")
|
||
|
|
|
||
|
|
# Init the Telegram Bot
|
||
|
|
updater = Updater(token=config["Telegram"]["bot_token"])
|
||
|
|
|
||
|
|
def message_sync(bot: Bot, update: Update):
|
||
|
|
tg_user = update.message.from_user
|
||
|
|
db_user = session.query(Telegram).filter(Telegram.telegram_id == tg_user.id).first()
|
||
|
|
if db_user is None:
|
||
|
|
# Find the royals table record matching the command argument
|
||
|
|
try:
|
||
|
|
tg_royal = update.message.text.split(" ", 1)[1]
|
||
|
|
except IndexError:
|
||
|
|
bot.send_message(update.message.chat.id, "⚠️ Non hai specificato nessun username!")
|
||
|
|
return
|
||
|
|
db_royal = session.query(Royal).filter(Royal.username == tg_royal).first()
|
||
|
|
if db_royal is None:
|
||
|
|
bot.send_message(update.message.chat.id, "⚠️ L'username che hai specificato non è valido!")
|
||
|
|
return
|
||
|
|
# Create the new user and link it to the royals table
|
||
|
|
db_user = Telegram(royal_id=db_royal.id,
|
||
|
|
telegram_id=tg_user.id,
|
||
|
|
first_name=tg_user.first_name,
|
||
|
|
last_name=tg_user.last_name,
|
||
|
|
username=tg_user.username)
|
||
|
|
session.add(db_user)
|
||
|
|
session.commit()
|
||
|
|
else:
|
||
|
|
# Update user data
|
||
|
|
db_user.first_name = tg_user.first_name
|
||
|
|
db_user.last_name = tg_user.last_name
|
||
|
|
db_user.username = tg_user.username
|
||
|
|
session.commit()
|
||
|
|
bot.send_message(update.message.chat.id, "✅ Sincronizzazione completata!")
|
||
|
|
|
||
|
|
updater.dispatcher.add_handler(CommandHandler("sync", message_sync))
|
||
|
|
updater.start_polling()
|