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

Implement DiscordCall.get_author

Browse source
This commit is contained in:
Steffo 2019-04-11 18:40:51 +02:00
parent 818b504ec5
commit 0e00e69650
1 changed files with 16 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
royalnet/bots/discord.py
View file

@ -21,7 +21,7 @@ class DiscordBot:
pass pass
def __init__(self, def __init__(self,
api_key: str, token: str,
master_server_uri: str, master_server_uri: str,
master_server_secret: str, master_server_secret: str,
commands: typing.List[typing.Type[Command]], commands: typing.List[typing.Type[Command]],
@ -31,6 +31,7 @@ class DiscordBot:
identity_column_name: str, identity_column_name: str,
missing_command: typing.Type[Command] = NullCommand, missing_command: typing.Type[Command] = NullCommand,
error_command: typing.Type[Command] = NullCommand): error_command: typing.Type[Command] = NullCommand):
self.token = token
self.bot = self.DiscordClient() self.bot = self.DiscordClient()
self.network: RoyalnetLink = RoyalnetLink(master_server_uri, master_server_secret, "discord", todo) self.network: RoyalnetLink = RoyalnetLink(master_server_uri, master_server_secret, "discord", todo)
# Generate commands # Generate commands
@ -70,10 +71,21 @@ class DiscordBot:
response = await self.network.request(message, destination) response = await self.network.request(message, destination)
return response return response
async def get_author(self, error_if_none=False): async def get_author(call, error_if_none=False):
raise NotImplementedError() message: discord.Message = call.kwargs["message"]
user: discord.Member = message.author
query = call.session.query(self.master_table)
for link in self.identity_chain:
query = query.join(link.mapper.class_)
query = query.filter(self.identity_column == user.id)
result = await asyncify(query.one_or_none)
if result is None and error_if_none:
raise UnregisteredError("Author is not registered")
return result
self.DiscordCall = DiscordCall self.DiscordCall = DiscordCall
async def run(self): async def run(self):
raise NotImplementedError() await self.bot.login(self.token)
await self.bot.connect()
# TODO: how to stop?