v6ole
754214692e
- 新增 ai_enabled/ai_api_key/ai_base_url/ai_model 等配置项，通过 .env 管理 - 新增 app/services/ai_analyzer.py — DeepSeek API 调用 + 详情页正文提取 - 新增 extract_page_content() 从详情页抓取正文供 AI 分析 - Announcement 模型新增 ai_relevant / ai_analysis 字段 - 流水线集成 AI 分析步骤（关键词匹配后、通知前） - AI 标记为可承接的项目额外发送 markdown 着重通知 - 创建 alembic 迁移版本 6e8f4c2d1b0a
45 lines
2.1 KiB
Python
45 lines
2.1 KiB
Python
import hashlib
|
|
from datetime import datetime
|
|
|
|
from sqlalchemy import Boolean, DateTime, Integer, String, Text, func
|
|
from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
|
|
|
|
|
|
class Base(DeclarativeBase):
|
|
pass
|
|
|
|
|
|
class Announcement(Base):
|
|
__tablename__ = "announcements"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
|
|
title: Mapped[str] = mapped_column(String(500), nullable=False)
|
|
publish_date: Mapped[datetime] = mapped_column(DateTime, nullable=False)
|
|
purchase_name: Mapped[str] = mapped_column(String(200), default="")
|
|
content_url: Mapped[str] = mapped_column(Text, default="")
|
|
source_code: Mapped[str] = mapped_column(String(50), nullable=False)
|
|
source_name: Mapped[str] = mapped_column(String(100), nullable=False)
|
|
announcement_type: Mapped[str] = mapped_column(String(50), default="purchase")
|
|
content_hash: Mapped[str] = mapped_column(String(64), unique=True, nullable=False)
|
|
crawl_mode: Mapped[str] = mapped_column(String(20), default="auto")
|
|
is_new: Mapped[bool] = mapped_column(Boolean, default=True)
|
|
is_sent: Mapped[bool] = mapped_column(Boolean, default=False)
|
|
keyword_matched: Mapped[bool] = mapped_column(Boolean, default=False)
|
|
ai_relevant: Mapped[bool | None] = mapped_column(Boolean, nullable=True, default=None)
|
|
ai_analysis: Mapped[str | None] = mapped_column(Text, nullable=True, default=None)
|
|
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()) # noqa: E501
|
|
|
|
@staticmethod
|
|
def generate_hash(title: str, publish_date: str, purchase_name: str,
|
|
content_url: str, source_code: str) -> str:
|
|
content = f"{title}|{publish_date}|{purchase_name}|{content_url}|{source_code}"
|
|
return hashlib.sha256(content.encode("utf-8")).hexdigest()
|
|
|
|
@staticmethod
|
|
def source_map() -> dict:
|
|
import json
|
|
|
|
from app.config import settings
|
|
return json.loads(settings.announcement_sources)
|