fix: auto-fix code issues (cron)

- 修复重复导入/字段
- 修复异常处理
- 修复PEP8格式问题
- 添加类型注解
This commit is contained in:
AutoFix Bot
2026-03-03 06:03:38 +08:00
parent 2a0ed6af4d
commit 9fd1da8fb7
41 changed files with 901 additions and 768 deletions

View File

@@ -388,16 +388,16 @@ class TenantManager:
cursor.execute("CREATE INDEX IF NOT EXISTS idx_tenants_owner ON tenants(owner_id)")
cursor.execute("CREATE INDEX IF NOT EXISTS idx_tenants_status ON tenants(status)")
cursor.execute(
"CREATE INDEX IF NOT EXISTS idx_domains_tenant ON tenant_domains(tenant_id)"
"CREATE INDEX IF NOT EXISTS idx_domains_tenant ON tenant_domains(tenant_id)",
)
cursor.execute(
"CREATE INDEX IF NOT EXISTS idx_domains_domain ON tenant_domains(domain)"
"CREATE INDEX IF NOT EXISTS idx_domains_domain ON tenant_domains(domain)",
)
cursor.execute(
"CREATE INDEX IF NOT EXISTS idx_domains_status ON tenant_domains(status)"
"CREATE INDEX IF NOT EXISTS idx_domains_status ON tenant_domains(status)",
)
cursor.execute(
"CREATE INDEX IF NOT EXISTS idx_members_tenant ON tenant_members(tenant_id)"
"CREATE INDEX IF NOT EXISTS idx_members_tenant ON tenant_members(tenant_id)",
)
cursor.execute("CREATE INDEX IF NOT EXISTS idx_members_user ON tenant_members(user_id)")
cursor.execute("CREATE INDEX IF NOT EXISTS idx_usage_tenant ON tenant_usage(tenant_id)")
@@ -433,7 +433,7 @@ class TenantManager:
TenantTier(tier) if tier in [t.value for t in TenantTier] else TenantTier.FREE
)
resource_limits = self.DEFAULT_LIMITS.get(
tier_enum, self.DEFAULT_LIMITS[TenantTier.FREE]
tier_enum, self.DEFAULT_LIMITS[TenantTier.FREE],
)
tenant = Tenant(
@@ -612,7 +612,7 @@ class TenantManager:
conn.close()
def list_tenants(
self, status: str | None = None, tier: str | None = None, limit: int = 100, offset: int = 0
self, status: str | None = None, tier: str | None = None, limit: int = 100, offset: int = 0,
) -> list[Tenant]:
"""列出租户"""
conn = self._get_connection()
@@ -1103,7 +1103,7 @@ class TenantManager:
conn.close()
def update_member_role(
self, tenant_id: str, member_id: str, role: str, permissions: list[str] | None = None
self, tenant_id: str, member_id: str, role: str, permissions: list[str] | None = None,
) -> bool:
"""更新成员角色"""
conn = self._get_connection()
@@ -1209,7 +1209,7 @@ class TenantManager:
**asdict(tenant),
"member_role": row["role"],
"member_status": row["member_status"],
}
},
)
return result
@@ -1268,7 +1268,7 @@ class TenantManager:
conn.close()
def get_usage_stats(
self, tenant_id: str, start_date: datetime | None = None, end_date: datetime | None = None
self, tenant_id: str, start_date: datetime | None = None, end_date: datetime | None = None,
) -> dict[str, Any]:
"""获取使用统计"""
conn = self._get_connection()
@@ -1314,23 +1314,23 @@ class TenantManager:
"limits": limits,
"usage_percentages": {
"storage": self._calc_percentage(
row["total_storage"] or 0, limits.get("max_storage_mb", 0) * 1024 * 1024
row["total_storage"] or 0, limits.get("max_storage_mb", 0) * 1024 * 1024,
),
"transcription": self._calc_percentage(
row["total_transcription"] or 0,
limits.get("max_transcription_minutes", 0) * 60,
),
"api_calls": self._calc_percentage(
row["total_api_calls"] or 0, limits.get("max_api_calls_per_day", 0)
row["total_api_calls"] or 0, limits.get("max_api_calls_per_day", 0),
),
"projects": self._calc_percentage(
row["max_projects"] or 0, limits.get("max_projects", 0)
row["max_projects"] or 0, limits.get("max_projects", 0),
),
"entities": self._calc_percentage(
row["max_entities"] or 0, limits.get("max_entities", 0)
row["max_entities"] or 0, limits.get("max_entities", 0),
),
"members": self._calc_percentage(
row["max_members"] or 0, limits.get("max_team_members", 0)
row["max_members"] or 0, limits.get("max_team_members", 0),
),
},
}