JiuContinent
d7db84571d
feat: 添加飞书用户模块和订阅功能支持 - 新增feishu_users模块用于处理飞书用户身份验证和权限管理 - 新增subscriptions模块用于处理订阅相关功能 - 新增personalization模块用于个性化服务 - 在alembic迁移配置中注册新的模型模块 - 在API路由器中添加feishu_users和subscriptions路由 - 实现事件调度服务的改进，包括错误处理和状态更新优化 - 添加飞书命令处理的权限检查机制 - 实现飞书应用票据事件处理 - 改进审计日志记录功能 ```
132 lines
4.2 KiB
Python
132 lines
4.2 KiB
Python
from datetime import datetime
|
|
from uuid import uuid4
|
|
|
|
from sqlalchemy import DateTime, ForeignKey, Integer, String, Text, UniqueConstraint
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from app.core.database import Base
|
|
from app.core.utils.time import utc_now
|
|
from app.modules.feishu_users.models import FeishuUser
|
|
from app.modules.personalization.constants import (
|
|
CONVERSATION_CODE_PREFIX,
|
|
PREFERENCE_CODE_PREFIX,
|
|
PreferenceSource,
|
|
)
|
|
|
|
|
|
def _public_code(prefix: str) -> str:
|
|
return f"{prefix}-{uuid4().hex}"
|
|
|
|
|
|
class UserPreference(Base):
|
|
__tablename__ = "user_preferences"
|
|
__table_args__ = (
|
|
UniqueConstraint(
|
|
"owner_id",
|
|
"category",
|
|
"normalized_value",
|
|
name="uq_user_preference_owner_category_value",
|
|
),
|
|
)
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
code: Mapped[str] = mapped_column(
|
|
String(64),
|
|
default=lambda: _public_code(PREFERENCE_CODE_PREFIX),
|
|
unique=True,
|
|
index=True,
|
|
)
|
|
owner_id: Mapped[int] = mapped_column(
|
|
ForeignKey("feishu_users.id", ondelete="CASCADE"),
|
|
index=True,
|
|
)
|
|
owner: Mapped[FeishuUser] = relationship()
|
|
category: Mapped[str] = mapped_column(String(32), index=True)
|
|
value: Mapped[str] = mapped_column(Text)
|
|
normalized_value: Mapped[str] = mapped_column(String(1000))
|
|
source: Mapped[str] = mapped_column(
|
|
String(32),
|
|
default=PreferenceSource.EXPLICIT,
|
|
index=True,
|
|
)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=utc_now, index=True)
|
|
updated_at: Mapped[datetime] = mapped_column(
|
|
DateTime,
|
|
default=utc_now,
|
|
onupdate=utc_now,
|
|
)
|
|
|
|
|
|
class AIConversation(Base):
|
|
__tablename__ = "ai_conversations"
|
|
__table_args__ = (
|
|
UniqueConstraint(
|
|
"owner_id",
|
|
"chat_type",
|
|
"chat_key",
|
|
name="uq_ai_conversation_owner_chat",
|
|
),
|
|
)
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
code: Mapped[str] = mapped_column(
|
|
String(64),
|
|
default=lambda: _public_code(CONVERSATION_CODE_PREFIX),
|
|
unique=True,
|
|
index=True,
|
|
)
|
|
owner_id: Mapped[int] = mapped_column(
|
|
ForeignKey("feishu_users.id", ondelete="CASCADE"),
|
|
index=True,
|
|
)
|
|
owner: Mapped[FeishuUser] = relationship()
|
|
chat_type: Mapped[str] = mapped_column(String(32), index=True)
|
|
chat_key: Mapped[str] = mapped_column(String(256), index=True)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=utc_now, index=True)
|
|
updated_at: Mapped[datetime] = mapped_column(
|
|
DateTime,
|
|
default=utc_now,
|
|
onupdate=utc_now,
|
|
index=True,
|
|
)
|
|
messages: Mapped[list["AIConversationMessage"]] = relationship(
|
|
back_populates="conversation",
|
|
cascade="all, delete-orphan",
|
|
passive_deletes=True,
|
|
order_by="AIConversationMessage.id",
|
|
)
|
|
|
|
|
|
class AIConversationMessage(Base):
|
|
__tablename__ = "ai_conversation_messages"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
conversation_id: Mapped[int] = mapped_column(
|
|
ForeignKey("ai_conversations.id", ondelete="CASCADE"),
|
|
index=True,
|
|
)
|
|
role: Mapped[str] = mapped_column(String(32), index=True)
|
|
content: Mapped[str] = mapped_column(Text)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=utc_now, index=True)
|
|
conversation: Mapped[AIConversation] = relationship(back_populates="messages")
|
|
|
|
|
|
class PersonalDataErasureRequest(Base):
|
|
__tablename__ = "personal_data_erasure_requests"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
owner_id: Mapped[int] = mapped_column(
|
|
ForeignKey("feishu_users.id", ondelete="CASCADE"),
|
|
unique=True,
|
|
index=True,
|
|
)
|
|
owner: Mapped[FeishuUser] = relationship()
|
|
token_hash: Mapped[str] = mapped_column(String(64))
|
|
expires_at: Mapped[datetime] = mapped_column(DateTime, index=True)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=utc_now, index=True)
|
|
updated_at: Mapped[datetime] = mapped_column(
|
|
DateTime,
|
|
default=utc_now,
|
|
onupdate=utc_now,
|
|
)
|