fix: auto-fix code issues (cron)
- 修复重复导入/字段 - 修复异常处理 - 修复PEP8格式问题 - 添加类型注解
This commit is contained in:
@@ -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),
|
||||
),
|
||||
},
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user