6635b8fea8
Source/SourceKind、UserRole、SubscriptionMatch 三个 enum 改用 values_callable 保证 PG 看到 'rss' / 'owner' / 'any' 等小写 value,而非 'RSS' 大写 name。 seed_sources 同步改为显式字符串。
53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
"""关键词订阅(命中即通知)。"""
|
|
from __future__ import annotations
|
|
|
|
import enum
|
|
from datetime import datetime
|
|
|
|
from sqlalchemy import (
|
|
Boolean,
|
|
DateTime,
|
|
Enum,
|
|
ForeignKey,
|
|
String,
|
|
Text,
|
|
func,
|
|
)
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from app.database import Base
|
|
|
|
|
|
class SubscriptionMatch(str, enum.Enum):
|
|
ANY = "any" # 标题或正文
|
|
TITLE = "title"
|
|
BODY = "body"
|
|
|
|
|
|
class Subscription(Base):
|
|
__tablename__ = "subscriptions"
|
|
|
|
id: Mapped[int] = mapped_column(primary_key=True)
|
|
user_id: Mapped[int] = mapped_column(
|
|
ForeignKey("users.id", ondelete="CASCADE"), nullable=False, index=True
|
|
)
|
|
keyword: Mapped[str] = mapped_column(String(255), nullable=False)
|
|
# 简单关键词,匹配走 ILIKE '%kw%';后续可加 regex/lucene
|
|
match_in: Mapped[SubscriptionMatch] = mapped_column(
|
|
Enum(
|
|
SubscriptionMatch,
|
|
name="subscription_match",
|
|
values_callable=lambda x: [e.value for e in x],
|
|
),
|
|
default=SubscriptionMatch.ANY,
|
|
nullable=False,
|
|
)
|
|
channel: Mapped[str] = mapped_column(String(32), default="telegram", nullable=False)
|
|
# telegram / email / web
|
|
target: Mapped[str | None] = mapped_column(Text) # chat_id / email
|
|
enabled: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False)
|
|
last_hit_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))
|
|
created_at: Mapped[datetime] = mapped_column(
|
|
DateTime(timezone=True), server_default=func.now(), nullable=False
|
|
)
|