From 25900a991112ed787bae1219c30cbc11c21b45d0 Mon Sep 17 00:00:00 2001 From: Colin Powell Date: Wed, 29 Jan 2025 00:23:46 -0500 Subject: [PATCH] [boardgames] Send notifications on chess imports --- vrobbler/apps/boardgames/sources/lichess.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/vrobbler/apps/boardgames/sources/lichess.py b/vrobbler/apps/boardgames/sources/lichess.py index da06995..d388d28 100644 --- a/vrobbler/apps/boardgames/sources/lichess.py +++ b/vrobbler/apps/boardgames/sources/lichess.py @@ -1,9 +1,9 @@ import berserk -from django.conf import settings - from boardgames.models import BoardGame -from scrobbles.models import Scrobble +from django.conf import settings from django.contrib.auth import get_user_model +from scrobbles.models import Scrobble +from scrobbles.utils import send_notifications_for_scrobble User = get_user_model() @@ -40,9 +40,6 @@ def import_chess_games_for_all_users(): "players": [], } - chess_status = game_dict.get("status") - chess_source = game_dict.get("source") - winner = game_dict.get("winner") black_player = game_dict.get("players", {}).get("black", {}) white_player = game_dict.get("players", {}).get("white", {}) @@ -117,5 +114,7 @@ def import_chess_games_for_all_users(): scrobbles_to_create.append(Scrobble(**scrobble_dict)) if scrobbles_to_create: - Scrobble.objects.bulk_create(scrobbles_to_create) + created = Scrobble.objects.bulk_create(scrobbles_to_create) + for scrobble in created: + send_notifications_for_scrobble(scrobble.id) return scrobbles_to_create