fix: auto-fix code issues (cron)

- 修复重复导入/字段
- 修复异常处理
- 修复PEP8格式问题
- 添加类型注解
This commit is contained in:
OpenClaw Bot
2026-03-01 00:08:06 +08:00
parent 8492e7a0d3
commit ea58b6fe43
75 changed files with 1179 additions and 1207 deletions

File diff suppressed because it is too large Load Diff

View File

@@ -27,7 +27,6 @@ import httpx
# Database path
DB_PATH = os.path.join(os.path.dirname(__file__), "insightflow.db")
class ModelType(StrEnum):
"""模型类型"""
@@ -36,7 +35,6 @@ class ModelType(StrEnum):
SUMMARIZATION = "summarization" # 摘要
PREDICTION = "prediction" # 预测
class ModelStatus(StrEnum):
"""模型状态"""
@@ -46,7 +44,6 @@ class ModelStatus(StrEnum):
FAILED = "failed"
ARCHIVED = "archived"
class MultimodalProvider(StrEnum):
"""多模态模型提供商"""
@@ -55,7 +52,6 @@ class MultimodalProvider(StrEnum):
GEMINI = "gemini-pro-vision"
KIMI_VL = "kimi-vl"
class PredictionType(StrEnum):
"""预测类型"""
@@ -64,7 +60,6 @@ class PredictionType(StrEnum):
ENTITY_GROWTH = "entity_growth" # 实体增长预测
RELATION_EVOLUTION = "relation_evolution" # 关系演变预测
@dataclass
class CustomModel:
"""自定义模型"""
@@ -84,7 +79,6 @@ class CustomModel:
trained_at: str | None
created_by: str
@dataclass
class TrainingSample:
"""训练样本"""
@@ -96,7 +90,6 @@ class TrainingSample:
metadata: dict
created_at: str
@dataclass
class MultimodalAnalysis:
"""多模态分析结果"""
@@ -113,7 +106,6 @@ class MultimodalAnalysis:
cost: float
created_at: str
@dataclass
class KnowledgeGraphRAG:
"""基于知识图谱的 RAG 配置"""
@@ -130,7 +122,6 @@ class KnowledgeGraphRAG:
created_at: str
updated_at: str
@dataclass
class RAGQuery:
"""RAG 查询记录"""
@@ -146,7 +137,6 @@ class RAGQuery:
latency_ms: int
created_at: str
@dataclass
class PredictionModel:
"""预测模型"""
@@ -166,7 +156,6 @@ class PredictionModel:
created_at: str
updated_at: str
@dataclass
class PredictionResult:
"""预测结果"""
@@ -182,7 +171,6 @@ class PredictionResult:
is_correct: bool | None
created_at: str
@dataclass
class SmartSummary:
"""智能摘要"""
@@ -200,7 +188,6 @@ class SmartSummary:
tokens_used: int
created_at: str
class AIManager:
"""AI 能力管理主类"""
@@ -1500,11 +1487,9 @@ class AIManager:
created_at=row["created_at"],
)
# Singleton instance
_ai_manager = None
def get_ai_manager() -> AIManager:
global _ai_manager
if _ai_manager is None:

View File

@@ -15,13 +15,11 @@ from enum import Enum
DB_PATH = os.getenv("DB_PATH", "/app/data/insightflow.db")
class ApiKeyStatus(Enum):
ACTIVE = "active"
REVOKED = "revoked"
EXPIRED = "expired"
@dataclass
class ApiKey:
id: str
@@ -39,7 +37,6 @@ class ApiKey:
revoked_reason: str | None
total_calls: int = 0
class ApiKeyManager:
"""API Key 管理器"""
@@ -525,11 +522,9 @@ class ApiKeyManager:
total_calls=row["total_calls"],
)
# 全局实例
_api_key_manager: ApiKeyManager | None = None
def get_api_key_manager() -> ApiKeyManager:
"""获取 API Key 管理器实例"""
global _api_key_manager

View File

@@ -11,7 +11,6 @@ from datetime import datetime, timedelta
from enum import Enum
from typing import Any
class SharePermission(Enum):
"""分享权限级别"""
@@ -20,7 +19,6 @@ class SharePermission(Enum):
EDIT = "edit" # 可编辑
ADMIN = "admin" # 管理员
class CommentTargetType(Enum):
"""评论目标类型"""
@@ -29,7 +27,6 @@ class CommentTargetType(Enum):
TRANSCRIPT = "transcript" # 转录文本评论
PROJECT = "project" # 项目级评论
class ChangeType(Enum):
"""变更类型"""
@@ -39,7 +36,6 @@ class ChangeType(Enum):
MERGE = "merge" # 合并
SPLIT = "split" # 拆分
@dataclass
class ProjectShare:
"""项目分享链接"""
@@ -58,7 +54,6 @@ class ProjectShare:
allow_download: bool # 允许下载
allow_export: bool # 允许导出
@dataclass
class Comment:
"""评论/批注"""
@@ -79,7 +74,6 @@ class Comment:
mentions: list[str] # 提及的用户
attachments: list[dict] # 附件
@dataclass
class ChangeRecord:
"""变更记录"""
@@ -101,7 +95,6 @@ class ChangeRecord:
reverted_at: str | None # 回滚时间
reverted_by: str | None # 回滚者
@dataclass
class TeamMember:
"""团队成员"""
@@ -117,7 +110,6 @@ class TeamMember:
last_active_at: str | None # 最后活跃时间
permissions: list[str] # 具体权限列表
@dataclass
class TeamSpace:
"""团队空间"""
@@ -132,7 +124,6 @@ class TeamSpace:
project_count: int
settings: dict[str, Any] # 团队设置
class CollaborationManager:
"""协作管理主类"""
@@ -991,11 +982,9 @@ class CollaborationManager:
)
self.db.conn.commit()
# 全局协作管理器实例
_collaboration_manager = None
def get_collaboration_manager(db_manager=None) -> None:
"""获取协作管理器单例"""
global _collaboration_manager

View File

@@ -14,7 +14,6 @@ from datetime import datetime
DB_PATH = os.getenv("DB_PATH", "/app/data/insightflow.db")
@dataclass
class Project:
id: str
@@ -23,7 +22,6 @@ class Project:
created_at: str = ""
updated_at: str = ""
@dataclass
class Entity:
id: str
@@ -44,7 +42,6 @@ class Entity:
if self.attributes is None:
self.attributes = {}
@dataclass
class AttributeTemplate:
"""属性模板定义"""
@@ -65,7 +62,6 @@ class AttributeTemplate:
if self.options is None:
self.options = []
@dataclass
class EntityAttribute:
"""实体属性值"""
@@ -86,7 +82,6 @@ class EntityAttribute:
if self.options is None:
self.options = []
@dataclass
class AttributeHistory:
"""属性变更历史"""
@@ -100,7 +95,6 @@ class AttributeHistory:
changed_at: str = ""
change_reason: str = ""
@dataclass
class EntityMention:
id: str
@@ -111,7 +105,6 @@ class EntityMention:
text_snippet: str
confidence: float = 1.0
class DatabaseManager:
def __init__(self, db_path: str = DB_PATH):
self.db_path = db_path
@@ -1394,11 +1387,9 @@ class DatabaseManager:
conn.close()
return stats
# Singleton instance
_db_manager = None
def get_db_manager() -> DatabaseManager:
global _db_manager
if _db_manager is None:

View File

@@ -21,7 +21,6 @@ from enum import StrEnum
# Database path
DB_PATH = os.path.join(os.path.dirname(__file__), "insightflow.db")
class SDKLanguage(StrEnum):
"""SDK 语言类型"""
@@ -32,7 +31,6 @@ class SDKLanguage(StrEnum):
JAVA = "java"
RUST = "rust"
class SDKStatus(StrEnum):
"""SDK 状态"""
@@ -42,7 +40,6 @@ class SDKStatus(StrEnum):
DEPRECATED = "deprecated" # 已弃用
ARCHIVED = "archived" # 已归档
class TemplateCategory(StrEnum):
"""模板分类"""
@@ -53,7 +50,6 @@ class TemplateCategory(StrEnum):
TECH = "tech" # 科技
GENERAL = "general" # 通用
class TemplateStatus(StrEnum):
"""模板状态"""
@@ -63,7 +59,6 @@ class TemplateStatus(StrEnum):
PUBLISHED = "published" # 已发布
UNLISTED = "unlisted" # 未列出
class PluginStatus(StrEnum):
"""插件状态"""
@@ -74,7 +69,6 @@ class PluginStatus(StrEnum):
PUBLISHED = "published" # 已发布
SUSPENDED = "suspended" # 已暂停
class PluginCategory(StrEnum):
"""插件分类"""
@@ -85,7 +79,6 @@ class PluginCategory(StrEnum):
SECURITY = "security" # 安全
CUSTOM = "custom" # 自定义
class DeveloperStatus(StrEnum):
"""开发者认证状态"""
@@ -95,7 +88,6 @@ class DeveloperStatus(StrEnum):
CERTIFIED = "certified" # 已认证(高级)
SUSPENDED = "suspended" # 已暂停
@dataclass
class SDKRelease:
"""SDK 发布"""
@@ -121,7 +113,6 @@ class SDKRelease:
published_at: str | None
created_by: str
@dataclass
class SDKVersion:
"""SDK 版本历史"""
@@ -138,7 +129,6 @@ class SDKVersion:
download_count: int
created_at: str
@dataclass
class TemplateMarketItem:
"""模板市场项目"""
@@ -170,7 +160,6 @@ class TemplateMarketItem:
updated_at: str
published_at: str | None
@dataclass
class TemplateReview:
"""模板评价"""
@@ -186,7 +175,6 @@ class TemplateReview:
created_at: str
updated_at: str
@dataclass
class PluginMarketItem:
"""插件市场项目"""
@@ -225,7 +213,6 @@ class PluginMarketItem:
reviewed_at: str | None
review_notes: str | None
@dataclass
class PluginReview:
"""插件评价"""
@@ -241,7 +228,6 @@ class PluginReview:
created_at: str
updated_at: str
@dataclass
class DeveloperProfile:
"""开发者档案"""
@@ -265,7 +251,6 @@ class DeveloperProfile:
updated_at: str
verified_at: str | None
@dataclass
class DeveloperRevenue:
"""开发者收益"""
@@ -283,7 +268,6 @@ class DeveloperRevenue:
transaction_id: str
created_at: str
@dataclass
class CodeExample:
"""代码示例"""
@@ -306,7 +290,6 @@ class CodeExample:
created_at: str
updated_at: str
@dataclass
class APIDocumentation:
"""API 文档生成记录"""
@@ -320,7 +303,6 @@ class APIDocumentation:
generated_at: str
generated_by: str
@dataclass
class DeveloperPortalConfig:
"""开发者门户配置"""
@@ -344,7 +326,6 @@ class DeveloperPortalConfig:
created_at: str
updated_at: str
class DeveloperEcosystemManager:
"""开发者生态系统管理主类"""
@@ -2052,11 +2033,9 @@ class DeveloperEcosystemManager:
updated_at=row["updated_at"],
)
# Singleton instance
_developer_ecosystem_manager = None
def get_developer_ecosystem_manager() -> DeveloperEcosystemManager:
"""获取开发者生态系统管理器单例"""
global _developer_ecosystem_manager

View File

@@ -7,7 +7,6 @@ Document Processor - Phase 3
import io
import os
class DocumentProcessor:
"""文档处理器 - 提取 PDF/DOCX 文本"""
@@ -156,7 +155,6 @@ class DocumentProcessor:
ext = os.path.splitext(filename.lower())[1]
return ext in self.supported_formats
# 简单的文本提取器(不需要外部依赖)
class SimpleTextExtractor:
"""简单的文本提取器,用于测试"""
@@ -173,7 +171,6 @@ class SimpleTextExtractor:
return content.decode("latin-1", errors="ignore")
if __name__ == "__main__":
# 测试
processor = DocumentProcessor()

View File

@@ -21,7 +21,6 @@ from typing import Any
logger = logging.getLogger(__name__)
class SSOProvider(StrEnum):
"""SSO 提供商类型"""
@@ -33,7 +32,6 @@ class SSOProvider(StrEnum):
GOOGLE = "google" # Google Workspace
CUSTOM_SAML = "custom_saml" # 自定义 SAML
class SSOStatus(StrEnum):
"""SSO 配置状态"""
@@ -42,7 +40,6 @@ class SSOStatus(StrEnum):
ACTIVE = "active" # 已启用
ERROR = "error" # 配置错误
class SCIMSyncStatus(StrEnum):
"""SCIM 同步状态"""
@@ -51,7 +48,6 @@ class SCIMSyncStatus(StrEnum):
SUCCESS = "success" # 同步成功
FAILED = "failed" # 同步失败
class AuditLogExportFormat(StrEnum):
"""审计日志导出格式"""
@@ -60,7 +56,6 @@ class AuditLogExportFormat(StrEnum):
PDF = "pdf"
XLSX = "xlsx"
class DataRetentionAction(StrEnum):
"""数据保留策略动作"""
@@ -68,7 +63,6 @@ class DataRetentionAction(StrEnum):
DELETE = "delete" # 删除
ANONYMIZE = "anonymize" # 匿名化
class ComplianceStandard(StrEnum):
"""合规标准"""
@@ -78,7 +72,6 @@ class ComplianceStandard(StrEnum):
HIPAA = "hipaa"
PCI_DSS = "pci_dss"
@dataclass
class SSOConfig:
"""SSO 配置数据类"""
@@ -111,7 +104,6 @@ class SSOConfig:
last_tested_at: datetime | None
last_error: str | None
@dataclass
class SCIMConfig:
"""SCIM 配置数据类"""
@@ -136,7 +128,6 @@ class SCIMConfig:
created_at: datetime
updated_at: datetime
@dataclass
class SCIMUser:
"""SCIM 用户数据类"""
@@ -156,7 +147,6 @@ class SCIMUser:
created_at: datetime
updated_at: datetime
@dataclass
class AuditLogExport:
"""审计日志导出记录"""
@@ -181,7 +171,6 @@ class AuditLogExport:
completed_at: datetime | None
error_message: str | None
@dataclass
class DataRetentionPolicy:
"""数据保留策略"""
@@ -209,7 +198,6 @@ class DataRetentionPolicy:
created_at: datetime
updated_at: datetime
@dataclass
class DataRetentionJob:
"""数据保留任务"""
@@ -227,7 +215,6 @@ class DataRetentionJob:
details: dict[str, Any]
created_at: datetime
@dataclass
class SAMLAuthRequest:
"""SAML 认证请求"""
@@ -242,7 +229,6 @@ class SAMLAuthRequest:
used: bool
used_at: datetime | None
@dataclass
class SAMLAuthResponse:
"""SAML 认证响应"""
@@ -259,7 +245,6 @@ class SAMLAuthResponse:
processed_at: datetime | None
created_at: datetime
class EnterpriseManager:
"""企业级功能管理器"""
@@ -2200,11 +2185,9 @@ class EnterpriseManager:
),
)
# 全局实例
_enterprise_manager = None
def get_enterprise_manager(db_path: str = "insightflow.db") -> EnterpriseManager:
"""获取 EnterpriseManager 单例"""
global _enterprise_manager

View File

@@ -15,7 +15,6 @@ import numpy as np
KIMI_API_KEY = os.getenv("KIMI_API_KEY", "")
KIMI_BASE_URL = os.getenv("KIMI_BASE_URL", "https://api.kimi.com/coding")
@dataclass
class EntityEmbedding:
entity_id: str
@@ -23,7 +22,6 @@ class EntityEmbedding:
definition: str
embedding: list[float]
class EntityAligner:
"""实体对齐器 - 使用 embedding 进行相似度匹配"""
@@ -318,7 +316,6 @@ class EntityAligner:
return []
# 简单的字符串相似度计算(不使用 embedding
def simple_similarity(str1: str, str2: str) -> float:
"""
@@ -350,7 +347,6 @@ def simple_similarity(str1: str, str2: str) -> float:
return SequenceMatcher(None, s1, s2).ratio()
if __name__ == "__main__":
# 测试
aligner = EntityAligner()

View File

@@ -36,7 +36,6 @@ try:
except ImportError:
REPORTLAB_AVAILABLE = False
@dataclass
class ExportEntity:
id: str
@@ -47,7 +46,6 @@ class ExportEntity:
mention_count: int
attributes: dict[str, Any]
@dataclass
class ExportRelation:
id: str
@@ -57,7 +55,6 @@ class ExportRelation:
confidence: float
evidence: str
@dataclass
class ExportTranscript:
id: str
@@ -67,7 +64,6 @@ class ExportTranscript:
segments: list[dict]
entity_mentions: list[dict]
class ExportManager:
"""导出管理器 - 处理各种导出需求"""
@@ -615,11 +611,9 @@ class ExportManager:
return json.dumps(data, ensure_ascii=False, indent=2)
# 全局导出管理器实例
_export_manager = None
def get_export_manager(db_manager=None) -> None:
"""获取导出管理器实例"""
global _export_manager

View File

@@ -28,7 +28,6 @@ import httpx
# Database path
DB_PATH = os.path.join(os.path.dirname(__file__), "insightflow.db")
class EventType(StrEnum):
"""事件类型"""
@@ -44,7 +43,6 @@ class EventType(StrEnum):
INVITE_ACCEPTED = "invite_accepted" # 接受邀请
REFERRAL_REWARD = "referral_reward" # 推荐奖励
class ExperimentStatus(StrEnum):
"""实验状态"""
@@ -54,7 +52,6 @@ class ExperimentStatus(StrEnum):
COMPLETED = "completed" # 已完成
ARCHIVED = "archived" # 已归档
class TrafficAllocationType(StrEnum):
"""流量分配类型"""
@@ -62,7 +59,6 @@ class TrafficAllocationType(StrEnum):
STRATIFIED = "stratified" # 分层分配
TARGETED = "targeted" # 定向分配
class EmailTemplateType(StrEnum):
"""邮件模板类型"""
@@ -74,7 +70,6 @@ class EmailTemplateType(StrEnum):
REFERRAL = "referral" # 推荐邀请
NEWSLETTER = "newsletter" # 新闻通讯
class EmailStatus(StrEnum):
"""邮件状态"""
@@ -88,7 +83,6 @@ class EmailStatus(StrEnum):
BOUNCED = "bounced" # 退信
FAILED = "failed" # 失败
class WorkflowTriggerType(StrEnum):
"""工作流触发类型"""
@@ -100,7 +94,6 @@ class WorkflowTriggerType(StrEnum):
MILESTONE = "milestone" # 里程碑
CUSTOM_EVENT = "custom_event" # 自定义事件
class ReferralStatus(StrEnum):
"""推荐状态"""
@@ -109,7 +102,6 @@ class ReferralStatus(StrEnum):
REWARDED = "rewarded" # 已奖励
EXPIRED = "expired" # 已过期
@dataclass
class AnalyticsEvent:
"""分析事件"""
@@ -128,7 +120,6 @@ class AnalyticsEvent:
utm_medium: str | None
utm_campaign: str | None
@dataclass
class UserProfile:
"""用户画像"""
@@ -148,7 +139,6 @@ class UserProfile:
created_at: datetime
updated_at: datetime
@dataclass
class Funnel:
"""转化漏斗"""
@@ -161,7 +151,6 @@ class Funnel:
created_at: datetime
updated_at: datetime
@dataclass
class FunnelAnalysis:
"""漏斗分析结果"""
@@ -174,7 +163,6 @@ class FunnelAnalysis:
overall_conversion: float # 总体转化率
drop_off_points: list[dict] # 流失点
@dataclass
class Experiment:
"""A/B 测试实验"""
@@ -199,7 +187,6 @@ class Experiment:
updated_at: datetime
created_by: str
@dataclass
class ExperimentResult:
"""实验结果"""
@@ -217,7 +204,6 @@ class ExperimentResult:
uplift: float # 提升幅度
created_at: datetime
@dataclass
class EmailTemplate:
"""邮件模板"""
@@ -238,7 +224,6 @@ class EmailTemplate:
created_at: datetime
updated_at: datetime
@dataclass
class EmailCampaign:
"""邮件营销活动"""
@@ -260,7 +245,6 @@ class EmailCampaign:
completed_at: datetime | None
created_at: datetime
@dataclass
class EmailLog:
"""邮件发送记录"""
@@ -282,7 +266,6 @@ class EmailLog:
error_message: str | None
created_at: datetime
@dataclass
class AutomationWorkflow:
"""自动化工作流"""
@@ -299,7 +282,6 @@ class AutomationWorkflow:
created_at: datetime
updated_at: datetime
@dataclass
class ReferralProgram:
"""推荐计划"""
@@ -319,7 +301,6 @@ class ReferralProgram:
created_at: datetime
updated_at: datetime
@dataclass
class Referral:
"""推荐记录"""
@@ -340,7 +321,6 @@ class Referral:
expires_at: datetime
created_at: datetime
@dataclass
class TeamIncentive:
"""团队升级激励"""
@@ -358,7 +338,6 @@ class TeamIncentive:
is_active: bool
created_at: datetime
class GrowthManager:
"""运营与增长管理主类"""
@@ -2147,11 +2126,9 @@ class GrowthManager:
created_at=row["created_at"],
)
# Singleton instance
_growth_manager = None
def get_growth_manager() -> GrowthManager:
global _growth_manager
if _growth_manager is None:

View File

@@ -33,7 +33,6 @@ try:
except ImportError:
PYTESSERACT_AVAILABLE = False
@dataclass
class ImageEntity:
"""图片中检测到的实体"""
@@ -43,7 +42,6 @@ class ImageEntity:
confidence: float
bbox: tuple[int, int, int, int] | None = None # (x, y, width, height)
@dataclass
class ImageRelation:
"""图片中检测到的关系"""
@@ -53,7 +51,6 @@ class ImageRelation:
relation_type: str
confidence: float
@dataclass
class ImageProcessingResult:
"""图片处理结果"""
@@ -69,7 +66,6 @@ class ImageProcessingResult:
success: bool
error_message: str = ""
@dataclass
class BatchProcessingResult:
"""批量图片处理结果"""
@@ -79,7 +75,6 @@ class BatchProcessingResult:
success_count: int
failed_count: int
class ImageProcessor:
"""图片处理器 - 处理各种类型图片"""
@@ -553,11 +548,9 @@ class ImageProcessor:
print(f"Thumbnail generation error: {e}")
return image_data
# Singleton instance
_image_processor = None
def get_image_processor(temp_dir: str = None) -> ImageProcessor:
"""获取图片处理器单例"""
global _image_processor

View File

@@ -15,7 +15,6 @@ import httpx
KIMI_API_KEY = os.getenv("KIMI_API_KEY", "")
KIMI_BASE_URL = os.getenv("KIMI_BASE_URL", "https://api.kimi.com/coding")
class ReasoningType(Enum):
"""推理类型"""
@@ -25,7 +24,6 @@ class ReasoningType(Enum):
COMPARATIVE = "comparative" # 对比推理
SUMMARY = "summary" # 总结推理
@dataclass
class ReasoningResult:
"""推理结果"""
@@ -37,7 +35,6 @@ class ReasoningResult:
related_entities: list[str] # 相关实体
gaps: list[str] # 知识缺口
@dataclass
class InferencePath:
"""推理路径"""
@@ -47,7 +44,6 @@ class InferencePath:
path: list[dict] # 路径上的节点和关系
strength: float # 路径强度
class KnowledgeReasoner:
"""知识推理引擎"""
@@ -502,11 +498,9 @@ class KnowledgeReasoner:
"confidence": 0.5,
}
# Singleton instance
_reasoner = None
def get_knowledge_reasoner() -> KnowledgeReasoner:
global _reasoner
if _reasoner is None:

View File

@@ -15,13 +15,11 @@ import httpx
KIMI_API_KEY = os.getenv("KIMI_API_KEY", "")
KIMI_BASE_URL = os.getenv("KIMI_BASE_URL", "https://api.kimi.com/coding")
@dataclass
class ChatMessage:
role: str
content: str
@dataclass
class EntityExtractionResult:
name: str
@@ -29,7 +27,6 @@ class EntityExtractionResult:
definition: str
confidence: float
@dataclass
class RelationExtractionResult:
source: str
@@ -37,7 +34,6 @@ class RelationExtractionResult:
type: str
confidence: float
class LLMClient:
"""Kimi API 客户端"""
@@ -258,11 +254,9 @@ class LLMClient:
messages = [ChatMessage(role="user", content=prompt)]
return await self.chat(messages, temperature=0.3)
# Singleton instance
_llm_client = None
def get_llm_client() -> LLMClient:
global _llm_client
if _llm_client is None:

View File

@@ -35,7 +35,6 @@ except ImportError:
logger = logging.getLogger(__name__)
class LanguageCode(StrEnum):
"""支持的语言代码"""
@@ -52,7 +51,6 @@ class LanguageCode(StrEnum):
AR = "ar"
HI = "hi"
class RegionCode(StrEnum):
"""区域代码"""
@@ -64,7 +62,6 @@ class RegionCode(StrEnum):
LATIN_AMERICA = "latam"
MIDDLE_EAST = "me"
class DataCenterRegion(StrEnum):
"""数据中心区域"""
@@ -78,7 +75,6 @@ class DataCenterRegion(StrEnum):
CN_NORTH = "cn-north"
CN_EAST = "cn-east"
class PaymentProvider(StrEnum):
"""支付提供商"""
@@ -95,7 +91,6 @@ class PaymentProvider(StrEnum):
SEPA = "sepa"
UNIONPAY = "unionpay"
class CalendarType(StrEnum):
"""日历类型"""
@@ -107,7 +102,6 @@ class CalendarType(StrEnum):
PERSIAN = "persian"
BUDDHIST = "buddhist"
@dataclass
class Translation:
id: str
@@ -122,7 +116,6 @@ class Translation:
reviewed_by: str | None
reviewed_at: datetime | None
@dataclass
class LanguageConfig:
code: str
@@ -140,7 +133,6 @@ class LanguageConfig:
first_day_of_week: int
calendar_type: str
@dataclass
class DataCenter:
id: str
@@ -155,7 +147,6 @@ class DataCenter:
created_at: datetime
updated_at: datetime
@dataclass
class TenantDataCenterMapping:
id: str
@@ -167,7 +158,6 @@ class TenantDataCenterMapping:
created_at: datetime
updated_at: datetime
@dataclass
class LocalizedPaymentMethod:
id: str
@@ -185,7 +175,6 @@ class LocalizedPaymentMethod:
created_at: datetime
updated_at: datetime
@dataclass
class CountryConfig:
code: str
@@ -207,7 +196,6 @@ class CountryConfig:
vat_rate: float | None
is_active: bool
@dataclass
class TimezoneConfig:
id: str
@@ -218,7 +206,6 @@ class TimezoneConfig:
region: str
is_active: bool
@dataclass
class CurrencyConfig:
code: str
@@ -230,7 +217,6 @@ class CurrencyConfig:
thousands_separator: str
is_active: bool
@dataclass
class LocalizationSettings:
id: str
@@ -250,7 +236,6 @@ class LocalizationSettings:
created_at: datetime
updated_at: datetime
class LocalizationManager:
DEFAULT_LANGUAGES = {
LanguageCode.EN: {
@@ -1698,10 +1683,8 @@ class LocalizationManager:
),
)
_localization_manager = None
def get_localization_manager(db_path: str = "insightflow.db") -> LocalizationManager:
global _localization_manager
if _localization_manager is None:

File diff suppressed because it is too large Load Diff

View File

@@ -14,7 +14,6 @@ try:
except ImportError:
NUMPY_AVAILABLE = False
@dataclass
class MultimodalEntity:
"""多模态实体"""
@@ -33,7 +32,6 @@ class MultimodalEntity:
if self.modality_features is None:
self.modality_features = {}
@dataclass
class EntityLink:
"""实体关联"""
@@ -48,7 +46,6 @@ class EntityLink:
confidence: float
evidence: str
@dataclass
class AlignmentResult:
"""对齐结果"""
@@ -59,7 +56,6 @@ class AlignmentResult:
match_type: str # exact, fuzzy, embedding
confidence: float
@dataclass
class FusionResult:
"""知识融合结果"""
@@ -70,7 +66,6 @@ class FusionResult:
source_modalities: list[str]
confidence: float
class MultimodalEntityLinker:
"""多模态实体关联器 - 跨模态实体对齐和知识融合"""
@@ -512,11 +507,9 @@ class MultimodalEntityLinker:
else 0,
}
# Singleton instance
_multimodal_entity_linker = None
def get_multimodal_entity_linker(similarity_threshold: float = 0.85) -> MultimodalEntityLinker:
"""获取多模态实体关联器单例"""
global _multimodal_entity_linker

View File

@@ -35,7 +35,6 @@ try:
except ImportError:
FFMPEG_AVAILABLE = False
@dataclass
class VideoFrame:
"""视频关键帧数据类"""
@@ -53,7 +52,6 @@ class VideoFrame:
if self.entities_detected is None:
self.entities_detected = []
@dataclass
class VideoInfo:
"""视频信息数据类"""
@@ -77,7 +75,6 @@ class VideoInfo:
if self.metadata is None:
self.metadata = {}
@dataclass
class VideoProcessingResult:
"""视频处理结果"""
@@ -90,7 +87,6 @@ class VideoProcessingResult:
success: bool
error_message: str = ""
class MultimodalProcessor:
"""多模态处理器 - 处理视频文件"""
@@ -449,11 +445,9 @@ class MultimodalProcessor:
shutil.rmtree(dir_path)
os.makedirs(dir_path, exist_ok=True)
# Singleton instance
_multimodal_processor = None
def get_multimodal_processor(temp_dir: str = None, frame_interval: int = 5) -> MultimodalProcessor:
"""获取多模态处理器单例"""
global _multimodal_processor

View File

@@ -26,7 +26,6 @@ except ImportError:
NEO4J_AVAILABLE = False
logger.warning("Neo4j driver not installed. Neo4j features will be disabled.")
@dataclass
class GraphEntity:
"""图数据库中的实体节点"""
@@ -45,7 +44,6 @@ class GraphEntity:
if self.properties is None:
self.properties = {}
@dataclass
class GraphRelation:
"""图数据库中的关系边"""
@@ -61,7 +59,6 @@ class GraphRelation:
if self.properties is None:
self.properties = {}
@dataclass
class PathResult:
"""路径查询结果"""
@@ -71,7 +68,6 @@ class PathResult:
length: int
total_weight: float = 0.0
@dataclass
class CommunityResult:
"""社区发现结果"""
@@ -81,7 +77,6 @@ class CommunityResult:
size: int
density: float = 0.0
@dataclass
class CentralityResult:
"""中心性分析结果"""
@@ -91,7 +86,6 @@ class CentralityResult:
score: float
rank: int = 0
class Neo4jManager:
"""Neo4j 图数据库管理器"""
@@ -968,11 +962,9 @@ class Neo4jManager:
return {"nodes": nodes, "relationships": relationships}
# 全局单例
_neo4j_manager = None
def get_neo4j_manager() -> Neo4jManager:
"""获取 Neo4j 管理器单例"""
global _neo4j_manager
@@ -980,7 +972,6 @@ def get_neo4j_manager() -> Neo4jManager:
_neo4j_manager = Neo4jManager()
return _neo4j_manager
def close_neo4j_manager() -> None:
"""关闭 Neo4j 连接"""
global _neo4j_manager
@@ -988,7 +979,6 @@ def close_neo4j_manager() -> None:
_neo4j_manager.close()
_neo4j_manager = None
# 便捷函数
def sync_project_to_neo4j(
project_id: str, project_name: str, entities: list[dict], relations: list[dict]
@@ -1043,7 +1033,6 @@ def sync_project_to_neo4j(
f"Synced project {project_id} to Neo4j: {len(entities)} entities, {len(relations)} relations"
)
if __name__ == "__main__":
# 测试代码
logging.basicConfig(level=logging.INFO)

View File

@@ -29,7 +29,6 @@ import httpx
# Database path
DB_PATH = os.path.join(os.path.dirname(__file__), "insightflow.db")
class AlertSeverity(StrEnum):
"""告警严重级别 P0-P3"""
@@ -38,7 +37,6 @@ class AlertSeverity(StrEnum):
P2 = "p2" # 一般 - 部分功能受影响需要4小时内处理
P3 = "p3" # 轻微 - 非核心功能问题24小时内处理
class AlertStatus(StrEnum):
"""告警状态"""
@@ -47,7 +45,6 @@ class AlertStatus(StrEnum):
ACKNOWLEDGED = "acknowledged" # 已确认
SUPPRESSED = "suppressed" # 已抑制
class AlertChannelType(StrEnum):
"""告警渠道类型"""
@@ -60,7 +57,6 @@ class AlertChannelType(StrEnum):
SMS = "sms"
WEBHOOK = "webhook"
class AlertRuleType(StrEnum):
"""告警规则类型"""
@@ -69,7 +65,6 @@ class AlertRuleType(StrEnum):
PREDICTIVE = "predictive" # 预测性告警
COMPOSITE = "composite" # 复合告警
class ResourceType(StrEnum):
"""资源类型"""
@@ -82,7 +77,6 @@ class ResourceType(StrEnum):
CACHE = "cache"
QUEUE = "queue"
class ScalingAction(StrEnum):
"""扩缩容动作"""
@@ -90,7 +84,6 @@ class ScalingAction(StrEnum):
SCALE_DOWN = "scale_down" # 缩容
MAINTAIN = "maintain" # 保持
class HealthStatus(StrEnum):
"""健康状态"""
@@ -99,7 +92,6 @@ class HealthStatus(StrEnum):
UNHEALTHY = "unhealthy"
UNKNOWN = "unknown"
class BackupStatus(StrEnum):
"""备份状态"""
@@ -109,7 +101,6 @@ class BackupStatus(StrEnum):
FAILED = "failed"
VERIFIED = "verified"
@dataclass
class AlertRule:
"""告警规则"""
@@ -133,7 +124,6 @@ class AlertRule:
updated_at: str
created_by: str
@dataclass
class AlertChannel:
"""告警渠道配置"""
@@ -151,7 +141,6 @@ class AlertChannel:
created_at: str
updated_at: str
@dataclass
class Alert:
"""告警实例"""
@@ -175,7 +164,6 @@ class Alert:
notification_sent: dict[str, bool] # 渠道发送状态
suppression_count: int # 抑制计数
@dataclass
class AlertSuppressionRule:
"""告警抑制规则"""
@@ -189,7 +177,6 @@ class AlertSuppressionRule:
created_at: str
expires_at: str | None
@dataclass
class AlertGroup:
"""告警聚合组"""
@@ -201,7 +188,6 @@ class AlertGroup:
created_at: str
updated_at: str
@dataclass
class ResourceMetric:
"""资源指标"""
@@ -216,7 +202,6 @@ class ResourceMetric:
timestamp: str
metadata: dict
@dataclass
class CapacityPlan:
"""容量规划"""
@@ -232,7 +217,6 @@ class CapacityPlan:
estimated_cost: float
created_at: str
@dataclass
class AutoScalingPolicy:
"""自动扩缩容策略"""
@@ -253,7 +237,6 @@ class AutoScalingPolicy:
created_at: str
updated_at: str
@dataclass
class ScalingEvent:
"""扩缩容事件"""
@@ -271,7 +254,6 @@ class ScalingEvent:
completed_at: str | None
error_message: str | None
@dataclass
class HealthCheck:
"""健康检查配置"""
@@ -292,7 +274,6 @@ class HealthCheck:
created_at: str
updated_at: str
@dataclass
class HealthCheckResult:
"""健康检查结果"""
@@ -306,7 +287,6 @@ class HealthCheckResult:
details: dict
checked_at: str
@dataclass
class FailoverConfig:
"""故障转移配置"""
@@ -324,7 +304,6 @@ class FailoverConfig:
created_at: str
updated_at: str
@dataclass
class FailoverEvent:
"""故障转移事件"""
@@ -340,7 +319,6 @@ class FailoverEvent:
completed_at: str | None
rolled_back_at: str | None
@dataclass
class BackupJob:
"""备份任务"""
@@ -360,7 +338,6 @@ class BackupJob:
created_at: str
updated_at: str
@dataclass
class BackupRecord:
"""备份记录"""
@@ -377,7 +354,6 @@ class BackupRecord:
error_message: str | None
storage_path: str
@dataclass
class CostReport:
"""成本报告"""
@@ -392,7 +368,6 @@ class CostReport:
anomalies: list[dict] # 异常检测
created_at: str
@dataclass
class ResourceUtilization:
"""资源利用率"""
@@ -408,7 +383,6 @@ class ResourceUtilization:
report_date: str
recommendations: list[str]
@dataclass
class IdleResource:
"""闲置资源"""
@@ -425,7 +399,6 @@ class IdleResource:
recommendation: str
detected_at: str
@dataclass
class CostOptimizationSuggestion:
"""成本优化建议"""
@@ -445,7 +418,6 @@ class CostOptimizationSuggestion:
created_at: str
applied_at: str | None
class OpsManager:
"""运维与监控管理主类"""
@@ -3098,11 +3070,9 @@ class OpsManager:
applied_at=row["applied_at"],
)
# Singleton instance
_ops_manager = None
def get_ops_manager() -> OpsManager:
global _ops_manager
if _ops_manager is None:

View File

@@ -9,7 +9,6 @@ from datetime import datetime
import oss2
class OSSUploader:
def __init__(self):
self.access_key = os.getenv("ALI_ACCESS_KEY")
@@ -41,11 +40,9 @@ class OSSUploader:
"""删除 OSS 对象"""
self.bucket.delete_object(object_name)
# 单例
_oss_uploader = None
def get_oss_uploader() -> OSSUploader:
global _oss_uploader
if _oss_uploader is None:

View File

@@ -42,7 +42,6 @@ except ImportError:
# ==================== 数据模型 ====================
@dataclass
class CacheStats:
"""缓存统计数据模型"""
@@ -59,7 +58,6 @@ class CacheStats:
if self.total_requests > 0:
self.hit_rate = round(self.hits / self.total_requests, 4)
@dataclass
class CacheEntry:
"""缓存条目数据模型"""
@@ -72,7 +70,6 @@ class CacheEntry:
last_accessed: float = 0
size_bytes: int = 0
@dataclass
class PerformanceMetric:
"""性能指标数据模型"""
@@ -94,7 +91,6 @@ class PerformanceMetric:
"metadata": self.metadata,
}
@dataclass
class TaskInfo:
"""任务信息数据模型"""
@@ -126,7 +122,6 @@ class TaskInfo:
"max_retries": self.max_retries,
}
@dataclass
class ShardInfo:
"""分片信息数据模型"""
@@ -139,10 +134,8 @@ class ShardInfo:
created_at: str = ""
last_accessed: str = ""
# ==================== Redis 缓存层 ====================
class CacheManager:
"""
缓存管理器
@@ -599,10 +592,8 @@ class CacheManager:
return count
# ==================== 数据库分片 ====================
class DatabaseSharding:
"""
数据库分片管理器
@@ -902,10 +893,8 @@ class DatabaseSharding:
"message": "Rebalancing analysis completed",
}
# ==================== 异步任务队列 ====================
class TaskQueue:
"""
异步任务队列管理器
@@ -1287,10 +1276,8 @@ class TaskQueue:
"backend": "celery" if self.use_celery else "memory",
}
# ==================== 性能监控 ====================
class PerformanceMonitor:
"""
性能监控器
@@ -1607,10 +1594,8 @@ class PerformanceMonitor:
return deleted
# ==================== 性能装饰器 ====================
def cached(
cache_manager: CacheManager,
key_prefix: str = "",
@@ -1655,7 +1640,6 @@ def cached(
return decorator
def monitored(monitor: PerformanceMonitor, metric_type: str, endpoint: str | None = None) -> None:
"""
性能监控装饰器
@@ -1683,10 +1667,8 @@ def monitored(monitor: PerformanceMonitor, metric_type: str, endpoint: str | Non
return decorator
# ==================== 性能管理器 ====================
class PerformanceManager:
"""
性能管理器 - 统一入口
@@ -1748,11 +1730,9 @@ class PerformanceManager:
return stats
# 单例模式
_performance_manager = None
def get_performance_manager(
db_path: str = "insightflow.db", redis_url: str | None = None, enable_sharding: bool = False
) -> PerformanceManager:

View File

@@ -28,7 +28,6 @@ try:
except ImportError:
WEBDAV_AVAILABLE = False
class PluginType(Enum):
"""插件类型"""
@@ -40,7 +39,6 @@ class PluginType(Enum):
WEBDAV = "webdav"
CUSTOM = "custom"
class PluginStatus(Enum):
"""插件状态"""
@@ -49,7 +47,6 @@ class PluginStatus(Enum):
ERROR = "error"
PENDING = "pending"
@dataclass
class Plugin:
"""插件配置"""
@@ -65,7 +62,6 @@ class Plugin:
last_used_at: str | None = None
use_count: int = 0
@dataclass
class PluginConfig:
"""插件详细配置"""
@@ -78,7 +74,6 @@ class PluginConfig:
created_at: str = ""
updated_at: str = ""
@dataclass
class BotSession:
"""机器人会话"""
@@ -96,7 +91,6 @@ class BotSession:
last_message_at: str | None = None
message_count: int = 0
@dataclass
class WebhookEndpoint:
"""Webhook 端点配置Zapier/Make集成"""
@@ -115,7 +109,6 @@ class WebhookEndpoint:
last_triggered_at: str | None = None
trigger_count: int = 0
@dataclass
class WebDAVSync:
"""WebDAV 同步配置"""
@@ -137,7 +130,6 @@ class WebDAVSync:
updated_at: str = ""
sync_count: int = 0
@dataclass
class ChromeExtensionToken:
"""Chrome 扩展令牌"""
@@ -154,7 +146,6 @@ class ChromeExtensionToken:
use_count: int = 0
is_revoked: bool = False
class PluginManager:
"""插件管理主类"""
@@ -394,7 +385,6 @@ class PluginManager:
conn.commit()
conn.close()
class ChromeExtensionHandler:
"""Chrome 扩展处理器"""
@@ -598,7 +588,6 @@ class ChromeExtensionHandler:
"content_length": len(content),
}
class BotHandler:
"""飞书/钉钉机器人处理器"""
@@ -926,7 +915,6 @@ class BotHandler:
)
return response.status_code == 200
class WebhookIntegration:
"""Zapier/Make Webhook 集成"""
@@ -1149,7 +1137,6 @@ class WebhookIntegration:
"message": "Test event sent successfully" if success else "Failed to send test event",
}
class WebDAVSyncManager:
"""WebDAV 同步管理"""
@@ -1410,11 +1397,9 @@ class WebDAVSyncManager:
return {"success": False, "error": str(e)}
# Singleton instance
_plugin_manager = None
def get_plugin_manager(db_manager=None) -> None:
"""获取 PluginManager 单例"""
global _plugin_manager

View File

@@ -12,7 +12,6 @@ from collections.abc import Callable
from dataclasses import dataclass
from functools import wraps
@dataclass
class RateLimitConfig:
"""限流配置"""
@@ -21,7 +20,6 @@ class RateLimitConfig:
burst_size: int = 10 # 突发请求数
window_size: int = 60 # 窗口大小(秒)
@dataclass
class RateLimitInfo:
"""限流信息"""
@@ -31,7 +29,6 @@ class RateLimitInfo:
reset_time: int # 重置时间戳
retry_after: int # 需要等待的秒数
class SlidingWindowCounter:
"""滑动窗口计数器"""
@@ -63,7 +60,6 @@ class SlidingWindowCounter:
for k in old_keys:
self.requests.pop(k, None)
class RateLimiter:
"""API 限流器"""
@@ -159,11 +155,9 @@ class RateLimiter:
self.counters.clear()
self.configs.clear()
# 全局限流器实例
_rate_limiter: RateLimiter | None = None
def get_rate_limiter() -> RateLimiter:
"""获取限流器实例"""
global _rate_limiter
@@ -171,7 +165,6 @@ def get_rate_limiter() -> RateLimiter:
_rate_limiter = RateLimiter()
return _rate_limiter
# 限流装饰器(用于函数级别限流)
def rate_limit(requests_per_minute: int = 60, key_func: Callable | None = None) -> None:
"""
@@ -215,6 +208,5 @@ def rate_limit(requests_per_minute: int = 60, key_func: Callable | None = None)
return decorator
class RateLimitExceeded(Exception):
"""限流异常"""

View File

@@ -19,7 +19,6 @@ from dataclasses import dataclass, field
from datetime import datetime
from enum import Enum
class SearchOperator(Enum):
"""搜索操作符"""
@@ -27,7 +26,6 @@ class SearchOperator(Enum):
OR = "OR"
NOT = "NOT"
# 尝试导入 sentence-transformers 用于语义搜索
try:
from sentence_transformers import SentenceTransformer
@@ -39,7 +37,6 @@ except ImportError:
# ==================== 数据模型 ====================
@dataclass
class SearchResult:
"""搜索结果数据模型"""
@@ -63,7 +60,6 @@ class SearchResult:
"metadata": self.metadata,
}
@dataclass
class SemanticSearchResult:
"""语义搜索结果数据模型"""
@@ -89,7 +85,6 @@ class SemanticSearchResult:
result["embedding_dim"] = len(self.embedding)
return result
@dataclass
class EntityPath:
"""实体关系路径数据模型"""
@@ -119,7 +114,6 @@ class EntityPath:
"path_description": self.path_description,
}
@dataclass
class KnowledgeGap:
"""知识缺口数据模型"""
@@ -147,7 +141,6 @@ class KnowledgeGap:
"metadata": self.metadata,
}
@dataclass
class SearchIndex:
"""搜索索引数据模型"""
@@ -161,7 +154,6 @@ class SearchIndex:
created_at: str
updated_at: str
@dataclass
class TextEmbedding:
"""文本 Embedding 数据模型"""
@@ -174,10 +166,8 @@ class TextEmbedding:
model_name: str
created_at: str
# ==================== 全文搜索 ====================
class FullTextSearch:
"""
全文搜索模块
@@ -786,10 +776,8 @@ class FullTextSearch:
conn.close()
return stats
# ==================== 语义搜索 ====================
class SemanticSearch:
"""
语义搜索模块
@@ -1150,10 +1138,8 @@ class SemanticSearch:
print(f"删除 embedding 失败: {e}")
return False
# ==================== 实体关系路径发现 ====================
class EntityPathDiscovery:
"""
实体关系路径发现模块
@@ -1623,10 +1609,8 @@ class EntityPathDiscovery:
bridge_scores.sort(key=lambda x: x["bridge_score"], reverse=True)
return bridge_scores[:20] # 返回前20
# ==================== 知识缺口识别 ====================
class KnowledgeGapDetection:
"""
知识缺口识别模块
@@ -2029,10 +2013,8 @@ class KnowledgeGapDetection:
return recommendations
# ==================== 搜索管理器 ====================
class SearchManager:
"""
搜索管理器 - 统一入口
@@ -2205,11 +2187,9 @@ class SearchManager:
"semantic_search_available": self.semantic_search.is_available(),
}
# 单例模式
_search_manager = None
def get_search_manager(db_path: str = "insightflow.db") -> SearchManager:
"""获取搜索管理器单例"""
global _search_manager
@@ -2217,7 +2197,6 @@ def get_search_manager(db_path: str = "insightflow.db") -> SearchManager:
_search_manager = SearchManager(db_path)
return _search_manager
# 便捷函数
def fulltext_search(
query: str, project_id: str | None = None, limit: int = 20
@@ -2226,7 +2205,6 @@ def fulltext_search(
manager = get_search_manager()
return manager.fulltext_search.search(query, project_id, limit=limit)
def semantic_search(
query: str, project_id: str | None = None, top_k: int = 10
) -> list[SemanticSearchResult]:
@@ -2234,13 +2212,11 @@ def semantic_search(
manager = get_search_manager()
return manager.semantic_search.search(query, project_id, top_k=top_k)
def find_entity_path(source_id: str, target_id: str, max_depth: int = 5) -> EntityPath | None:
"""查找实体路径便捷函数"""
manager = get_search_manager()
return manager.path_discovery.find_shortest_path(source_id, target_id, max_depth)
def detect_knowledge_gaps(project_id: str) -> list[KnowledgeGap]:
"""知识缺口检测便捷函数"""
manager = get_search_manager()

View File

@@ -25,7 +25,6 @@ except ImportError:
CRYPTO_AVAILABLE = False
print("Warning: cryptography not available, encryption features disabled")
class AuditActionType(Enum):
"""审计动作类型"""
@@ -48,7 +47,6 @@ class AuditActionType(Enum):
WEBHOOK_SEND = "webhook_send"
BOT_MESSAGE = "bot_message"
class DataSensitivityLevel(Enum):
"""数据敏感度级别"""
@@ -57,7 +55,6 @@ class DataSensitivityLevel(Enum):
CONFIDENTIAL = "confidential" # 机密
SECRET = "secret" # 绝密
class MaskingRuleType(Enum):
"""脱敏规则类型"""
@@ -69,7 +66,6 @@ class MaskingRuleType(Enum):
ADDRESS = "address" # 地址
CUSTOM = "custom" # 自定义
@dataclass
class AuditLog:
"""审计日志条目"""
@@ -91,7 +87,6 @@ class AuditLog:
def to_dict(self) -> dict[str, Any]:
return asdict(self)
@dataclass
class EncryptionConfig:
"""加密配置"""
@@ -109,7 +104,6 @@ class EncryptionConfig:
def to_dict(self) -> dict[str, Any]:
return asdict(self)
@dataclass
class MaskingRule:
"""脱敏规则"""
@@ -129,7 +123,6 @@ class MaskingRule:
def to_dict(self) -> dict[str, Any]:
return asdict(self)
@dataclass
class DataAccessPolicy:
"""数据访问策略"""
@@ -151,7 +144,6 @@ class DataAccessPolicy:
def to_dict(self) -> dict[str, Any]:
return asdict(self)
@dataclass
class AccessRequest:
"""访问请求(用于需要审批的访问)"""
@@ -169,7 +161,6 @@ class AccessRequest:
def to_dict(self) -> dict[str, Any]:
return asdict(self)
class SecurityManager:
"""安全管理器"""
@@ -1240,11 +1231,9 @@ class SecurityManager:
created_at=row[8],
)
# 全局安全管理器实例
_security_manager = None
def get_security_manager(db_path: str = "insightflow.db") -> SecurityManager:
"""获取安全管理器实例"""
global _security_manager

View File

@@ -21,7 +21,6 @@ from typing import Any
logger = logging.getLogger(__name__)
class SubscriptionStatus(StrEnum):
"""订阅状态"""
@@ -32,7 +31,6 @@ class SubscriptionStatus(StrEnum):
TRIAL = "trial" # 试用中
PENDING = "pending" # 待支付
class PaymentProvider(StrEnum):
"""支付提供商"""
@@ -41,7 +39,6 @@ class PaymentProvider(StrEnum):
WECHAT = "wechat" # 微信支付
BANK_TRANSFER = "bank_transfer" # 银行转账
class PaymentStatus(StrEnum):
"""支付状态"""
@@ -52,7 +49,6 @@ class PaymentStatus(StrEnum):
REFUNDED = "refunded" # 已退款
PARTIAL_REFUNDED = "partial_refunded" # 部分退款
class InvoiceStatus(StrEnum):
"""发票状态"""
@@ -63,7 +59,6 @@ class InvoiceStatus(StrEnum):
VOID = "void" # 作废
CREDIT_NOTE = "credit_note" # 贷项通知单
class RefundStatus(StrEnum):
"""退款状态"""
@@ -73,7 +68,6 @@ class RefundStatus(StrEnum):
COMPLETED = "completed" # 已完成
FAILED = "failed" # 失败
@dataclass
class SubscriptionPlan:
"""订阅计划数据类"""
@@ -92,7 +86,6 @@ class SubscriptionPlan:
updated_at: datetime
metadata: dict[str, Any]
@dataclass
class Subscription:
"""订阅数据类"""
@@ -113,7 +106,6 @@ class Subscription:
updated_at: datetime
metadata: dict[str, Any]
@dataclass
class UsageRecord:
"""用量记录数据类"""
@@ -128,7 +120,6 @@ class UsageRecord:
description: str | None
metadata: dict[str, Any]
@dataclass
class Payment:
"""支付记录数据类"""
@@ -150,7 +141,6 @@ class Payment:
created_at: datetime
updated_at: datetime
@dataclass
class Invoice:
"""发票数据类"""
@@ -174,7 +164,6 @@ class Invoice:
created_at: datetime
updated_at: datetime
@dataclass
class Refund:
"""退款数据类"""
@@ -197,7 +186,6 @@ class Refund:
created_at: datetime
updated_at: datetime
@dataclass
class BillingHistory:
"""账单历史数据类"""
@@ -213,7 +201,6 @@ class BillingHistory:
created_at: datetime
metadata: dict[str, Any]
class SubscriptionManager:
"""订阅与计费管理器"""
@@ -2200,11 +2187,9 @@ class SubscriptionManager:
metadata=json.loads(row["metadata"] or "{}"),
)
# 全局订阅管理器实例
subscription_manager = None
def get_subscription_manager(db_path: str = "insightflow.db") -> SubscriptionManager:
"""获取订阅管理器实例(单例模式)"""
global subscription_manager

View File

@@ -23,7 +23,6 @@ from typing import Any
logger = logging.getLogger(__name__)
class TenantLimits:
"""租户资源限制常量"""
@@ -43,7 +42,6 @@ class TenantLimits:
UNLIMITED = -1
class TenantStatus(StrEnum):
"""租户状态"""
@@ -53,7 +51,6 @@ class TenantStatus(StrEnum):
EXPIRED = "expired" # 过期
PENDING = "pending" # 待激活
class TenantTier(StrEnum):
"""租户订阅层级"""
@@ -61,7 +58,6 @@ class TenantTier(StrEnum):
PRO = "pro" # 专业版
ENTERPRISE = "enterprise" # 企业版
class TenantRole(StrEnum):
"""租户角色"""
@@ -70,7 +66,6 @@ class TenantRole(StrEnum):
MEMBER = "member" # 成员
VIEWER = "viewer" # 查看者
class DomainStatus(StrEnum):
"""域名状态"""
@@ -79,7 +74,6 @@ class DomainStatus(StrEnum):
FAILED = "failed" # 验证失败
EXPIRED = "expired" # 已过期
@dataclass
class Tenant:
"""租户数据类"""
@@ -98,7 +92,6 @@ class Tenant:
resource_limits: dict[str, Any] # 资源限制
metadata: dict[str, Any] # 元数据
@dataclass
class TenantDomain:
"""租户域名数据类"""
@@ -116,7 +109,6 @@ class TenantDomain:
ssl_enabled: bool # SSL 是否启用
ssl_expires_at: datetime | None
@dataclass
class TenantBranding:
"""租户品牌配置数据类"""
@@ -134,7 +126,6 @@ class TenantBranding:
created_at: datetime
updated_at: datetime
@dataclass
class TenantMember:
"""租户成员数据类"""
@@ -151,7 +142,6 @@ class TenantMember:
last_active_at: datetime | None
status: str # active/pending/suspended
@dataclass
class TenantPermission:
"""租户权限定义数据类"""
@@ -166,7 +156,6 @@ class TenantPermission:
conditions: dict | None # 条件限制
created_at: datetime
class TenantManager:
"""租户管理器 - 多租户 SaaS 架构核心"""
@@ -1610,10 +1599,8 @@ class TenantManager:
status=row["status"],
)
# ==================== 租户上下文管理 ====================
class TenantContext:
"""租户上下文管理器 - 用于请求级别的租户隔离"""
@@ -1646,11 +1633,9 @@ class TenantContext:
cls._current_tenant_id = None
cls._current_user_id = None
# 全局租户管理器实例
tenant_manager = None
def get_tenant_manager(db_path: str = "insightflow.db") -> TenantManager:
"""获取租户管理器实例(单例模式)"""
global tenant_manager

View File

@@ -20,7 +20,6 @@ from search_manager import (
# 添加 backend 到路径
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
def test_fulltext_search():
"""测试全文搜索"""
print("\n" + "=" * 60)
@@ -63,7 +62,6 @@ def test_fulltext_search():
print("\n✓ 全文搜索测试完成")
return True
def test_semantic_search():
"""测试语义搜索"""
print("\n" + "=" * 60)
@@ -99,7 +97,6 @@ def test_semantic_search():
print("\n✓ 语义搜索测试完成")
return True
def test_entity_path_discovery():
"""测试实体路径发现"""
print("\n" + "=" * 60)
@@ -118,7 +115,6 @@ def test_entity_path_discovery():
print("\n✓ 实体路径发现测试完成")
return True
def test_knowledge_gap_detection():
"""测试知识缺口识别"""
print("\n" + "=" * 60)
@@ -137,7 +133,6 @@ def test_knowledge_gap_detection():
print("\n✓ 知识缺口识别测试完成")
return True
def test_cache_manager():
"""测试缓存管理器"""
print("\n" + "=" * 60)
@@ -185,7 +180,6 @@ def test_cache_manager():
print("\n✓ 缓存管理器测试完成")
return True
def test_task_queue():
"""测试任务队列"""
print("\n" + "=" * 60)
@@ -226,7 +220,6 @@ def test_task_queue():
print("\n✓ 任务队列测试完成")
return True
def test_performance_monitor():
"""测试性能监控"""
print("\n" + "=" * 60)
@@ -273,7 +266,6 @@ def test_performance_monitor():
print("\n✓ 性能监控测试完成")
return True
def test_search_manager():
"""测试搜索管理器"""
print("\n" + "=" * 60)
@@ -294,7 +286,6 @@ def test_search_manager():
print("\n✓ 搜索管理器测试完成")
return True
def test_performance_manager():
"""测试性能管理器"""
print("\n" + "=" * 60)
@@ -319,7 +310,6 @@ def test_performance_manager():
print("\n✓ 性能管理器测试完成")
return True
def run_all_tests():
"""运行所有测试"""
print("\n" + "=" * 60)
@@ -406,7 +396,6 @@ def run_all_tests():
return passed == total
if __name__ == "__main__":
success = run_all_tests()
sys.exit(0 if success else 1)

View File

@@ -17,7 +17,6 @@ from tenant_manager import get_tenant_manager
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
def test_tenant_management():
"""测试租户管理功能"""
print("=" * 60)
@@ -65,7 +64,6 @@ def test_tenant_management():
return tenant.id
def test_domain_management(tenant_id: str):
"""测试域名管理功能"""
print("\n" + "=" * 60)
@@ -111,7 +109,6 @@ def test_domain_management(tenant_id: str):
return domain.id
def test_branding_management(tenant_id: str):
"""测试品牌白标功能"""
print("\n" + "=" * 60)
@@ -151,7 +148,6 @@ def test_branding_management(tenant_id: str):
return branding.id
def test_member_management(tenant_id: str):
"""测试成员管理功能"""
print("\n" + "=" * 60)
@@ -206,7 +202,6 @@ def test_member_management(tenant_id: str):
return member1.id, member2.id
def test_usage_tracking(tenant_id: str):
"""测试资源使用统计功能"""
print("\n" + "=" * 60)
@@ -248,7 +243,6 @@ def test_usage_tracking(tenant_id: str):
return stats
def cleanup(tenant_id: str, domain_id: str, member_ids: list):
"""清理测试数据"""
print("\n" + "=" * 60)
@@ -272,7 +266,6 @@ 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)
@@ -310,6 +303,5 @@ def main():
except Exception as e:
print(f"⚠️ 清理失败: {e}")
if __name__ == "__main__":
main()

View File

@@ -11,7 +11,6 @@ from subscription_manager import PaymentProvider, SubscriptionManager
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
def test_subscription_manager():
"""测试订阅管理器"""
print("=" * 60)
@@ -224,7 +223,6 @@ def test_subscription_manager():
os.remove(db_path)
print(f"\n清理临时数据库: {db_path}")
if __name__ == "__main__":
try:
test_subscription_manager()

View File

@@ -13,7 +13,6 @@ from ai_manager import ModelType, PredictionType, get_ai_manager
# Add backend directory to path
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
def test_custom_model():
"""测试自定义模型功能"""
print("\n=== 测试自定义模型 ===")
@@ -88,7 +87,6 @@ def test_custom_model():
return model.id
async def test_train_and_predict(model_id: str):
"""测试训练和预测"""
print("\n=== 测试模型训练和预测 ===")
@@ -115,7 +113,6 @@ async def test_train_and_predict(model_id: str):
except Exception as e:
print(f" 预测失败: {e}")
def test_prediction_models():
"""测试预测模型"""
print("\n=== 测试预测模型 ===")
@@ -157,7 +154,6 @@ def test_prediction_models():
return trend_model.id, anomaly_model.id
async def test_predictions(trend_model_id: str, anomaly_model_id: str):
"""测试预测功能"""
print("\n=== 测试预测功能 ===")
@@ -192,7 +188,6 @@ async def test_predictions(trend_model_id: str, anomaly_model_id: str):
)
print(f" 检测结果: {anomaly_result.prediction_data}")
def test_kg_rag():
"""测试知识图谱 RAG"""
print("\n=== 测试知识图谱 RAG ===")
@@ -222,7 +217,6 @@ def test_kg_rag():
return rag.id
async def test_kg_rag_query(rag_id: str):
"""测试 RAG 查询"""
print("\n=== 测试知识图谱 RAG 查询 ===")
@@ -293,7 +287,6 @@ async def test_kg_rag_query(rag_id: str):
except Exception as e:
print(f" 查询失败: {e}")
async def test_smart_summary():
"""测试智能摘要"""
print("\n=== 测试智能摘要 ===")
@@ -341,7 +334,6 @@ async def test_smart_summary():
except Exception as e:
print(f" 生成失败: {e}")
async def main():
"""主测试函数"""
print("=" * 60)
@@ -380,6 +372,5 @@ async def main():
traceback.print_exc()
if __name__ == "__main__":
asyncio.run(main())

View File

@@ -32,7 +32,6 @@ backend_dir = os.path.dirname(os.path.abspath(__file__))
if backend_dir not in sys.path:
sys.path.insert(0, backend_dir)
class TestGrowthManager:
"""测试 Growth Manager 功能"""
@@ -735,12 +734,10 @@ class TestGrowthManager:
print("✨ 测试完成!")
print("=" * 60)
async def main():
"""主函数"""
tester = TestGrowthManager()
await tester.run_all_tests()
if __name__ == "__main__":
asyncio.run(main())

View File

@@ -29,7 +29,6 @@ backend_dir = os.path.dirname(os.path.abspath(__file__))
if backend_dir not in sys.path:
sys.path.insert(0, backend_dir)
class TestDeveloperEcosystem:
"""开发者生态系统测试类"""
@@ -688,12 +687,10 @@ console.log('Upload complete:', result.id);
print("=" * 60)
def main():
"""主函数"""
test = TestDeveloperEcosystem()
test.run_all_tests()
if __name__ == "__main__":
main()

View File

@@ -30,7 +30,6 @@ backend_dir = os.path.dirname(os.path.abspath(__file__))
if backend_dir not in sys.path:
sys.path.insert(0, backend_dir)
class TestOpsManager:
"""测试运维与监控管理器"""
@@ -722,12 +721,10 @@ class TestOpsManager:
print("=" * 60)
def main():
"""主函数"""
test = TestOpsManager()
test.run_all_tests()
if __name__ == "__main__":
main()

View File

@@ -8,7 +8,6 @@ import time
from datetime import datetime
from typing import Any
class TingwuClient:
def __init__(self):
self.access_key = os.getenv("ALI_ACCESS_KEY", "")

View File

@@ -33,7 +33,6 @@ from apscheduler.triggers.interval import IntervalTrigger
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)
class WorkflowStatus(Enum):
"""工作流状态"""
@@ -42,7 +41,6 @@ class WorkflowStatus(Enum):
ERROR = "error"
COMPLETED = "completed"
class WorkflowType(Enum):
"""工作流类型"""
@@ -52,7 +50,6 @@ class WorkflowType(Enum):
SCHEDULED_REPORT = "scheduled_report" # 定时报告
CUSTOM = "custom" # 自定义工作流
class WebhookType(Enum):
"""Webhook 类型"""
@@ -61,7 +58,6 @@ class WebhookType(Enum):
SLACK = "slack"
CUSTOM = "custom"
class TaskStatus(Enum):
"""任务执行状态"""
@@ -71,7 +67,6 @@ class TaskStatus(Enum):
FAILED = "failed"
CANCELLED = "cancelled"
@dataclass
class WorkflowTask:
"""工作流任务定义"""
@@ -95,7 +90,6 @@ class WorkflowTask:
if not self.updated_at:
self.updated_at = self.created_at
@dataclass
class WebhookConfig:
"""Webhook 配置"""
@@ -120,7 +114,6 @@ class WebhookConfig:
if not self.updated_at:
self.updated_at = self.created_at
@dataclass
class Workflow:
"""工作流定义"""
@@ -150,7 +143,6 @@ class Workflow:
if not self.updated_at:
self.updated_at = self.created_at
@dataclass
class WorkflowLog:
"""工作流执行日志"""
@@ -171,7 +163,6 @@ class WorkflowLog:
if not self.created_at:
self.created_at = datetime.now().isoformat()
class WebhookNotifier:
"""Webhook 通知器 - 支持飞书、钉钉、Slack"""
@@ -327,7 +318,6 @@ class WebhookNotifier:
"""关闭 HTTP 客户端"""
await self.http_client.aclose()
class WorkflowManager:
"""工作流管理器 - 核心管理类"""
@@ -1498,11 +1488,9 @@ class WorkflowManager:
]
}
# Singleton instance
_workflow_manager = None
def get_workflow_manager(db_manager=None) -> WorkflowManager:
"""获取 WorkflowManager 单例"""
global _workflow_manager