ryg/royalnet
ryg/royalnet
1
Fork 0
mirror of https://github.com/RYGhub/royalnet.git synced 2024-11-23 19:44:20 +00:00
Code Activity

Add Discord sync command

Browse source
This commit is contained in:
Steffo 2019-04-11 19:21:09 +02:00
parent 34b04acd5d
commit 526a2a493a
1 changed files with 35 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
royalnet/commands/sync.py
View file

@ -2,7 +2,7 @@ import typing
from telegram import Update, User
from discord import Message, Member
from ..utils import Command, Call, asyncify, UnsupportedError
from ..database.tables import Royal, Telegram
from ..database.tables import Royal, Telegram, Discord
class SyncCommand(Command):
@ -11,7 +11,7 @@ class SyncCommand(Command):
command_description = "Connetti il tuo account attuale a Royalnet!"
command_syntax = "(royalnetusername)"
require_alchemy_tables = {Royal, Telegram}
require_alchemy_tables = {Royal, Telegram, Discord}
@classmethod
async def common(cls, call: Call):
@ -27,7 +27,7 @@ class SyncCommand(Command):
# Find the Royal
royal = await asyncify(call.session.query(call.alchemy.Royal).filter_by(username=call.args[0]).one_or_none)
if royal is None:
await call.reply("⚠️ Non esiste alcun account Royalnet con quel nome.")
await call.reply("⚠️ Non esiste alcun account Royalnet con quel nome. Ricorda, gli username sono [b]case-sensitive[/b]!")
return
# Find if the user is already synced
telegram = await asyncify(call.session.query(call.alchemy.Telegram).filter_by(tg_id=user.id).one_or_none)
@ -43,10 +43,38 @@ class SyncCommand(Command):
await call.reply(f"✅ Connessione completata: {str(royal)} ⬌ {str(telegram)}")
else:
# Update the Telegram data
# Avatar is WIP
telegram.tg_first_name = user.first_name
telegram.tg_last_name = user.last_name
telegram.tg_username = user.username
telegram.first_name = user.first_name
telegram.last_name = user.last_name
telegram.username = user.username
await call.reply(f"✅ Dati di {str(telegram)} aggiornati.")
# Commit the session
await asyncify(call.session.commit)
@classmethod
async def discord(cls, call: Call):
message: Message = call.kwargs["message"]
user: typing.Optional[Member] = message.author
# Find the Royal
royal = await asyncify(call.session.query(call.alchemy.Royal).filter_by(username=call.args[0]).one_or_none)
if royal is None:
await call.reply("⚠️ Non esiste alcun account Royalnet con quel nome. Ricorda, gli username sono [b]case-sensitive[/b]!")
return
# Find if the user is already synced
discord = await asyncify(call.session.query(call.alchemy.Discord).filter_by(discord_id=user.id).one_or_none)
if discord is None:
# Create a Discord to connect to the Royal
discord = call.alchemy.Discord(royal=royal,
discord_id=user.id,
username=user.name,
discriminator=user.discriminator,
avatar_hash=user.avatar)
call.session.add(discord)
await call.reply(f"✅ Connessione completata: {str(royal)} ⬌ {str(discord)}")
else:
# Update the Discord data
discord.username = user.name
discord.discriminator = user.discriminator
discord.avatar_hash = user.avatar
await call.reply(f"✅ Dati di {str(discord)} aggiornati.")
# Commit the session
await asyncify(call.session.commit)