From 74e4b1cb530a67e8dcd85c29073562a5343719f0 Mon Sep 17 00:00:00 2001
From: MingxuanGame <MingxuanGame@outlook.com>
Date: Fri, 1 Aug 2025 04:27:44 +0000
Subject: [PATCH] fix(relationship): fix unique relationship

---
 app/database/relationship.py                  |  7 ++-
 ...02_relationship_fix_unique_relationship.py | 54 +++++++++++++++++++
 2 files changed, 59 insertions(+), 2 deletions(-)
 create mode 100644 migrations/versions/58a11441d302_relationship_fix_unique_relationship.py

diff --git a/app/database/relationship.py b/app/database/relationship.py
index 7a351aa..b941c28 100644
--- a/app/database/relationship.py
+++ b/app/database/relationship.py
@@ -22,12 +22,16 @@ class RelationshipType(str, Enum):
 
 class Relationship(SQLModel, table=True):
     __tablename__ = "relationship"  # pyright: ignore[reportAssignmentType]
+    id: int | None = Field(
+        default=None,
+        sa_column=Column(BigInteger, autoincrement=True, primary_key=True),
+        exclude=True,
+    )
     user_id: int = Field(
         default=None,
         sa_column=Column(
             BigInteger,
             ForeignKey("lazer_users.id"),
-            primary_key=True,
             index=True,
         ),
     )
@@ -36,7 +40,6 @@ class Relationship(SQLModel, table=True):
         sa_column=Column(
             BigInteger,
             ForeignKey("lazer_users.id"),
-            primary_key=True,
             index=True,
         ),
     )
diff --git a/migrations/versions/58a11441d302_relationship_fix_unique_relationship.py b/migrations/versions/58a11441d302_relationship_fix_unique_relationship.py
new file mode 100644
index 0000000..e383621
--- /dev/null
+++ b/migrations/versions/58a11441d302_relationship_fix_unique_relationship.py
@@ -0,0 +1,54 @@
+"""relationship: fix unique relationship
+
+Revision ID: 58a11441d302
+Revises: 1178d0758ebf
+Create Date: 2025-08-01 04:23:02.498166
+
+"""
+
+from __future__ import annotations
+
+from collections.abc import Sequence
+
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import mysql
+
+# revision identifiers, used by Alembic.
+revision: str = "58a11441d302"
+down_revision: str | Sequence[str] | None = "1178d0758ebf"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    """Upgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column(
+        "relationship",
+        sa.Column("id", sa.BigInteger(), autoincrement=True, nullable=False),
+    )
+    op.drop_constraint("PRIMARY", "relationship", type_="primary")
+    op.create_primary_key("pk_relationship", "relationship", ["id"])
+    op.alter_column(
+        "relationship", "user_id", existing_type=mysql.BIGINT(), nullable=True
+    )
+    op.alter_column(
+        "relationship", "target_id", existing_type=mysql.BIGINT(), nullable=True
+    )
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade schema."""
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint("pk_relationship", "relationship", type_="primary")
+    op.create_primary_key("PRIMARY", "relationship", ["user_id", "target_id"])
+    op.alter_column(
+        "relationship", "target_id", existing_type=mysql.BIGINT(), nullable=False
+    )
+    op.alter_column(
+        "relationship", "user_id", existing_type=mysql.BIGINT(), nullable=False
+    )
+    op.drop_column("relationship", "id")
+    # ### end Alembic commands ###