密码登录，只允许配送员和商家登录。

app/api/endpoints/user.py

@@ -21,6 +21,7 @@ from sqlalchemy import text
 from app.models.community import CommunityDB
 from app.models.user_auth import UserAuthDB, UserAuthCreate, UserAuthInfo
 from app.core.qcloud import qcloud_manager
+from app.models.merchant import MerchantDB
 
 router = APIRouter()
 
@@ -258,7 +259,7 @@ async def password_login(
 ):
     """密码登录"""
     user = db.query(UserDB).filter(UserDB.phone == login_data.phone).first()
-    
+
     if not user:
         return error_response(code=401, message="用户不存在")
     
@@ -267,6 +268,18 @@ async def password_login(
     
     if not verify_password(login_data.password, user.password):
         return error_response(code=401, message="密码错误")
+
+    if user.roles not in [UserRole.DELIVERYMAN, UserRole.MERCHANT]:
+        return error_response(code=401, message="只有配送员和商家可以登录")
+    
+    if user.roles == UserRole.MERCHANT:
+        # 检查是否有商家设置了当前用户 id
+        merchant = db.query(MerchantDB).filter(MerchantDB.user_id == user.userid).first()
+        if not merchant:
+            return error_response(code=401, message="商家账户，请先关联商家")
+    
+    if user.roles == UserRole.DELIVERYMAN and not user.community_id:
+        return error_response(code=401, message="配送员账户，请先设置归属小区")
     
     # 生成访问令牌
     access_token = create_access_token(data={"phone": user.phone,"userid":user.userid})