瀏覽代碼

Fix looking up games by hltb only

Colin Powell 1 年之前
父節點
當前提交
39add993e0
共有 1 個文件被更改,包括 10 次插入3 次删除
  1. 10 3
      vrobbler/apps/videogames/utils.py

+ 10 - 3
vrobbler/apps/videogames/utils.py

@@ -45,9 +45,16 @@ def get_or_create_videogame(
     if "genres" in game_dict.keys():
         genres = game_dict.pop("genres")
 
-    game, game_created = VideoGame.objects.get_or_create(
-        hltb_id=game_dict.get("hltb_id")
-    )
+    hltb_id = game_dict.get("hltb_id")
+    igdb_id = game_dict.get("igdb_id")
+    if hltb_id:
+        game, game_created = VideoGame.objects.get_or_create(hltb_id=hltb_id)
+    elif igdb_id:
+        game, game_created = VideoGame.objects.get_or_create(igdb_id=igdb_id)
+    else:
+        game, game_created = VideoGame.objects.get_or_create(
+            title=game_dict.get("title")
+        )
 
     if game_created or force_update:
         VideoGame.objects.filter(pk=game.id).update(**game_dict)