Add WIP

core/views/index.py

@@ -1,6 +1,5 @@
 from __future__ import annotations
 
-import datetime
 import logging
 from typing import TYPE_CHECKING
 
@@ -9,17 +8,15 @@ from django.conf import settings
 from django.http import HttpRequest, HttpResponse
 from django.template.response import TemplateResponse
 
-from core.data import CampaignContext, DropContext, GameContext, WebhookData
+from core.data import WebhookData
 from twitch_app.models import (
-    DropBenefit,
-    DropCampaign,
-    Game,
-    TimeBasedDrop,
+    Organization,
 )
 
 if TYPE_CHECKING:
     from pathlib import Path
 
+    from django.db.models.manager import BaseManager
     from django.http import (
         HttpRequest,
         HttpResponse,
@@ -44,111 +41,12 @@ controller = hishel.Controller(
 )
 
 
-def fetch_campaigns(game: Game) -> list[CampaignContext]:
-    """Fetch active campaigns for a given game."""
-    campaigns: list[CampaignContext] = []
-    for campaign in DropCampaign.objects.filter(
-        game=game,
-        status="ACTIVE",
-        end_at__gt=datetime.datetime.now(tz=datetime.UTC),
-    ).only(
-        "id",
-        "name",
-        "image_url",
-        "status",
-        "account_link_url",
-        "description",
-        "details_url",
-        "start_at",
-        "end_at",
-    ):
-        drops = fetch_drops(campaign)
-        if not drops:
-            logger.info("No drops found for %s", campaign.name)
-            continue
-
-        campaigns.append(
-            CampaignContext(
-                drop_id=campaign.id,
-                name=campaign.name,
-                image_url=campaign.image_url,
-                status=campaign.status,
-                account_link_url=campaign.account_link_url,
-                description=campaign.description,
-                details_url=campaign.details_url,
-                start_at=campaign.start_at,
-                end_at=campaign.end_at,
-                drops=drops,
-            ),
-        )
-    return campaigns
-
-
-def fetch_drops(campaign: DropCampaign) -> list[DropContext]:
-    """Fetch drops for a given campaign."""
-    drops: list[DropContext] = []
-    drop: TimeBasedDrop
-    for drop in campaign.time_based_drops.all().only(
-        "id",
-        "name",
-        "required_minutes_watched",
-        "required_subs",
-    ):
-        benefit: DropBenefit | None = drop.benefits.first()
-
-        image_asset_url: str = "https://static-cdn.jtvnw.net/twitch-quests-assets/CAMPAIGN/default.png"
-        if benefit and benefit.image_asset_url:
-            image_asset_url = benefit.image_asset_url
-
-        drops.append(
-            DropContext(
-                drops_id=drop.id,
-                image_url=image_asset_url,
-                name=drop.name,
-                required_minutes_watched=drop.required_minutes_watched,
-                required_subs=drop.required_subs,
-            ),
-        )
-    return drops
-
-
-def sort_games_by_campaign_start(list_of_games: list[GameContext]) -> list[GameContext]:
-    """Sort games by the start date of the first campaign and reverse the list so the latest games are first."""
-    if list_of_games and list_of_games[0].campaigns:
-        list_of_games.sort(
-            key=lambda x: x.campaigns[0].start_at
-            if x.campaigns and x.campaigns[0].start_at is not None
-            else datetime.datetime.min,
-        )
-    list_of_games.reverse()
-    return list_of_games
-
-
 def get_webhooks(request: HttpRequest) -> list[str]:
     """Get the webhooks from the cookie."""
     cookie: str = request.COOKIES.get("webhooks", "")
     return list(filter(None, cookie.split(",")))
 
 
-def prepare_game_contexts() -> list[GameContext]:
-    """Prepare game contexts with their respective campaigns and drops."""
-    list_of_games: list[GameContext] = []
-    for game in list(Game.objects.all().only("id", "image_url", "display_name", "slug")):
-        campaigns: list[CampaignContext] = fetch_campaigns(game)
-        if not campaigns:
-            continue
-        list_of_games.append(
-            GameContext(
-                game_id=game.id,
-                campaigns=campaigns,
-                image_url=game.image_url,
-                display_name=game.display_name,
-                twitch_url=game.twitch_url,
-            ),
-        )
-    return list_of_games
-
-
 def get_avatar(webhook_response: Response) -> str:
     """Get the avatar URL from the webhook response."""
     avatar: str = "https://cdn.discordapp.com/embed/avatars/0.png"
@@ -173,12 +71,11 @@ def get_webhook_data(webhook: str) -> WebhookData:
 
 def index(request: HttpRequest) -> HttpResponse:
     """Render the index page."""
-    list_of_games: list[GameContext] = prepare_game_contexts()
-    sorted_list_of_games: list[GameContext] = sort_games_by_campaign_start(list_of_games)
+    orgs: BaseManager[Organization] = Organization.objects.all()
     webhooks: list[WebhookData] = [get_webhook_data(webhook) for webhook in get_webhooks(request)]
 
     return TemplateResponse(
         request=request,
         template="index.html",
-        context={"games": sorted_list_of_games, "webhooks": webhooks},
+        context={"orgs": orgs, "webhooks": webhooks},
     )