From 727943faadaa6258c6b55290bafe5f9465253168 Mon Sep 17 00:00:00 2001 From: MingxuanGame Date: Mon, 28 Jul 2025 16:03:33 +0000 Subject: [PATCH] fix(relationship): fix add friends --- app/router/relationship.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/app/router/relationship.py b/app/router/relationship.py index 4c4bb99..9ed5b0f 100644 --- a/app/router/relationship.py +++ b/app/router/relationship.py @@ -70,7 +70,8 @@ async def add_relationship( type=relationship_type, ) db.add(relationship) - if relationship.type == RelationshipType.BLOCK: + origin_type = relationship.type + if origin_type == RelationshipType.BLOCK: target_relationship = ( await db.exec( select(Relationship).where( @@ -81,13 +82,14 @@ async def add_relationship( ).first() if target_relationship and target_relationship.type == RelationshipType.FOLLOW: await db.delete(target_relationship) + current_user_id = current_user.id await db.commit() - if relationship.type == RelationshipType.FOLLOW: + if origin_type == RelationshipType.FOLLOW: relationship = ( await db.exec( select(Relationship) .where( - Relationship.user_id == current_user.id, + Relationship.user_id == current_user_id, Relationship.target_id == target, ) .options(