forked from mttu-developers/konabot
修正若干拼写错误,增强相关逻辑
This commit is contained in:
@ -91,7 +91,10 @@ def create_startup(): # pragma: no cover
|
||||
|
||||
@driver.on_shutdown
|
||||
async def _():
|
||||
try:
|
||||
await db.close_all_connections()
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
|
||||
DepPermManager = Annotated[PermManager, Depends(perm_manager)]
|
||||
|
||||
@ -121,7 +121,7 @@ class PermRepo:
|
||||
async def get_entity_id_batch(
|
||||
self, entities: list[PermEntity]
|
||||
) -> dict[PermEntity, int]:
|
||||
"""批量获取 Entity 的 eneity_id
|
||||
"""批量获取 Entity 的 entity_id
|
||||
|
||||
Args:
|
||||
entities: PermEntity 列表
|
||||
@ -130,10 +130,14 @@ class PermRepo:
|
||||
字典,键为 PermEntity,值为对应的 ID
|
||||
"""
|
||||
|
||||
for entity in entities:
|
||||
await self.conn.execute(
|
||||
# for entity in entities:
|
||||
# await self.conn.execute(
|
||||
# s("create_entity.sql"),
|
||||
# (entity.platform, entity.entity_type, entity.external_id),
|
||||
# )
|
||||
await self.conn.executemany(
|
||||
s("create_entity.sql"),
|
||||
(entity.platform, entity.entity_type, entity.external_id),
|
||||
[(e.platform, e.entity_type, e.external_id) for e in entities],
|
||||
)
|
||||
await self.conn.commit()
|
||||
val_placeholders = ", ".join(["(?, ?, ?)"] * len(entities))
|
||||
|
||||
@ -58,7 +58,7 @@ def make_formatter(parent: PermEntity):
|
||||
|
||||
|
||||
@cmd.assign("list")
|
||||
async def list_permisison(
|
||||
async def list_permission(
|
||||
pm: DepPermManager,
|
||||
ec: _DepEntityChain,
|
||||
event: Event,
|
||||
@ -71,7 +71,7 @@ async def list_permisison(
|
||||
|
||||
|
||||
@cmd.assign("get")
|
||||
async def get_permisison(
|
||||
async def get_permission(
|
||||
pm: DepPermManager,
|
||||
ec: _DepEntityChain,
|
||||
perm: str,
|
||||
@ -96,7 +96,7 @@ async def get_permisison(
|
||||
|
||||
|
||||
@cmd.assign("set")
|
||||
async def set_permisison(
|
||||
async def set_permission(
|
||||
pm: DepPermManager,
|
||||
ec: _DepEntityChain,
|
||||
perm: str,
|
||||
@ -105,8 +105,8 @@ async def set_permisison(
|
||||
):
|
||||
if any(i == val.lower() for i in ("y", "yes", "allow", "true", "t")):
|
||||
await pm.update_permission(ec[0], perm, True)
|
||||
if any(i == val.lower() for i in ("n", "no", "deny", "false", "f")):
|
||||
elif any(i == val.lower() for i in ("n", "no", "deny", "false", "f")):
|
||||
await pm.update_permission(ec[0], perm, False)
|
||||
if any(i == val.lower() for i in ("null", "none")):
|
||||
elif any(i == val.lower() for i in ("null", "none")):
|
||||
await pm.update_permission(ec[0], perm, None)
|
||||
await get_permisison(pm, ec, perm, event)
|
||||
await get_permission(pm, ec, perm, event)
|
||||
|
||||
Reference in New Issue
Block a user