earthjasonlin/mailova
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(models): resolve ambiguous foreign keys between User and InviteCode

Browse Source
This commit is contained in:
Zichao Lin
2026-07-06 08:39:33 +08:00
parent 755409f386
commit 501e27400b
2 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/models.py
View File

@@ -23,7 +23,7 @@ class User(Base):
profiles = relationship("Profile", back_populates="user", cascade="all, delete-orphan") profiles = relationship("Profile", back_populates="user", cascade="all, delete-orphan")
sessions = relationship("UserSession", back_populates="user", cascade="all, delete-orphan") sessions = relationship("UserSession", back_populates="user", cascade="all, delete-orphan")
invite_codes = relationship("InviteCode", back_populates="creator", cascade="all, delete-orphan") invite_codes = relationship("InviteCode", back_populates="creator", foreign_keys="InviteCode.created_by", cascade="all, delete-orphan")
invited_by_code = relationship("InviteCode", foreign_keys=[invited_by_code_id]) invited_by_code = relationship("InviteCode", foreign_keys=[invited_by_code_id])
@@ -38,7 +38,7 @@ class InviteCode(Base):
created_by = Column(Integer, ForeignKey("users.id", ondelete="SET NULL"), nullable=True) created_by = Column(Integer, ForeignKey("users.id", ondelete="SET NULL"), nullable=True)
created_at = Column(DateTime, default=_utcnow) created_at = Column(DateTime, default=_utcnow)
creator = relationship("User", back_populates="invite_codes") creator = relationship("User", back_populates="invite_codes", foreign_keys=[created_by])
registered_users = relationship("User", back_populates="invited_by_code", foreign_keys="User.invited_by_code_id") registered_users = relationship("User", back_populates="invited_by_code", foreign_keys="User.invited_by_code_id")

8
test_error.py Normal file
View File

@@ -0,0 +1,8 @@
from app.main import app
from fastapi.testclient import TestClient
c = TestClient(app)
import sqlite3
conn = sqlite3.connect('data.db')
cur = conn.execute('SELECT id, username, invited_by_code_id FROM users')
print(cur.fetchall())
conn.close()