Skip to content

Commit ce7829e

Browse files
committed
Added ignoring bots for bot recommendations
1 parent 483c703 commit ce7829e

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

backend/PyMatcha/utils/bot_actions.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -39,10 +39,10 @@ def _prepare_chatbot(bot_name):
3939
return chatbot
4040

4141

42-
def _get_recommendations(bot_user: User):
42+
def _get_recommendations(bot_user: User, ignore_bots: bool):
4343
recommendations = redis.get(f"user_recommendations:{str(bot_user.id)}")
4444
if not recommendations:
45-
create_user_recommendations(bot_user)
45+
create_user_recommendations(bot_user, ignore_bots)
4646
recommendations = redis.get(f"user_recommendations:{str(bot_user.id)}")
4747
if not recommendations:
4848
raise ValueError("Recommendations could not be calculated")
@@ -136,7 +136,7 @@ def _botaction_send_message_over_old_one(bot_user: User, chatbot):
136136

137137

138138
def decide_bot_action(bot_user: User):
139-
recommendations = _get_recommendations(bot_user)
139+
recommendations = _get_recommendations(bot_user, ignore_bots=True)
140140

141141
# The bot will first view 0 to 10 profiles
142142
for _ in range(0, randrange(0, 10)):

backend/PyMatcha/utils/recommendations.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ def default_date_converter(o):
1313
return o.__str__()
1414

1515

16-
def create_user_recommendations(user_to_update):
16+
def create_user_recommendations(user_to_update, ignore_bots: bool = False):
1717
today = datetime.datetime.utcnow()
1818
user_to_update_recommendations = []
1919
if not user_to_update.birthdate:
@@ -38,6 +38,8 @@ def create_user_recommendations(user_to_update):
3838
blocked_ids = [u.blocked_id for u in user_to_update.get_blocks()]
3939

4040
for user in query:
41+
if user.is_bot and ignore_bots:
42+
continue
4143
if user.id == user_to_update.id:
4244
continue
4345
if user.id in matches_id or user.id in likes_sent_user_ids:

0 commit comments

Comments
 (0)