chore: prepare yanting monorepo handoff
This commit is contained in:
@@ -0,0 +1,302 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import datetime as dt
|
||||
|
||||
from sqlalchemy import BigInteger, Boolean, DateTime, ForeignKey, Index, Integer, String, Text, UniqueConstraint
|
||||
from sqlalchemy.dialects.mysql import MEDIUMTEXT
|
||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
|
||||
from app.db import Base
|
||||
|
||||
|
||||
def utcnow() -> dt.datetime:
|
||||
return dt.datetime.now(dt.UTC).replace(tzinfo=None)
|
||||
|
||||
|
||||
MediumText = Text().with_variant(MEDIUMTEXT, "mysql")
|
||||
|
||||
|
||||
class Institution(Base):
|
||||
__tablename__ = "institutions"
|
||||
|
||||
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
||||
institution_id: Mapped[str] = mapped_column(String(64), unique=True, nullable=False)
|
||||
name_cn: Mapped[str] = mapped_column(String(255), nullable=False)
|
||||
name_en: Mapped[str | None] = mapped_column(String(255))
|
||||
institution_type: Mapped[str] = mapped_column(String(32), nullable=False)
|
||||
source_tier: Mapped[str] = mapped_column(String(16), nullable=False)
|
||||
website_url: Mapped[str | None] = mapped_column(String(512))
|
||||
covered_topics: Mapped[str | None] = mapped_column(Text)
|
||||
intro_cn: Mapped[str | None] = mapped_column(Text)
|
||||
credibility_note: Mapped[str | None] = mapped_column(Text)
|
||||
report_count: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
|
||||
latest_report_id: Mapped[str | None] = mapped_column(String(64))
|
||||
latest_report_at: Mapped[dt.datetime | None] = mapped_column(DateTime)
|
||||
status: Mapped[str] = mapped_column(String(16), nullable=False, default="active")
|
||||
created_at: Mapped[dt.datetime] = mapped_column(DateTime, nullable=False, default=utcnow)
|
||||
updated_at: Mapped[dt.datetime] = mapped_column(DateTime, nullable=False, default=utcnow, onupdate=utcnow)
|
||||
|
||||
reports: Mapped[list[Report]] = relationship(back_populates="institution")
|
||||
|
||||
__table_args__ = (Index("ix_institutions_status_latest", "status", "latest_report_at"),)
|
||||
|
||||
|
||||
class Report(Base):
|
||||
__tablename__ = "reports"
|
||||
|
||||
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
||||
report_id: Mapped[str] = mapped_column(String(64), unique=True, nullable=False)
|
||||
report_type: Mapped[str] = mapped_column(String(16), nullable=False, default="single")
|
||||
title_cn: Mapped[str] = mapped_column(String(512), nullable=False)
|
||||
subtitle_cn: Mapped[str | None] = mapped_column(String(512))
|
||||
original_title: Mapped[str | None] = mapped_column(String(512))
|
||||
one_liner: Mapped[str | None] = mapped_column(String(512))
|
||||
institution_id: Mapped[str] = mapped_column(String(64), ForeignKey("institutions.institution_id"), nullable=False)
|
||||
co_institution_ids: Mapped[str | None] = mapped_column(Text)
|
||||
source_tier: Mapped[str] = mapped_column(String(32), nullable=False)
|
||||
source_url: Mapped[str | None] = mapped_column(String(512))
|
||||
source_note: Mapped[str] = mapped_column(Text, nullable=False)
|
||||
published_at: Mapped[dt.datetime | None] = mapped_column(DateTime)
|
||||
interpreted_at: Mapped[dt.datetime | None] = mapped_column(DateTime)
|
||||
released_at: Mapped[dt.datetime | None] = mapped_column(DateTime)
|
||||
topics: Mapped[str | None] = mapped_column(Text)
|
||||
language: Mapped[str] = mapped_column(String(8), nullable=False, default="en")
|
||||
has_audio: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
|
||||
display_status: Mapped[str] = mapped_column(String(16), nullable=False, default="draft")
|
||||
display_version: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
|
||||
cache_version: Mapped[str] = mapped_column(String(128), nullable=False)
|
||||
risk_disclaimer: Mapped[str | None] = mapped_column(Text)
|
||||
interpretation_label: Mapped[str | None] = mapped_column(String(64), default="研报解读")
|
||||
created_at: Mapped[dt.datetime] = mapped_column(DateTime, nullable=False, default=utcnow)
|
||||
updated_at: Mapped[dt.datetime] = mapped_column(DateTime, nullable=False, default=utcnow, onupdate=utcnow)
|
||||
|
||||
institution: Mapped[Institution] = relationship(back_populates="reports")
|
||||
|
||||
__table_args__ = (
|
||||
Index("ix_reports_status_released", "display_status", "released_at"),
|
||||
Index("ix_reports_institution_released", "institution_id", "released_at"),
|
||||
Index("ix_reports_audio_released", "has_audio", "released_at"),
|
||||
)
|
||||
|
||||
|
||||
class RawArtifact(Base):
|
||||
__tablename__ = "raw_artifacts"
|
||||
|
||||
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
||||
raw_artifact_id: Mapped[str] = mapped_column(String(64), unique=True, nullable=False)
|
||||
report_id: Mapped[str] = mapped_column(String(64), ForeignKey("reports.report_id"), nullable=False)
|
||||
provider: Mapped[str] = mapped_column(String(32), nullable=False, default="notebooklm")
|
||||
artifact_type: Mapped[str] = mapped_column(String(64), nullable=False)
|
||||
conversation_id: Mapped[str | None] = mapped_column(String(128))
|
||||
source_id: Mapped[str | None] = mapped_column(String(128))
|
||||
notebook_id: Mapped[str | None] = mapped_column(String(128))
|
||||
source_language: Mapped[str | None] = mapped_column(String(8))
|
||||
payload_format: Mapped[str] = mapped_column(String(16), nullable=False)
|
||||
payload_ref: Mapped[str | None] = mapped_column(String(512))
|
||||
sha256: Mapped[str | None] = mapped_column(String(128))
|
||||
status: Mapped[str] = mapped_column(String(16), nullable=False, default="pending")
|
||||
error: Mapped[str | None] = mapped_column(Text)
|
||||
size_bytes: Mapped[int | None] = mapped_column(BigInteger)
|
||||
generated_at: Mapped[dt.datetime | None] = mapped_column(DateTime)
|
||||
ingested_at: Mapped[dt.datetime | None] = mapped_column(DateTime)
|
||||
is_publish_blocking: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
|
||||
requires_human_review: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
|
||||
quality_flags: Mapped[str | None] = mapped_column(Text)
|
||||
retention_status: Mapped[str] = mapped_column(String(32), nullable=False, default="retained")
|
||||
created_at: Mapped[dt.datetime] = mapped_column(DateTime, nullable=False, default=utcnow)
|
||||
|
||||
__table_args__ = (
|
||||
Index("ix_raw_report_type", "report_id", "artifact_type"),
|
||||
Index("ix_raw_report_status", "report_id", "status"),
|
||||
Index("ix_raw_retention", "retention_status"),
|
||||
)
|
||||
|
||||
|
||||
class DisplayArtifact(Base):
|
||||
__tablename__ = "display_artifacts"
|
||||
|
||||
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
||||
display_artifact_id: Mapped[str] = mapped_column(String(64), unique=True, nullable=False)
|
||||
report_id: Mapped[str] = mapped_column(String(64), ForeignKey("reports.report_id"), nullable=False)
|
||||
display_version: Mapped[int] = mapped_column(Integer, nullable=False, default=1)
|
||||
title_cn: Mapped[str] = mapped_column(String(512), nullable=False)
|
||||
summary_cn: Mapped[str | None] = mapped_column(Text)
|
||||
source_label: Mapped[str | None] = mapped_column(String(255))
|
||||
interpretation_label: Mapped[str | None] = mapped_column(String(64), default="研报解读")
|
||||
ai_generated_label: Mapped[str | None] = mapped_column(String(128))
|
||||
synthesis_type: Mapped[str | None] = mapped_column(String(16))
|
||||
source_disclosure_text: Mapped[str | None] = mapped_column(Text)
|
||||
review_status: Mapped[str] = mapped_column(String(16), nullable=False, default="review")
|
||||
reviewed_by: Mapped[str | None] = mapped_column(String(128))
|
||||
reviewed_at: Mapped[dt.datetime | None] = mapped_column(DateTime)
|
||||
published_at: Mapped[dt.datetime | None] = mapped_column(DateTime)
|
||||
created_at: Mapped[dt.datetime] = mapped_column(DateTime, nullable=False, default=utcnow)
|
||||
updated_at: Mapped[dt.datetime] = mapped_column(DateTime, nullable=False, default=utcnow, onupdate=utcnow)
|
||||
|
||||
__table_args__ = (Index("ix_display_artifacts_report_status_version", "report_id", "review_status", "display_version"),)
|
||||
|
||||
|
||||
class DisplayModule(Base):
|
||||
__tablename__ = "display_modules"
|
||||
|
||||
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
||||
module_id: Mapped[str] = mapped_column(String(64), unique=True, nullable=False)
|
||||
report_id: Mapped[str] = mapped_column(String(64), ForeignKey("reports.report_id"), nullable=False)
|
||||
display_artifact_id: Mapped[str] = mapped_column(String(64), ForeignKey("display_artifacts.display_artifact_id"), nullable=False)
|
||||
type: Mapped[str] = mapped_column(String(32), nullable=False)
|
||||
title_cn: Mapped[str | None] = mapped_column(String(255))
|
||||
content_format: Mapped[str] = mapped_column(String(16), nullable=False)
|
||||
content: Mapped[str | None] = mapped_column(MediumText)
|
||||
content_ref: Mapped[str | None] = mapped_column(String(512))
|
||||
content_etag: Mapped[str | None] = mapped_column(String(64))
|
||||
source_raw_artifact_ids: Mapped[str | None] = mapped_column(Text)
|
||||
status: Mapped[str] = mapped_column(String(16), nullable=False, default="missing")
|
||||
sort_order: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
|
||||
version: Mapped[int] = mapped_column(Integer, nullable=False, default=1)
|
||||
updated_at: Mapped[dt.datetime] = mapped_column(DateTime, nullable=False, default=utcnow, onupdate=utcnow)
|
||||
|
||||
__table_args__ = (
|
||||
Index("ix_display_modules_report_status_sort", "report_id", "status", "sort_order"),
|
||||
Index("ix_display_modules_artifact_status", "display_artifact_id", "status"),
|
||||
)
|
||||
|
||||
|
||||
class AudioAsset(Base):
|
||||
__tablename__ = "audio_assets"
|
||||
|
||||
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
||||
audio_id: Mapped[str] = mapped_column(String(64), unique=True, nullable=False)
|
||||
report_id: Mapped[str] = mapped_column(String(64), ForeignKey("reports.report_id"), nullable=False)
|
||||
source_raw_artifact_id: Mapped[str | None] = mapped_column(String(64), ForeignKey("raw_artifacts.raw_artifact_id"))
|
||||
title_cn: Mapped[str] = mapped_column(String(512), nullable=False)
|
||||
duration_sec: Mapped[int | None] = mapped_column(Integer)
|
||||
oss_key: Mapped[str | None] = mapped_column(String(512))
|
||||
waveform_ref: Mapped[str | None] = mapped_column(String(512))
|
||||
chapters: Mapped[str | None] = mapped_column(Text)
|
||||
status: Mapped[str] = mapped_column(String(16), nullable=False, default="missing")
|
||||
published_at: Mapped[dt.datetime | None] = mapped_column(DateTime)
|
||||
created_at: Mapped[dt.datetime] = mapped_column(DateTime, nullable=False, default=utcnow)
|
||||
updated_at: Mapped[dt.datetime] = mapped_column(DateTime, nullable=False, default=utcnow, onupdate=utcnow)
|
||||
|
||||
__table_args__ = (Index("ix_audio_report_status", "report_id", "status"),)
|
||||
|
||||
|
||||
class RelatedNews(Base):
|
||||
__tablename__ = "related_news"
|
||||
|
||||
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
||||
related_news_id: Mapped[str] = mapped_column(String(64), unique=True, nullable=False)
|
||||
report_id: Mapped[str] = mapped_column(String(64), ForeignKey("reports.report_id"), nullable=False)
|
||||
title: Mapped[str] = mapped_column(String(512), nullable=False)
|
||||
source_name: Mapped[str | None] = mapped_column(String(255))
|
||||
source_url: Mapped[str | None] = mapped_column(String(512))
|
||||
published_at: Mapped[dt.datetime | None] = mapped_column(DateTime)
|
||||
language: Mapped[str | None] = mapped_column(String(8))
|
||||
summary_cn: Mapped[str | None] = mapped_column(Text)
|
||||
match_method: Mapped[str] = mapped_column(String(32), nullable=False, default="manual_curated")
|
||||
match_keywords: Mapped[str | None] = mapped_column(Text)
|
||||
match_confidence: Mapped[str | None] = mapped_column(String(8))
|
||||
status: Mapped[str] = mapped_column(String(16), nullable=False, default="candidate")
|
||||
created_at: Mapped[dt.datetime] = mapped_column(DateTime, nullable=False, default=utcnow)
|
||||
|
||||
__table_args__ = (Index("ix_related_news_report_status", "report_id", "status"),)
|
||||
|
||||
|
||||
class User(Base):
|
||||
__tablename__ = "users"
|
||||
|
||||
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
||||
user_id: Mapped[str] = mapped_column(String(64), unique=True, nullable=False)
|
||||
phone_hash: Mapped[str | None] = mapped_column(String(128), unique=True)
|
||||
wechat_openid: Mapped[str | None] = mapped_column(String(128), unique=True)
|
||||
apple_user_id: Mapped[str | None] = mapped_column(String(256), unique=True)
|
||||
display_name: Mapped[str | None] = mapped_column(String(128))
|
||||
avatar_url: Mapped[str | None] = mapped_column(String(512))
|
||||
created_at: Mapped[dt.datetime] = mapped_column(DateTime, nullable=False, default=utcnow)
|
||||
last_login_at: Mapped[dt.datetime | None] = mapped_column(DateTime)
|
||||
status: Mapped[str] = mapped_column(String(16), nullable=False, default="active")
|
||||
|
||||
|
||||
class Favorite(Base):
|
||||
__tablename__ = "favorites"
|
||||
|
||||
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
||||
favorite_id: Mapped[str] = mapped_column(String(64), unique=True, nullable=False)
|
||||
user_id: Mapped[str] = mapped_column(String(64), ForeignKey("users.user_id"), nullable=False)
|
||||
report_id: Mapped[str] = mapped_column(String(64), ForeignKey("reports.report_id"), nullable=False)
|
||||
created_at: Mapped[dt.datetime] = mapped_column(DateTime, nullable=False, default=utcnow)
|
||||
status: Mapped[str] = mapped_column(String(16), nullable=False, default="active")
|
||||
|
||||
__table_args__ = (UniqueConstraint("user_id", "report_id", name="uq_favorites_user_report"), Index("ix_favorites_user_report", "user_id", "report_id"))
|
||||
|
||||
|
||||
class ReadingHistory(Base):
|
||||
__tablename__ = "reading_history"
|
||||
|
||||
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
||||
history_id: Mapped[str] = mapped_column(String(64), unique=True, nullable=False)
|
||||
user_id: Mapped[str] = mapped_column(String(64), ForeignKey("users.user_id"), nullable=False)
|
||||
report_id: Mapped[str] = mapped_column(String(64), ForeignKey("reports.report_id"), nullable=False)
|
||||
event_type: Mapped[str] = mapped_column(String(32), nullable=False, default="view_detail")
|
||||
last_position: Mapped[str | None] = mapped_column(Text)
|
||||
last_seen_at: Mapped[dt.datetime] = mapped_column(DateTime, nullable=False, default=utcnow)
|
||||
|
||||
__table_args__ = (Index("ix_reading_history_user_seen", "user_id", "last_seen_at"), Index("ix_reading_history_user_report", "user_id", "report_id"))
|
||||
|
||||
|
||||
class SavedListen(Base):
|
||||
__tablename__ = "saved_listens"
|
||||
|
||||
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
||||
saved_listen_id: Mapped[str] = mapped_column(String(64), unique=True, nullable=False)
|
||||
user_id: Mapped[str] = mapped_column(String(64), ForeignKey("users.user_id"), nullable=False)
|
||||
report_id: Mapped[str] = mapped_column(String(64), ForeignKey("reports.report_id"), nullable=False)
|
||||
audio_id: Mapped[str] = mapped_column(String(64), ForeignKey("audio_assets.audio_id"), nullable=False)
|
||||
created_at: Mapped[dt.datetime] = mapped_column(DateTime, nullable=False, default=utcnow)
|
||||
status: Mapped[str] = mapped_column(String(16), nullable=False, default="active")
|
||||
|
||||
__table_args__ = (UniqueConstraint("user_id", "audio_id", name="uq_saved_listens_user_audio"),)
|
||||
|
||||
|
||||
class PlaybackProgress(Base):
|
||||
__tablename__ = "playback_progress"
|
||||
|
||||
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
||||
progress_id: Mapped[str] = mapped_column(String(64), unique=True, nullable=False)
|
||||
user_id: Mapped[str] = mapped_column(String(64), ForeignKey("users.user_id"), nullable=False)
|
||||
audio_id: Mapped[str] = mapped_column(String(64), ForeignKey("audio_assets.audio_id"), nullable=False)
|
||||
report_id: Mapped[str] = mapped_column(String(64), ForeignKey("reports.report_id"), nullable=False)
|
||||
position_sec: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
|
||||
duration_sec: Mapped[int | None] = mapped_column(Integer)
|
||||
completed: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
|
||||
updated_at: Mapped[dt.datetime] = mapped_column(DateTime, nullable=False, default=utcnow, onupdate=utcnow)
|
||||
|
||||
__table_args__ = (UniqueConstraint("user_id", "audio_id", name="uq_playback_user_audio"), Index("ix_playback_user_audio", "user_id", "audio_id"))
|
||||
|
||||
|
||||
class OutboundEvent(Base):
|
||||
__tablename__ = "outbound_events"
|
||||
|
||||
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
||||
outbound_id: Mapped[str] = mapped_column(String(64), unique=True, nullable=False)
|
||||
click_id: Mapped[str] = mapped_column(String(64), unique=True, nullable=False)
|
||||
tracking_id: Mapped[str] = mapped_column(String(64), nullable=False)
|
||||
user_id: Mapped[str | None] = mapped_column(String(64), ForeignKey("users.user_id"))
|
||||
device_id: Mapped[str | None] = mapped_column(String(128))
|
||||
report_id: Mapped[str | None] = mapped_column(String(64), ForeignKey("reports.report_id"))
|
||||
institution_id: Mapped[str | None] = mapped_column(String(64))
|
||||
scene: Mapped[str | None] = mapped_column(String(64))
|
||||
ref: Mapped[str | None] = mapped_column(String(128))
|
||||
target: Mapped[str | None] = mapped_column(String(32))
|
||||
source_page: Mapped[str | None] = mapped_column(String(32))
|
||||
placement: Mapped[str | None] = mapped_column(String(64))
|
||||
campaign_id: Mapped[str | None] = mapped_column(String(64))
|
||||
target_app: Mapped[str | None] = mapped_column(String(64))
|
||||
commodity_tag: Mapped[str | None] = mapped_column(String(64))
|
||||
hook_type: Mapped[str | None] = mapped_column(String(64))
|
||||
user_state: Mapped[str | None] = mapped_column(String(16))
|
||||
ts: Mapped[int | None] = mapped_column(BigInteger)
|
||||
created_at: Mapped[dt.datetime] = mapped_column(DateTime, nullable=False, default=utcnow)
|
||||
|
||||
__table_args__ = (Index("ix_outbound_tracking", "tracking_id"), Index("ix_outbound_report_created", "report_id", "created_at"))
|
||||
|
||||
Reference in New Issue
Block a user