Colin Powell 3 роки тому
батько
коміт
3e633ffcd2
2 змінених файлів з 6 додано та 3 видалено
  1. 3 0
      games/api/views.py
  2. 3 3
      games/utils.py

+ 3 - 0
games/api/views.py

@@ -6,6 +6,7 @@ from games.api.serializers import (
     PublisherSerializer,
     GameCollectionSerializer,
 )
+from django_filters.rest_framework import DjangoFilterBackend
 from games.models import Developer, Game, GameSystem, Genre, Publisher, GameCollection
 from rest_framework import viewsets
 
@@ -13,6 +14,8 @@ from rest_framework import viewsets
 class GameViewSet(viewsets.ModelViewSet):
     queryset = Game.objects.all()
     serializer_class = GameSerializer
+    filter_backends = [DjangoFilterBackend]
+    filterset_fields = ['rating', 'english_patched', "undub", "kid_game","region", "hack"]
 
 
 class PublisherViewSet(viewsets.ModelViewSet):

+ 3 - 3
games/utils.py

@@ -79,9 +79,9 @@ def import_gamelist_file_to_db_for_system(
         if any(eu in game_path for eu in REGION_KEYWORDS["EU"]):
             region = Game.Region.EU.name
 
-        english_patched = game_path.lower() in ENGLISH_PATCHED_KEYWORDS
-        undub = game_path.lower() in UNDUB_KEYWORDS
-        hack = game_path.lower() in HACK_KEYWORDS
+        english_patched = any(key in game_path.lower() for key in ENGLISH_PATCHED_KEYWORDS)
+        undub = any(key in game_path.lower() for key in UNDUB_KEYWORDS)
+        hack = any(key in game_path.lower() for key in HACK_KEYWORDS)
 
         release_date_str = game.find("releasedate").text
         developer_str = game.find("developer").text