This commit is contained in:
root 2025-03-04 06:24:40 +00:00
commit 474d827de6

@ -28,7 +28,8 @@ def get_friends(user_id, num_friends):
random.seed(0) random.seed(0)
if not curr_user: if not curr_user:
return {} return {}
for f in random.sample(curr_user[0]["friends"], num_friends): sample_size = min(num_friends, len(curr_user[0]["friends"]))
for f in random.sample(curr_user[0]["friends"], sample_size):
friends[f] = db.search(User.user_id == user_id)[0] friends[f] = db.search(User.user_id == user_id)[0]
return friends return friends