style: 统一 Ruff 代码格式
This commit is contained in:
@@ -44,7 +44,9 @@ def _empty_task_summary() -> dict:
|
||||
|
||||
def _can_view_huya_all(user: User) -> bool:
|
||||
"""兼容旧 huya:account 全量权限。"""
|
||||
return user_has_permission(user, "huya:view_all") or user_has_permission(user, "huya:account")
|
||||
return user_has_permission(user, "huya:view_all") or user_has_permission(
|
||||
user, "huya:account"
|
||||
)
|
||||
|
||||
|
||||
@router.get("/summary")
|
||||
@@ -65,12 +67,17 @@ def dashboard_summary(
|
||||
)
|
||||
|
||||
login_tasks = _empty_task_summary()
|
||||
if any(user_has_permission(current, permission) for permission in (
|
||||
"login:batch",
|
||||
"login:view_all",
|
||||
"login:view_assigned",
|
||||
)):
|
||||
login_query = db.query(LoginTask).join(Account, LoginTask.account_id == Account.id)
|
||||
if any(
|
||||
user_has_permission(current, permission)
|
||||
for permission in (
|
||||
"login:batch",
|
||||
"login:view_all",
|
||||
"login:view_assigned",
|
||||
)
|
||||
):
|
||||
login_query = db.query(LoginTask).join(
|
||||
Account, LoginTask.account_id == Account.id
|
||||
)
|
||||
if not user_has_permission(current, "login:view_all"):
|
||||
login_query = login_query.filter(Account.assigned_to == current.id)
|
||||
login_tasks = _task_summary(login_query, LoginTask)
|
||||
@@ -87,27 +94,40 @@ def dashboard_summary(
|
||||
cookies = cookie_query.with_entities(func.count(LoginTask.id)).scalar() or 0
|
||||
|
||||
huya_accounts = 0
|
||||
can_view_huya_accounts = any(user_has_permission(current, permission) for permission in (
|
||||
"huya:account",
|
||||
"huya:view_all",
|
||||
"huya:view_assigned",
|
||||
))
|
||||
can_view_huya_accounts = any(
|
||||
user_has_permission(current, permission)
|
||||
for permission in (
|
||||
"huya:account",
|
||||
"huya:view_all",
|
||||
"huya:view_assigned",
|
||||
)
|
||||
)
|
||||
if can_view_huya_accounts:
|
||||
huya_account_query = db.query(HuyaAccount)
|
||||
if not _can_view_huya_all(current):
|
||||
huya_account_query = huya_account_query.filter(HuyaAccount.assigned_to == current.id)
|
||||
huya_accounts = huya_account_query.with_entities(func.count(HuyaAccount.id)).scalar() or 0
|
||||
huya_account_query = huya_account_query.filter(
|
||||
HuyaAccount.assigned_to == current.id
|
||||
)
|
||||
huya_accounts = (
|
||||
huya_account_query.with_entities(func.count(HuyaAccount.id)).scalar() or 0
|
||||
)
|
||||
|
||||
huya_tasks = _empty_task_summary()
|
||||
huya_goods = 0
|
||||
huya_recharge_goods = 0
|
||||
if user_has_permission(current, "huya:task"):
|
||||
huya_task_query = db.query(HuyaTask).join(HuyaAccount, HuyaTask.account_id == HuyaAccount.id)
|
||||
huya_task_query = db.query(HuyaTask).join(
|
||||
HuyaAccount, HuyaTask.account_id == HuyaAccount.id
|
||||
)
|
||||
if not _can_view_huya_all(current):
|
||||
huya_task_query = huya_task_query.filter(HuyaAccount.assigned_to == current.id)
|
||||
huya_task_query = huya_task_query.filter(
|
||||
HuyaAccount.assigned_to == current.id
|
||||
)
|
||||
huya_tasks = _task_summary(huya_task_query, HuyaTask)
|
||||
huya_goods = db.query(func.count(HuyaGoodsSnapshot.id)).scalar() or 0
|
||||
huya_recharge_goods = db.query(func.count(HuyaRechargeGoodsSnapshot.id)).scalar() or 0
|
||||
huya_recharge_goods = (
|
||||
db.query(func.count(HuyaRechargeGoodsSnapshot.id)).scalar() or 0
|
||||
)
|
||||
|
||||
return {
|
||||
"douyu": {
|
||||
|
||||
Reference in New Issue
Block a user