antialias/haunt-fm
1
0
Fork 0
Code Issues 4 Pull Requests Actions Packages Projects Releases Wiki Activity

Add ForeignKey declarations to SQLAlchemy model columns

Browse Source 
SQLAlchemy relationships require ForeignKey on the column definitions,
not just in the migration. Without them, mapper initialization fails.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Thomas Hallock
2026-02-22 09:05:50 -06:00
parent f8c13daa11
commit 92a75d6432
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/haunt_fm/models/track.py
View File

@@ -1,7 +1,7 @@
from datetime import datetime from datetime import datetime
from pgvector.sqlalchemy import Vector from pgvector.sqlalchemy import Vector
from sqlalchemy import REAL, BigInteger, DateTime, Index, Integer, Text, func from sqlalchemy import REAL, BigInteger, DateTime, ForeignKey, Index, Integer, Text, func
from sqlalchemy.dialects.postgresql import JSONB from sqlalchemy.dialects.postgresql import JSONB
from sqlalchemy.orm import Mapped, mapped_column, relationship from sqlalchemy.orm import Mapped, mapped_column, relationship
@@ -37,7 +37,7 @@ class ListenEvent(Base):
__tablename__ = "listen_events" __tablename__ = "listen_events"
id: Mapped[int] = mapped_column(BigInteger, primary_key=True) id: Mapped[int] = mapped_column(BigInteger, primary_key=True)
track_id: Mapped[int] = mapped_column(BigInteger, nullable=False) track_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("tracks.id"), nullable=False)
source: Mapped[str] = mapped_column(Text, nullable=False, default="music_assistant") source: Mapped[str] = mapped_column(Text, nullable=False, default="music_assistant")
speaker_name: Mapped[str | None] = mapped_column(Text) speaker_name: Mapped[str | None] = mapped_column(Text)
listened_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now()) listened_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())
@@ -51,7 +51,7 @@ class TrackEmbedding(Base):
__tablename__ = "track_embeddings" __tablename__ = "track_embeddings"
id: Mapped[int] = mapped_column(BigInteger, primary_key=True) id: Mapped[int] = mapped_column(BigInteger, primary_key=True)
track_id: Mapped[int] = mapped_column(BigInteger, unique=True, nullable=False) track_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("tracks.id"), unique=True, nullable=False)
embedding = mapped_column(Vector(512), nullable=False) embedding = mapped_column(Vector(512), nullable=False)
model_version: Mapped[str] = mapped_column(Text, nullable=False, default="laion/larger_clap_music") model_version: Mapped[str] = mapped_column(Text, nullable=False, default="laion/larger_clap_music")
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now()) created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())
@@ -67,8 +67,8 @@ class SimilarityLink(Base):
__tablename__ = "similarity_links" __tablename__ = "similarity_links"
id: Mapped[int] = mapped_column(BigInteger, primary_key=True) id: Mapped[int] = mapped_column(BigInteger, primary_key=True)
source_track_id: Mapped[int] = mapped_column(BigInteger, nullable=False) source_track_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("tracks.id"), nullable=False)
target_track_id: Mapped[int] = mapped_column(BigInteger, nullable=False) target_track_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("tracks.id"), nullable=False)
lastfm_match: Mapped[float | None] = mapped_column(REAL) lastfm_match: Mapped[float | None] = mapped_column(REAL)
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now()) created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())
@@ -103,8 +103,8 @@ class PlaylistTrack(Base):
__tablename__ = "playlist_tracks" __tablename__ = "playlist_tracks"
id: Mapped[int] = mapped_column(BigInteger, primary_key=True) id: Mapped[int] = mapped_column(BigInteger, primary_key=True)
playlist_id: Mapped[int] = mapped_column(BigInteger, nullable=False) playlist_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("playlists.id", ondelete="CASCADE"), nullable=False)
track_id: Mapped[int] = mapped_column(BigInteger, nullable=False) track_id: Mapped[int] = mapped_column(BigInteger, ForeignKey("tracks.id"), nullable=False)
position: Mapped[int] = mapped_column(Integer, nullable=False) position: Mapped[int] = mapped_column(Integer, nullable=False)
is_known: Mapped[bool] = mapped_column(nullable=False) is_known: Mapped[bool] = mapped_column(nullable=False)
similarity_score: Mapped[float | None] = mapped_column(REAL) similarity_score: Mapped[float | None] = mapped_column(REAL)