From d457c07f40623401081b3e5bc89117d90d3a644b Mon Sep 17 00:00:00 2001
From: Zayden Jung <zayden.jung@entroia.com>
Date: Mon, 18 May 2026 22:36:23 +0800
Subject: [PATCH] table follow requires the id length of user

---
 app/models/base.py   |  7 ++++++-
 app/models/follow.py | 19 +++++++++++++++++--
 2 files changed, 23 insertions(+), 3 deletions(-)

diff --git a/app/models/base.py b/app/models/base.py
index d825b9c..f01b555 100644
--- a/app/models/base.py
+++ b/app/models/base.py
@@ -60,6 +60,11 @@ class AutoID:
     _id_prefix_len = 3   # 前缀长度
     _id_payload_len = 22 # 主体长度（ULID 转 Base62 最大约 22 位）
     
+    @classmethod
+    def id_length(cls) -> int:
+        """返回当前类 ID 的总长度"""
+        return cls._id_prefix_len + cls._id_payload_len
+    
     @declared_attr
     def id(cls):
         # 运行时动态获取子类的 __tablename__
@@ -74,4 +79,4 @@ class AutoID:
             payload = gen_convex_payload(length=s_len)
             return f"{prefix}{payload}"
 
-        return Column(String(p_len + s_len), primary_key=True, default=id_factory)
\ No newline at end of file
+        return Column(String(cls.id_length()), primary_key=True, default=id_factory)
\ No newline at end of file
diff --git a/app/models/follow.py b/app/models/follow.py
index df02498..c9daeb4 100644
--- a/app/models/follow.py
+++ b/app/models/follow.py
@@ -1,13 +1,28 @@
+from typing import TYPE_CHECKING
 from sqlalchemy import Column, DateTime, String, ForeignKey, UniqueConstraint
 from sqlalchemy.sql import func
 from .base import AutoID, Base
 
+# 只有在静态检查时才导入
+# 运行阶段这段代码会被跳过，彻底避免循环导入
+# 不确定是否用得上，但感觉这是个好习惯，尤其在模型之间有外键关系时
+if TYPE_CHECKING:
+    from .user import User
+
 class Follow(Base, AutoID):
     """用户关注关系表"""
     __tablename__ = "follows"
 
-    follower_id = Column(String(32), ForeignKey("users.id", ondelete="CASCADE"), index=True)
-    followee_id = Column(String(32), ForeignKey("users.id", ondelete="CASCADE"), index=True)
+    follower_id = Column(
+        String(User.id_length()), 
+        ForeignKey("users.id", ondelete="CASCADE"), 
+        index=True
+    )
+    followee_id = Column(
+        String(User.id_length()), 
+        ForeignKey("users.id", ondelete="CASCADE"), 
+        index=True
+    )
     
     created_at = Column(DateTime(timezone=True), server_default=func.now())