from sqlalchemy import Column, Integer, String, BigInteger, JSON, DateTime, Boolean, func
from sqlalchemy.orm import declarative_base

Base = declarative_base()

class TaskKeyword(Base):
    __tablename__ = "wx_trend_keywords"
    id = Column(Integer, primary_key=True)
    keyword = Column(String(176), unique=True)
    is_active = Column(Boolean, default=True)
    priority = Column(Integer, default=0)

class WxTrendData(Base):
    __tablename__ = "wx_trend_data"
    id = Column(BigInteger, primary_key=True)
    keyword = Column(String(255))
    ymd = Column(String(20))
    channel_score = Column(JSON) # 使用 JSON 类型
    updated_at = Column(DateTime, server_default=func.now(), onupdate=func.now())