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