You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
77 lines
2.7 KiB
Python
77 lines
2.7 KiB
Python
"""
|
|
Overlay Registry models - lookup tables for entity resolution
|
|
"""
|
|
|
|
from sqlalchemy import Column, String, Float, Boolean, Index, UniqueConstraint, JSON
|
|
from sqlalchemy.dialects.postgresql import UUID, TIMESTAMP
|
|
from datetime import datetime, timezone
|
|
import uuid
|
|
|
|
from app.core.database import Base
|
|
|
|
|
|
class CompanyAlias(Base):
|
|
__tablename__ = "company_aliases"
|
|
|
|
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
|
symbol = Column(String(10), nullable=False, index=True)
|
|
alias_type = Column(String(20), nullable=False) # canonical/short/alias/wiki
|
|
alias_value = Column(String(255), nullable=False)
|
|
confidence = Column(Float, default=1.0)
|
|
active = Column(Boolean, default=True)
|
|
created_at = Column(
|
|
TIMESTAMP(timezone=True), default=lambda: datetime.now(timezone.utc)
|
|
)
|
|
|
|
__table_args__ = (
|
|
Index("idx_company_aliases_symbol", "symbol"),
|
|
Index("idx_company_aliases_value", "alias_value"),
|
|
)
|
|
|
|
|
|
class YouTubeChannelRegistry(Base):
|
|
__tablename__ = "youtube_channel_registry"
|
|
|
|
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
|
channel_id = Column(String(100), unique=True, nullable=False)
|
|
channel_title = Column(String(255), nullable=False)
|
|
category = Column(String(50), nullable=True)
|
|
channel_weight = Column(Float, default=0.5) # 0~1
|
|
active = Column(Boolean, default=True)
|
|
watch_mode = Column(String(20), default="recent") # recent / all
|
|
symbol_focus_tags = Column(JSON, default=list)
|
|
created_at = Column(
|
|
TIMESTAMP(timezone=True), default=lambda: datetime.now(timezone.utc)
|
|
)
|
|
|
|
|
|
class WikiPageMap(Base):
|
|
__tablename__ = "wiki_page_map"
|
|
|
|
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
|
symbol = Column(String(10), nullable=False)
|
|
wiki_page_title = Column(String(500), nullable=False)
|
|
confidence = Column(Float, default=1.0)
|
|
active = Column(Boolean, default=True)
|
|
created_at = Column(
|
|
TIMESTAMP(timezone=True), default=lambda: datetime.now(timezone.utc)
|
|
)
|
|
|
|
__table_args__ = (
|
|
UniqueConstraint("symbol", "wiki_page_title", name="uq_wiki_page_map"),
|
|
Index("idx_wiki_page_map_symbol", "symbol"),
|
|
)
|
|
|
|
|
|
class ThemeTopicMap(Base):
|
|
__tablename__ = "theme_topic_map"
|
|
|
|
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
|
topic_id = Column(String(100), nullable=False, unique=True)
|
|
topic_label = Column(String(255), nullable=False)
|
|
mapped_symbols = Column(JSON, default=list) # list of ticker strings
|
|
active = Column(Boolean, default=True)
|
|
created_at = Column(
|
|
TIMESTAMP(timezone=True), default=lambda: datetime.now(timezone.utc)
|
|
)
|