add check_role to dispatcher
This commit is contained in:
parent
8fdc34c4d3
commit
01834c6ba7
|
@ -48,6 +48,31 @@ class Controller:
|
||||||
**kwargs,
|
**kwargs,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
async def check_role(self,
|
||||||
|
uri: str,
|
||||||
|
username: str,
|
||||||
|
**kwargs: dict,
|
||||||
|
) -> None:
|
||||||
|
# create a new config
|
||||||
|
async with await Config(dispatcher.option) as config:
|
||||||
|
await config.property.read_write()
|
||||||
|
await config.option('message').value.set(uri)
|
||||||
|
subconfig = config.option(uri)
|
||||||
|
for key, value in kwargs.items():
|
||||||
|
try:
|
||||||
|
await subconfig.option(key).value.set(value)
|
||||||
|
except AttributeError:
|
||||||
|
if get_config()['global']['debug']:
|
||||||
|
print_exc()
|
||||||
|
raise ValueError(_(f'unknown parameter in "{uri}": "{key}"'))
|
||||||
|
except ValueOptionError as err:
|
||||||
|
raise ValueError(_(f'invalid parameter in "{uri}": {err}'))
|
||||||
|
await dispatcher.check_role(subconfig,
|
||||||
|
username,
|
||||||
|
uri,
|
||||||
|
)
|
||||||
|
|
||||||
async def on_join(self,
|
async def on_join(self,
|
||||||
risotto_context,
|
risotto_context,
|
||||||
):
|
):
|
||||||
|
|
|
@ -342,7 +342,8 @@ class Dispatcher(register.RegisterDispatcher,
|
||||||
async def check_role(self,
|
async def check_role(self,
|
||||||
config: Config,
|
config: Config,
|
||||||
user_login: str,
|
user_login: str,
|
||||||
uri: str) -> None:
|
uri: str,
|
||||||
|
) -> None:
|
||||||
async with self.pool.acquire() as connection:
|
async with self.pool.acquire() as connection:
|
||||||
async with connection.transaction():
|
async with connection.transaction():
|
||||||
# Verify if user exists and get ID
|
# Verify if user exists and get ID
|
||||||
|
|
Loading…
Reference in New Issue