Merge pull request #43 from cuappdev/advanced-filters

Added find by date query to support advanced filters for frontend

Author: TushigBili

src/queries/game_query.py

@@ -1,4 +1,4 @@
-from graphene import ObjectType, String, Field, List, Int
+from graphene import ObjectType, String, Field, List, Int, DateTime
 from src.services.game_service import GameService
 from src.types import GameType
 
@@ -27,6 +27,7 @@ class GameQuery(ObjectType):
     games_by_sport_gender = List(
         GameType, sport=String(required=True), gender=String(required=True)
     )
+    games_by_date = List(GameType, startDate=DateTime(required=True), endDate=DateTime(required=True))
 
     def resolve_games(self, info, limit=100, offset=0):
         """
@@ -67,3 +68,9 @@ def resolve_games_by_sport_gender(self, info, sport, gender):
         Resolver for retrieving all games by its sport and gender.
         """
         return GameService.get_games_by_sport_gender(sport, gender)
+    
+    def resolve_games_by_date(self, info, startDate, endDate):
+        """
+        Resolver for retrieving games by date.
+        """
+        return GameService.get_games_by_date(startDate, endDate)

src/repositories/game_repository.py

@@ -225,6 +225,27 @@ def find_games_by_sport_gender_after_date(sport, gender, after_date=None):
 
         games = game_collection.find(query)
         return [Game.from_dict(game) for game in games]
+    
+    @staticmethod
+    def find_by_date(startDate, endDate):
+        """
+        Retrieve all games from the 'game' collection in MongoDB for games
+        between certain dates. 
+        """
+        game_collection = db["game"]
+        
+        start_str = startDate.isoformat()
+        endDate = endDate.isoformat()
+        
+        query = {
+            "utc_date": {
+                "$gte": start_str,
+                "$lte": endDate
+            }
+        }
+        
+        games = game_collection.find(query)
+        return [Game.from_dict(game) for game in games]
 
     @staticmethod
     def delete_games_by_ids(game_ids):

src/services/game_service.py

@@ -101,6 +101,13 @@ def get_games_by_sport_gender(sport, gender):
         Retrieves all game by its sport and gender.
         """
         return GameRepository.find_by_sport_gender(sport, gender)
+    
+    @staticmethod
+    def get_games_by_date(startDate, endDate):
+        """
+        Retrieves all games between these two dates.
+        """
+        return GameRepository.find_by_date(startDate, endDate)
 
     @staticmethod
     def get_tournament_games_by_sport_gender(sport, gender, after_date=None):