fix: auto-fix code issues (cron)

- 修复重复导入/字段
- 修复异常处理
- 修复PEP8格式问题
- 添加类型注解
- 修复缺失的urllib.parse导入
This commit is contained in:
OpenClaw Bot
2026-02-28 06:03:09 +08:00
parent ff83cab6c7
commit fe3d64a1d2
41 changed files with 4501 additions and 1176 deletions

View File

@@ -17,6 +17,7 @@ from tenant_manager import get_tenant_manager
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
def test_tenant_management():
"""测试租户管理功能"""
print("=" * 60)
@@ -28,10 +29,7 @@ def test_tenant_management():
# 1. 创建租户
print("\n1.1 创建租户...")
tenant = manager.create_tenant(
name="Test Company",
owner_id="user_001",
tier="pro",
description="A test company tenant"
name="Test Company", owner_id="user_001", tier="pro", description="A test company tenant"
)
print(f"✅ 租户创建成功: {tenant.id}")
print(f" - 名称: {tenant.name}")
@@ -55,9 +53,7 @@ def test_tenant_management():
# 4. 更新租户
print("\n1.4 更新租户信息...")
updated = manager.update_tenant(
tenant_id=tenant.id,
name="Test Company Updated",
tier="enterprise"
tenant_id=tenant.id, name="Test Company Updated", tier="enterprise"
)
assert updated is not None, "更新租户失败"
print(f"✅ 租户更新成功: {updated.name}, 层级: {updated.tier}")
@@ -69,6 +65,7 @@ def test_tenant_management():
return tenant.id
def test_domain_management(tenant_id: str):
"""测试域名管理功能"""
print("\n" + "=" * 60)
@@ -79,11 +76,7 @@ def test_domain_management(tenant_id: str):
# 1. 添加域名
print("\n2.1 添加自定义域名...")
domain = manager.add_domain(
tenant_id=tenant_id,
domain="test.example.com",
is_primary=True
)
domain = manager.add_domain(tenant_id=tenant_id, domain="test.example.com", is_primary=True)
print(f"✅ 域名添加成功: {domain.domain}")
print(f" - ID: {domain.id}")
print(f" - 状态: {domain.status}")
@@ -118,6 +111,7 @@ def test_domain_management(tenant_id: str):
return domain.id
def test_branding_management(tenant_id: str):
"""测试品牌白标功能"""
print("\n" + "=" * 60)
@@ -136,7 +130,7 @@ def test_branding_management(tenant_id: str):
secondary_color="#52c41a",
custom_css=".header { background: #1890ff; }",
custom_js="console.log('Custom JS loaded');",
login_page_bg="https://example.com/bg.jpg"
login_page_bg="https://example.com/bg.jpg",
)
print("✅ 品牌配置更新成功")
print(f" - Logo: {branding.logo_url}")
@@ -157,6 +151,7 @@ def test_branding_management(tenant_id: str):
return branding.id
def test_member_management(tenant_id: str):
"""测试成员管理功能"""
print("\n" + "=" * 60)
@@ -168,10 +163,7 @@ def test_member_management(tenant_id: str):
# 1. 邀请成员
print("\n4.1 邀请成员...")
member1 = manager.invite_member(
tenant_id=tenant_id,
email="admin@test.com",
role="admin",
invited_by="user_001"
tenant_id=tenant_id, email="admin@test.com", role="admin", invited_by="user_001"
)
print(f"✅ 成员邀请成功: {member1.email}")
print(f" - ID: {member1.id}")
@@ -179,10 +171,7 @@ def test_member_management(tenant_id: str):
print(f" - 权限: {member1.permissions}")
member2 = manager.invite_member(
tenant_id=tenant_id,
email="member@test.com",
role="member",
invited_by="user_001"
tenant_id=tenant_id, email="member@test.com", role="member", invited_by="user_001"
)
print(f"✅ 成员邀请成功: {member2.email}")
@@ -217,6 +206,7 @@ def test_member_management(tenant_id: str):
return member1.id, member2.id
def test_usage_tracking(tenant_id: str):
"""测试资源使用统计功能"""
print("\n" + "=" * 60)
@@ -230,11 +220,11 @@ def test_usage_tracking(tenant_id: str):
manager.record_usage(
tenant_id=tenant_id,
storage_bytes=1024 * 1024 * 50, # 50MB
transcription_seconds=600, # 10分钟
transcription_seconds=600, # 10分钟
api_calls=100,
projects_count=5,
entities_count=50,
members_count=3
members_count=3,
)
print("✅ 资源使用记录成功")
@@ -258,6 +248,7 @@ def test_usage_tracking(tenant_id: str):
return stats
def cleanup(tenant_id: str, domain_id: str, member_ids: list):
"""清理测试数据"""
print("\n" + "=" * 60)
@@ -281,6 +272,7 @@ def cleanup(tenant_id: str, domain_id: str, member_ids: list):
manager.delete_tenant(tenant_id)
print(f"✅ 租户已删除: {tenant_id}")
def main():
"""主测试函数"""
print("\n" + "=" * 60)
@@ -307,6 +299,7 @@ def main():
except Exception as e:
print(f"\n❌ 测试失败: {e}")
import traceback
traceback.print_exc()
finally:
@@ -317,5 +310,6 @@ def main():
except Exception as e:
print(f"⚠️ 清理失败: {e}")
if __name__ == "__main__":
main()