Files
diary-news/backend/alembic/versions/0006_meituan_blocked_topics.py

65 lines
1.7 KiB
Python
Raw Normal View History

"""美团"无可奉告"主题清单 — 命中则不调美团 API,直接写固定文案
- llm_settings.meituan_blocked_topics JSONB (默认 ["时政", "国际", "军事", "政治", "战争", "冲突", "制裁", "选举"])
- llm_settings.meituan_blocked_keywords JSONB (默认 [])
- llm_settings.meituan_no_comment_text TEXT (默认 "无可奉告")
Revision ID: 0006
Revises: 0005
Create Date: 2026-06-12
"""
from __future__ import annotations
from typing import Sequence, Union
import sqlalchemy as sa
from alembic import op
from sqlalchemy.dialects.postgresql import JSONB
revision: str = "0006"
down_revision: Union[str, None] = "0005"
branch_labels = None
depends_on = None
DEFAULT_TOPICS = ["时政", "国际", "军事", "政治", "战争", "冲突", "制裁", "选举"]
def upgrade() -> None:
import json
default_topics_json = sa.text(repr(json.dumps(DEFAULT_TOPICS, ensure_ascii=False)))
op.add_column(
"llm_settings",
sa.Column(
"meituan_blocked_topics",
JSONB,
nullable=False,
server_default=default_topics_json,
),
)
op.add_column(
"llm_settings",
sa.Column(
"meituan_blocked_keywords",
JSONB,
nullable=False,
server_default=sa.text("'[]'::jsonb"),
),
)
op.add_column(
"llm_settings",
sa.Column(
"meituan_no_comment_text",
sa.Text,
nullable=False,
server_default="无可奉告",
),
)
def downgrade() -> None:
op.drop_column("llm_settings", "meituan_no_comment_text")
op.drop_column("llm_settings", "meituan_blocked_keywords")
op.drop_column("llm_settings", "meituan_blocked_topics")