hankin
423baff73b
- Docker bridge 网络隔离（8000 端口封死） - Gunicorn 4 Worker 多进程 - Alembic 数据库迁移基线 - 日志轮转 20m×3 - JWT 密钥 + DB 密码 + CORS 收紧 - 3-2-1 备份链路（NAS + R740-B 冷备） - 连接池 pool_pre_ping + pool_recycle=3600
56 lines
2.8 KiB
Python
56 lines
2.8 KiB
Python
"""
|
|
新增 ORM 模型 — ai_chat_sessions / sales_logs / ai_report_drafts
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import uuid
|
|
from datetime import date, datetime
|
|
|
|
from sqlalchemy import Boolean, Date, DateTime, ForeignKey, SmallInteger, String, Text, func
|
|
from sqlalchemy.dialects.postgresql import UUID, JSONB
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from app.models.base import Base
|
|
|
|
|
|
class AiChatSession(Base):
|
|
__tablename__ = "ai_chat_sessions"
|
|
|
|
id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
|
user_id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), ForeignKey("sys_users.id"), nullable=False)
|
|
role: Mapped[str] = mapped_column(String(10), nullable=False)
|
|
content: Mapped[str] = mapped_column(Text, nullable=False)
|
|
msg_type: Mapped[str] = mapped_column(String(20), default="text")
|
|
metadata_: Mapped[dict | None] = mapped_column("metadata", JSONB, default=dict)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now())
|
|
|
|
|
|
class SalesLog(Base):
|
|
__tablename__ = "sales_logs"
|
|
|
|
id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
|
salesperson_id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), ForeignKey("sys_users.id"), nullable=False)
|
|
customer_id: Mapped[uuid.UUID | None] = mapped_column(UUID(as_uuid=True), ForeignKey("crm_customers.id"), nullable=True)
|
|
content: Mapped[str] = mapped_column(Text, nullable=False)
|
|
log_date: Mapped[date] = mapped_column(Date, default=date.today)
|
|
contact_ids: Mapped[list | None] = mapped_column(JSONB, default=list, nullable=True)
|
|
ai_processed: Mapped[bool] = mapped_column(Boolean, default=False)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now())
|
|
updated_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now(), onupdate=func.now())
|
|
is_deleted: Mapped[bool] = mapped_column(Boolean, default=False)
|
|
|
|
|
|
class AiReportDraft(Base):
|
|
__tablename__ = "ai_report_drafts"
|
|
|
|
id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
|
author_id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), ForeignKey("sys_users.id"), nullable=False)
|
|
report_type: Mapped[str] = mapped_column(String(20), nullable=False)
|
|
period_start: Mapped[date] = mapped_column(Date, nullable=False)
|
|
period_end: Mapped[date] = mapped_column(Date, nullable=False)
|
|
content_md: Mapped[str] = mapped_column(Text, nullable=False)
|
|
status: Mapped[str] = mapped_column(String(20), default="draft")
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now())
|
|
updated_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now(), onupdate=func.now())
|
|
is_deleted: Mapped[bool] = mapped_column(Boolean, default=False)
|