Merge branch 'master' into claire/basketballScoreBreakdown

Author: claiireyu

src/services/game_service.py

@@ -2,6 +2,7 @@
 from src.models.game import Game
 from src.services.team_service import TeamService
 from src.utils.helpers import is_tournament_placeholder_team
+from pymongo.errors import DuplicateKeyError
 
 
 class GameService:
@@ -36,8 +37,11 @@ def create_game(data):
             raise ValueError(f"Opponent team with id {opponent_id} does not exist.")
 
         game = Game(**data)
-        GameRepository.insert(game)
-        return game
+        try:
+            GameRepository.insert(game)
+            return game
+        except DuplicateKeyError:
+            return None
 
     @staticmethod
     def delete_game(game_id):

src/types.py

@@ -156,6 +156,7 @@ class YoutubeVideoType(ObjectType):
         - url: The URL to the video.
         - published_at: The date and time the video was published.
         - duration: The duration of the video (optional).
+        - sportsType: The sport type extracted from the video title.
     """
     id = String(required=False)
     title = String(required=True)
@@ -165,11 +166,19 @@ class YoutubeVideoType(ObjectType):
     url = String(required=True)
     published_at = String(required=True)
     duration = String(required=False)
+    sportsType = String(required=False)
 
     def __init__(self, **kwargs):
         for key, value in kwargs.items():
             setattr(self, key, value)
 
+    def resolve_sportsType(video, info):
+        """
+        Resolver to extract sport type from the video title.
+        """
+        from src.utils.helpers import extract_sport_from_title
+        return extract_sport_from_title(video.title)
+
 class ArticleType(ObjectType):
     """
     A GraphQL type representing a news article.

src/utils/helpers.py

@@ -3,6 +3,7 @@
 from PIL import Image
 from io import BytesIO
 from collections import Counter
+import re
 
 
 def get_dominant_color(image_url, white_threshold=200, black_threshold=50):
@@ -96,6 +97,53 @@ def is_cornell_loss(result: str):
     loss_indicators = ["L", "Loss", "loss", "Defeated", "defeated"]
     return any(indicator in result for indicator in loss_indicators)
 
+def extract_sport_from_title(title):
+    """
+    Extracts the sport type from a YouTube video title.
+    
+    Args:
+        title (str): The title of the YouTube video
+        
+    Returns:
+        str: The sport type if found, None otherwise
+    """
+    if not title:
+        return None
+    
+    title_lower = title.lower()
+    
+    sport_patterns = [
+        # Ice Hockey
+        (r"ice\s+hockey", "Ice Hockey"),
+        (r"women'?s\s+ice\s+hockey", "Ice Hockey"),
+        (r"men'?s\s+ice\s+hockey", "Ice Hockey"),
+        # Field Hockey
+        (r"field\s+hockey", "Field Hockey"),
+        # Hockey
+        (r"\bhockey\b", "Ice Hockey"),
+        # Basketball
+        (r"basketball", "Basketball"),
+        # Football
+        (r"\bfootball\b", "Football"),
+        # Soccer
+        (r"\bsoccer\b", "Soccer"),
+        # Volleyball
+        (r"volleyball", "Volleyball"),
+        # Wrestling
+        (r"wrestling", "Wrestling"),
+        # Sprint Football
+        (r"sprint\s+football", "Sprint Football"),
+    ]
+    
+    for pattern, sport_name in sport_patterns:
+        if re.search(pattern, title_lower):
+            return sport_name
+    
+    if "ice" in title_lower and ("hockey" in title_lower or "cornell" in title_lower):
+        return "Ice Hockey"
+    
+    return None
+
 def extract_sport_type_from_title(title: str):
     """
     Extract the sport type from an article title by matching against known sports.