phase4: implement player search filters, htmx results, and detail page

apps/players/services/search.py

@@ -0,0 +1,149 @@
+from datetime import date, timedelta
+
+from django.db.models import Case, ExpressionWrapper, F, FloatField, Max, Q, Value, When
+from django.db.models.functions import Coalesce
+
+from apps.players.models import Player
+
+
+def _years_ago_today(years: int) -> date:
+    today = date.today()
+    try:
+        return today.replace(year=today.year - years)
+    except ValueError:
+        # Feb 29 -> Feb 28 when target year is not leap year.
+        return today.replace(month=2, day=28, year=today.year - years)
+
+
+def _apply_min_max_filter(queryset, min_key: str, max_key: str, field_name: str, data: dict):
+    min_value = data.get(min_key)
+    max_value = data.get(max_key)
+    if min_value is not None:
+        queryset = queryset.filter(**{f"{field_name}__gte": min_value})
+    if max_value is not None:
+        queryset = queryset.filter(**{f"{field_name}__lte": max_value})
+    return queryset
+
+
+def filter_players(queryset, data: dict):
+    query = data.get("q")
+    if query:
+        queryset = queryset.filter(Q(full_name__icontains=query) | Q(aliases__alias__icontains=query))
+
+    if data.get("nominal_position"):
+        queryset = queryset.filter(nominal_position=data["nominal_position"])
+    if data.get("inferred_role"):
+        queryset = queryset.filter(inferred_role=data["inferred_role"])
+    if data.get("nationality"):
+        queryset = queryset.filter(nationality=data["nationality"])
+
+    if data.get("team"):
+        queryset = queryset.filter(player_seasons__team=data["team"])
+    if data.get("competition"):
+        queryset = queryset.filter(player_seasons__competition=data["competition"])
+    if data.get("season"):
+        queryset = queryset.filter(player_seasons__season=data["season"])
+
+    queryset = _apply_min_max_filter(queryset, "height_min", "height_max", "height_cm", data)
+    queryset = _apply_min_max_filter(queryset, "weight_min", "weight_max", "weight_kg", data)
+
+    age_min = data.get("age_min")
+    age_max = data.get("age_max")
+    if age_min is not None:
+        queryset = queryset.filter(birth_date__lte=_years_ago_today(age_min))
+    if age_max is not None:
+        earliest_birth = _years_ago_today(age_max + 1) + timedelta(days=1)
+        queryset = queryset.filter(birth_date__gte=earliest_birth)
+
+    queryset = _apply_min_max_filter(
+        queryset,
+        "games_played_min",
+        "games_played_max",
+        "player_seasons__games_played",
+        data,
+    )
+
+    mpg_min = data.get("minutes_per_game_min")
+    mpg_max = data.get("minutes_per_game_max")
+    if mpg_min is not None:
+        queryset = queryset.filter(player_seasons__games_played__gt=0).filter(
+            player_seasons__minutes_played__gte=F("player_seasons__games_played") * mpg_min
+        )
+    if mpg_max is not None:
+        queryset = queryset.filter(player_seasons__games_played__gt=0).filter(
+            player_seasons__minutes_played__lte=F("player_seasons__games_played") * mpg_max
+        )
+
+    stat_pairs = (
+        ("points_per_game_min", "points_per_game_max", "player_seasons__stats__points"),
+        ("rebounds_per_game_min", "rebounds_per_game_max", "player_seasons__stats__rebounds"),
+        ("assists_per_game_min", "assists_per_game_max", "player_seasons__stats__assists"),
+        ("steals_per_game_min", "steals_per_game_max", "player_seasons__stats__steals"),
+        ("blocks_per_game_min", "blocks_per_game_max", "player_seasons__stats__blocks"),
+        ("turnovers_per_game_min", "turnovers_per_game_max", "player_seasons__stats__turnovers"),
+        ("fg_pct_min", "fg_pct_max", "player_seasons__stats__fg_pct"),
+        ("three_pct_min", "three_pct_max", "player_seasons__stats__three_pct"),
+        ("ft_pct_min", "ft_pct_max", "player_seasons__stats__ft_pct"),
+        (
+            "efficiency_metric_min",
+            "efficiency_metric_max",
+            "player_seasons__stats__player_efficiency_rating",
+        ),
+    )
+    for min_key, max_key, field_name in stat_pairs:
+        queryset = _apply_min_max_filter(queryset, min_key, max_key, field_name, data)
+
+    mpg_expression = Case(
+        When(
+            player_seasons__games_played__gt=0,
+            then=ExpressionWrapper(
+                F("player_seasons__minutes_played") * 1.0 / F("player_seasons__games_played"),
+                output_field=FloatField(),
+            ),
+        ),
+        default=Value(0.0),
+        output_field=FloatField(),
+    )
+
+    queryset = queryset.annotate(
+        games_played_value=Coalesce(Max("player_seasons__games_played"), Value(0)),
+        mpg_value=Coalesce(Max(mpg_expression), Value(0.0)),
+        ppg_value=Coalesce(Max("player_seasons__stats__points"), Value(0.0)),
+        rpg_value=Coalesce(Max("player_seasons__stats__rebounds"), Value(0.0)),
+        apg_value=Coalesce(Max("player_seasons__stats__assists"), Value(0.0)),
+        spg_value=Coalesce(Max("player_seasons__stats__steals"), Value(0.0)),
+        bpg_value=Coalesce(Max("player_seasons__stats__blocks"), Value(0.0)),
+        top_efficiency=Coalesce(Max("player_seasons__stats__player_efficiency_rating"), Value(0.0)),
+    )
+
+    return queryset.distinct()
+
+
+def apply_sorting(queryset, sort_key: str):
+    if sort_key == "name_desc":
+        return queryset.order_by("-full_name", "id")
+    if sort_key == "age_youngest":
+        return queryset.order_by(F("birth_date").desc(nulls_last=True), "full_name")
+    if sort_key == "age_oldest":
+        return queryset.order_by(F("birth_date").asc(nulls_last=True), "full_name")
+    if sort_key == "height_desc":
+        return queryset.order_by(F("height_cm").desc(nulls_last=True), "full_name")
+    if sort_key == "height_asc":
+        return queryset.order_by(F("height_cm").asc(nulls_last=True), "full_name")
+    if sort_key == "ppg_desc":
+        return queryset.order_by(F("ppg_value").desc(nulls_last=True), "full_name")
+    if sort_key == "ppg_asc":
+        return queryset.order_by(F("ppg_value").asc(nulls_last=True), "full_name")
+    if sort_key == "mpg_desc":
+        return queryset.order_by(F("mpg_value").desc(nulls_last=True), "full_name")
+    if sort_key == "mpg_asc":
+        return queryset.order_by(F("mpg_value").asc(nulls_last=True), "full_name")
+    return queryset.order_by("full_name", "id")
+
+
+def base_player_queryset():
+    return Player.objects.select_related(
+        "nationality",
+        "nominal_position",
+        "inferred_role",
+    ).prefetch_related("aliases")