JiuContinent
dc8605ce3f
refactor(core): 重构核心模块结构并更新导入路径 - 将配置相关的设置从 app.core.config 移除 - 将常量定义从 app.core.constants 移除 - 将数据库相关功能从 app.core.database 移除 - 将基础数据库模型从 app.core.db_base 移除 - 将敏感信息掩码功能从 app.core.masking 移除 - 将中间件定义从 app.core.middleware 移除 - 将操作保护功能从 app.core.operation_guard 移除 - 将分页工具从 app.core.pagination 移除 - 将请求上下文管理从 app.core.request_context 移除 - 将调度器功能从 app.core.scheduler 移除 - 将安全认证逻辑从 app.core.security 移除 - 将任务队列相关功能从 app.core.task_queue 移除 - 将时间工具从 app.core.time 移除 - 更新 alembic 配置中的 Base 模型导入路径 - 更新各模块中对重构后组件的引用路径 ```
297 lines
16 KiB
Python
297 lines
16 KiB
Python
from datetime import date, datetime
|
|
from decimal import Decimal
|
|
|
|
from sqlalchemy import JSON, Date, DateTime, Integer, Numeric, String, Text
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from app.core.constants import ActorValue
|
|
from app.core.database import Base
|
|
from app.core.utils.time import utc_now
|
|
from app.modules.business.constants import (
|
|
AccountType,
|
|
PriorityValue,
|
|
RiskLevel,
|
|
SourceSystem,
|
|
StatusValue,
|
|
VersionValue,
|
|
)
|
|
|
|
|
|
class TimestampMixin:
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=utc_now)
|
|
updated_at: Mapped[datetime] = mapped_column(
|
|
DateTime, default=utc_now, onupdate=utc_now
|
|
)
|
|
|
|
|
|
class Project(Base, TimestampMixin):
|
|
__tablename__ = "projects"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
|
|
name: Mapped[str] = mapped_column(String(255), index=True)
|
|
owner: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)
|
|
status: Mapped[str] = mapped_column(String(64), default=StatusValue.INITIATED, index=True)
|
|
priority: Mapped[str] = mapped_column(String(32), default=PriorityValue.P2)
|
|
progress_percent: Mapped[int] = mapped_column(Integer, default=0)
|
|
risk_level: Mapped[str] = mapped_column(String(32), default=RiskLevel.LOW, index=True)
|
|
budget_amount: Mapped[Decimal] = mapped_column(Numeric(14, 2), default=0)
|
|
actual_amount: Mapped[Decimal] = mapped_column(Numeric(14, 2), default=0)
|
|
start_date: Mapped[date | None] = mapped_column(Date, nullable=True)
|
|
due_date: Mapped[date | None] = mapped_column(Date, nullable=True, index=True)
|
|
description: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
source_system: Mapped[str] = mapped_column(String(64), default=SourceSystem.INTERNAL)
|
|
external_id: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)
|
|
|
|
|
|
class WorkTask(Base, TimestampMixin):
|
|
__tablename__ = "work_tasks"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
|
|
title: Mapped[str] = mapped_column(String(255), index=True)
|
|
project_code: Mapped[str | None] = mapped_column(String(64), nullable=True, index=True)
|
|
owner: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)
|
|
status: Mapped[str] = mapped_column(String(64), default=StatusValue.TODO, index=True)
|
|
priority: Mapped[str] = mapped_column(String(32), default=PriorityValue.P2)
|
|
due_date: Mapped[date | None] = mapped_column(Date, nullable=True, index=True)
|
|
completed_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
|
|
blocker: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
description: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
source_system: Mapped[str] = mapped_column(String(64), default=SourceSystem.INTERNAL)
|
|
external_id: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)
|
|
|
|
|
|
class Procurement(Base, TimestampMixin):
|
|
__tablename__ = "procurements"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
|
|
name: Mapped[str] = mapped_column(String(255), index=True)
|
|
applicant: Mapped[str | None] = mapped_column(String(128), nullable=True)
|
|
project_code: Mapped[str | None] = mapped_column(String(64), nullable=True, index=True)
|
|
supplier_name: Mapped[str | None] = mapped_column(String(255), nullable=True)
|
|
budget_subject: Mapped[str | None] = mapped_column(String(128), nullable=True)
|
|
expected_amount: Mapped[Decimal] = mapped_column(Numeric(14, 2), default=0)
|
|
actual_amount: Mapped[Decimal] = mapped_column(Numeric(14, 2), default=0)
|
|
approval_status: Mapped[str] = mapped_column(String(64), default=StatusValue.DRAFT, index=True)
|
|
delivery_status: Mapped[str] = mapped_column(
|
|
String(64),
|
|
default=StatusValue.UNDELIVERED,
|
|
index=True,
|
|
)
|
|
payment_status: Mapped[str] = mapped_column(String(64), default=StatusValue.UNPAID, index=True)
|
|
comparison_summary: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
|
|
|
|
class Expense(Base, TimestampMixin):
|
|
__tablename__ = "expenses"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
|
|
expense_type: Mapped[str] = mapped_column(String(128), index=True)
|
|
amount: Mapped[Decimal] = mapped_column(Numeric(14, 2), default=0)
|
|
applicant: Mapped[str | None] = mapped_column(String(128), nullable=True)
|
|
department: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)
|
|
project_code: Mapped[str | None] = mapped_column(String(64), nullable=True, index=True)
|
|
budget_subject: Mapped[str | None] = mapped_column(String(128), nullable=True)
|
|
payment_account: Mapped[str | None] = mapped_column(String(128), nullable=True)
|
|
invoice_status: Mapped[str] = mapped_column(
|
|
String(64),
|
|
default=StatusValue.INVOICE_NOT_RECEIVED,
|
|
)
|
|
approval_status: Mapped[str] = mapped_column(String(64), default=StatusValue.DRAFT, index=True)
|
|
payment_status: Mapped[str] = mapped_column(String(64), default=StatusValue.UNPAID, index=True)
|
|
|
|
|
|
class FundAccount(Base, TimestampMixin):
|
|
__tablename__ = "fund_accounts"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
|
|
name: Mapped[str] = mapped_column(String(255), index=True)
|
|
account_type: Mapped[str] = mapped_column(String(64), default=AccountType.BANK)
|
|
current_balance: Mapped[Decimal] = mapped_column(Numeric(16, 2), default=0)
|
|
expected_receivable: Mapped[Decimal] = mapped_column(Numeric(16, 2), default=0)
|
|
expected_payable: Mapped[Decimal] = mapped_column(Numeric(16, 2), default=0)
|
|
safety_line: Mapped[Decimal] = mapped_column(Numeric(16, 2), default=0)
|
|
risk_level: Mapped[str] = mapped_column(String(32), default=RiskLevel.LOW, index=True)
|
|
note: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
|
|
|
|
class Policy(Base, TimestampMixin):
|
|
__tablename__ = "policies"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
|
|
title: Mapped[str] = mapped_column(String(255), index=True)
|
|
policy_type: Mapped[str | None] = mapped_column(String(128), nullable=True)
|
|
owner_department: Mapped[str | None] = mapped_column(String(128), nullable=True)
|
|
version: Mapped[str] = mapped_column(String(32), default=VersionValue.V1_0)
|
|
status: Mapped[str] = mapped_column(String(64), default=StatusValue.POLICY_DRAFT, index=True)
|
|
effective_date: Mapped[date | None] = mapped_column(Date, nullable=True)
|
|
feishu_doc_url: Mapped[str | None] = mapped_column(String(1024), nullable=True)
|
|
summary: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
|
|
|
|
class Standard(Base, TimestampMixin):
|
|
__tablename__ = "standards"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
|
|
title: Mapped[str] = mapped_column(String(255), index=True)
|
|
standard_type: Mapped[str | None] = mapped_column(String(128), nullable=True)
|
|
applies_to: Mapped[str | None] = mapped_column(String(255), nullable=True)
|
|
status: Mapped[str] = mapped_column(String(64), default=StatusValue.VALID, index=True)
|
|
check_items: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
remediation: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
policy_code: Mapped[str | None] = mapped_column(String(64), nullable=True, index=True)
|
|
|
|
|
|
class PerformanceMetric(Base, TimestampMixin):
|
|
__tablename__ = "performance_metrics"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
|
|
name: Mapped[str] = mapped_column(String(255), index=True)
|
|
applies_to_role: Mapped[str | None] = mapped_column(String(128), nullable=True)
|
|
formula: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
weight: Mapped[Decimal] = mapped_column(Numeric(5, 2), default=0)
|
|
data_source: Mapped[str | None] = mapped_column(String(255), nullable=True)
|
|
auto_score: Mapped[Decimal] = mapped_column(Numeric(8, 2), default=0)
|
|
confirmed_score: Mapped[Decimal | None] = mapped_column(Numeric(8, 2), nullable=True)
|
|
status: Mapped[str] = mapped_column(String(64), default=StatusValue.DRAFT, index=True)
|
|
|
|
|
|
class Supplier(Base, TimestampMixin):
|
|
__tablename__ = "suppliers"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
|
|
name: Mapped[str] = mapped_column(String(255), index=True)
|
|
category: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)
|
|
contact: Mapped[str | None] = mapped_column(String(128), nullable=True)
|
|
quality_score: Mapped[Decimal] = mapped_column(Numeric(5, 2), default=0)
|
|
delivery_score: Mapped[Decimal] = mapped_column(Numeric(5, 2), default=0)
|
|
price_score: Mapped[Decimal] = mapped_column(Numeric(5, 2), default=0)
|
|
risk_level: Mapped[str] = mapped_column(String(32), default=RiskLevel.LOW, index=True)
|
|
blacklist_status: Mapped[str] = mapped_column(
|
|
String(32),
|
|
default=StatusValue.NORMAL,
|
|
index=True,
|
|
)
|
|
|
|
|
|
class AttendanceRecord(Base, TimestampMixin):
|
|
__tablename__ = "attendance_records"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
|
|
employee_name: Mapped[str] = mapped_column(String(128), index=True)
|
|
employee_id: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)
|
|
department: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)
|
|
project_code: Mapped[str | None] = mapped_column(String(64), nullable=True, index=True)
|
|
work_date: Mapped[date] = mapped_column(Date, index=True)
|
|
check_in_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
|
|
check_out_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
|
|
status: Mapped[str] = mapped_column(String(64), default=StatusValue.NORMAL_CN, index=True)
|
|
location: Mapped[str | None] = mapped_column(String(255), nullable=True)
|
|
source_system: Mapped[str] = mapped_column(String(64), default=SourceSystem.INTERNAL)
|
|
external_id: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)
|
|
note: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
|
|
|
|
class WorkReport(Base, TimestampMixin):
|
|
__tablename__ = "work_reports"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
|
|
report_type: Mapped[str] = mapped_column(String(32), index=True)
|
|
title: Mapped[str] = mapped_column(String(255), index=True)
|
|
reporter: Mapped[str] = mapped_column(String(128), default=ActorValue.SYSTEM, index=True)
|
|
department: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)
|
|
project_code: Mapped[str | None] = mapped_column(String(64), nullable=True, index=True)
|
|
period_start: Mapped[date] = mapped_column(Date, index=True)
|
|
period_end: Mapped[date] = mapped_column(Date, index=True)
|
|
content: Mapped[str] = mapped_column(Text)
|
|
metrics: Mapped[dict | None] = mapped_column(JSON, nullable=True)
|
|
risk_summary: Mapped[dict | None] = mapped_column(JSON, nullable=True)
|
|
status: Mapped[str] = mapped_column(String(64), default=StatusValue.GENERATED, index=True)
|
|
source_system: Mapped[str] = mapped_column(String(64), default=SourceSystem.INTERNAL)
|
|
|
|
|
|
class RiskEvent(Base, TimestampMixin):
|
|
__tablename__ = "risk_events"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
|
|
title: Mapped[str] = mapped_column(String(255), index=True)
|
|
risk_type: Mapped[str] = mapped_column(String(64), index=True)
|
|
risk_level: Mapped[str] = mapped_column(String(32), default=RiskLevel.MEDIUM, index=True)
|
|
status: Mapped[str] = mapped_column(String(32), default=StatusValue.OPEN, index=True)
|
|
source_domain: Mapped[str] = mapped_column(String(128), index=True)
|
|
source_record_id: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)
|
|
project_code: Mapped[str | None] = mapped_column(String(64), nullable=True, index=True)
|
|
owner: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)
|
|
detected_at: Mapped[datetime] = mapped_column(DateTime, default=utc_now, index=True)
|
|
due_date: Mapped[date | None] = mapped_column(Date, nullable=True, index=True)
|
|
assigned_to: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)
|
|
resolved_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
|
|
closed_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
|
|
closed_reason: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
review_summary: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
description: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
mitigation: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
evidence: Mapped[dict | None] = mapped_column(JSON, nullable=True)
|
|
|
|
|
|
class RiskEventAction(Base):
|
|
__tablename__ = "risk_event_actions"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
|
|
risk_event_id: Mapped[int] = mapped_column(Integer, index=True)
|
|
action: Mapped[str] = mapped_column(String(64), index=True)
|
|
actor: Mapped[str] = mapped_column(String(128), default=ActorValue.SYSTEM, index=True)
|
|
from_status: Mapped[str | None] = mapped_column(String(32), nullable=True)
|
|
to_status: Mapped[str | None] = mapped_column(String(32), nullable=True)
|
|
assigned_to: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)
|
|
comment: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
payload: Mapped[dict | None] = mapped_column(JSON, nullable=True)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=utc_now, index=True)
|
|
|
|
|
|
class LegacySyncRun(Base, TimestampMixin):
|
|
__tablename__ = "legacy_sync_runs"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
|
|
domain: Mapped[str] = mapped_column(String(128), index=True)
|
|
source_table: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)
|
|
status: Mapped[str] = mapped_column(String(32), default=StatusValue.RUNNING, index=True)
|
|
started_at: Mapped[datetime] = mapped_column(DateTime, default=utc_now, index=True)
|
|
finished_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
|
|
created_count: Mapped[int] = mapped_column(Integer, default=0)
|
|
updated_count: Mapped[int] = mapped_column(Integer, default=0)
|
|
skipped_count: Mapped[int] = mapped_column(Integer, default=0)
|
|
error_message: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
note: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
|
|
|
|
class ReportPushRun(Base, TimestampMixin):
|
|
__tablename__ = "report_push_runs"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
code: Mapped[str] = mapped_column(String(64), unique=True, index=True)
|
|
report_type: Mapped[str] = mapped_column(String(64), index=True)
|
|
title: Mapped[str | None] = mapped_column(String(255), nullable=True)
|
|
receive_id: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)
|
|
receive_id_type: Mapped[str] = mapped_column(String(64), index=True)
|
|
status: Mapped[str] = mapped_column(String(32), index=True)
|
|
task_id: Mapped[str | None] = mapped_column(String(128), nullable=True, index=True)
|
|
actor: Mapped[str] = mapped_column(String(128), default=ActorValue.SYSTEM, index=True)
|
|
provider_response: Mapped[dict | None] = mapped_column(JSON, nullable=True)
|
|
error_message: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
queued_at: Mapped[datetime] = mapped_column(DateTime, default=utc_now, index=True)
|
|
sent_at: Mapped[datetime | None] = mapped_column(DateTime, nullable=True)
|