Skip to content

Commit b63377b

Browse files
committed
유사 친구 - books 삭제, 이름 추가
1 parent 130521b commit b63377b

File tree

2 files changed

+4
-20
lines changed

2 files changed

+4
-20
lines changed

app/BE/models/user_similarity.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ class SimilarUser(BaseModel):
3535
"""유사 사용자 정보를 반환하기 위한 모델"""
3636
user_id: str
3737
similarity_score: float
38-
books: List[Book] # 책 정보 리스트
38+
name: str
3939

4040
class Config:
4141
json_encoders = {

app/BE/services/similarity_service.py

+3-19
Original file line numberDiff line numberDiff line change
@@ -50,31 +50,15 @@ async def get_similar_users(user_id: str, limit: int = 5) -> List[SimilarUser]:
5050
user = await db.users.find_one({"_id": ObjectId(other_id)})
5151
if not user:
5252
continue
53+
54+
user_name = user.get("name", "Unknown")
5355

54-
# 책 정보 조회
55-
books = []
56-
for book_id in user.get("book", []):
57-
book_doc = await book_db.books.find_one({"_id": book_id})
58-
if book_doc:
59-
try:
60-
book = Book(
61-
_id=str(book_doc["_id"]),
62-
title=book_doc.get("book_name", "Unknown"), # book_name 필드 사용
63-
author=book_doc.get("author", "Unknown"),
64-
cover_image=book_doc.get("cover_image"),
65-
description=book_doc.get("description")
66-
)
67-
books.append(book)
68-
except Exception as e:
69-
print(f"Error creating Book model: {str(e)}")
70-
continue
71-
7256
# SimilarUser 객체 생성
7357
try:
7458
similar_user = SimilarUser(
7559
user_id=str(user["_id"]),
7660
similarity_score=similarity_score,
77-
books=books
61+
name=user_name
7862
)
7963
similar_users.append(similar_user)
8064
except Exception as e:

0 commit comments

Comments
 (0)