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

🧹 Cleanup code a bit

Browse source
This commit is contained in:
Steffo 2021-04-19 16:23:41 +02:00
parent 8961b58078
commit 093618bc39
Signed by: steffo
GPG key ID: 6965406171929D01
1 changed files with 10 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
royalpack/commands/login.py
View file

@ -210,7 +210,6 @@ async def register_user_telethon(
return tg return tg
@engi.use_database(db.lazy_session_class) @engi.use_database(db.lazy_session_class)
@engi.TeleportingConversation @engi.TeleportingConversation
async def login(*, _msg: engi.Message, _session: so.Session, _imp, **__): async def login(*, _msg: engi.Message, _session: so.Session, _imp, **__):
@ -261,15 +260,21 @@ async def login(*, _msg: engi.Message, _session: so.Session, _imp, **__):
user = await register_user_generic(session=_session, user_info=ui) user = await register_user_generic(session=_session, user_info=ui)
if isinstance(_imp, royalnet_telethon.TelethonPDAImplementation):
sender = await _msg.sender
tg = await register_user_telethon(session=_session, user_info=ui, telethon_user=sender._user)
log.debug(f"Committing session...") log.debug(f"Committing session...")
_session.commit() _session.commit()
log.debug(f"Done, notifying the user...") log.debug(f"Done, notifying the user...")
await private.send_message(text=f"✅ Login riuscito! Sei loggato come {user.name}!") await private.send_message(text=f"✅ Login riuscito! Sei loggato come {user.name}!")
if isinstance(_imp, royalnet_telethon.TelethonPDAImplementation):
sender = await _msg.sender
tg = await register_user_telethon(session=_session, user_info=ui, telethon_user=sender._user)
log.debug(f"Committing session...")
_session.commit()
log.debug(f"Done, notifying the user...")
await private.send_message(text=f"↔️ Sincronizzazione con Telegram riuscita! Sei loggato come {tg.mention()}!")
__all__ = ("login",) __all__ = ("login",)