117 lines
3.5 KiB
Python
117 lines
3.5 KiB
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
from dataclasses import dataclass
|
|
from typing import TYPE_CHECKING
|
|
|
|
from django.db.models import Prefetch
|
|
from django.db.models.manager import BaseManager
|
|
from django.template.response import TemplateResponse
|
|
from django.utils import timezone # type: ignore # noqa: PGH003
|
|
|
|
from core.models import DropCampaign, Game, RewardCampaign
|
|
|
|
if TYPE_CHECKING:
|
|
from django.db.models.manager import BaseManager
|
|
from django.http import HttpRequest, HttpResponse
|
|
|
|
logger: logging.Logger = logging.getLogger(__name__)
|
|
|
|
|
|
@dataclass
|
|
class TOCItem:
|
|
"""Table of contents item."""
|
|
|
|
name: str
|
|
toc_id: str
|
|
|
|
|
|
def build_toc(list_of_things: list[TOCItem]) -> str:
|
|
"""Build the table of contents.
|
|
|
|
Args:
|
|
list_of_things (list[TOCItem]): The list of table of contents items.
|
|
|
|
Returns:
|
|
str: The HTML for the table of contents.
|
|
"""
|
|
html: str = """
|
|
<div class="position-sticky d-none d-lg-block toc">
|
|
<div class="card">
|
|
<div class="card-body">
|
|
<div id="toc-list" class="list-group">
|
|
"""
|
|
|
|
for item in list_of_things:
|
|
html += (
|
|
f'<a class="list-group-item list-group-item-action plain-text-item" href="#{item.toc_id}">{item.name}</a>'
|
|
)
|
|
html += """</div></div></div></div>"""
|
|
return html
|
|
|
|
|
|
def index(request: HttpRequest) -> HttpResponse:
|
|
"""Render the index page.
|
|
|
|
Args:
|
|
request (HttpRequest): The request object.
|
|
|
|
Returns:
|
|
HttpResponse: The response object
|
|
"""
|
|
reward_campaigns: BaseManager[RewardCampaign] = (
|
|
RewardCampaign.objects.all()
|
|
.prefetch_related("rewards")
|
|
.filter(ends_at__gt=timezone.now(), starts_at__lt=timezone.now())
|
|
)
|
|
future_campaigns: BaseManager[DropCampaign] = DropCampaign.objects.filter(
|
|
ends_at__gt=timezone.now(),
|
|
starts_at__lt=timezone.now(),
|
|
)
|
|
|
|
games: BaseManager[Game] = Game.objects.all().prefetch_related(
|
|
Prefetch("drop_campaigns", queryset=future_campaigns.prefetch_related("drops__benefits")),
|
|
)
|
|
tocs: list[TOCItem] = []
|
|
for game in games.all():
|
|
game_name: str = game.name or "<div class='text-muted'>Game name unknown</div>"
|
|
tocs.append(TOCItem(name=game_name, toc_id=f"#{game.twitch_id}"))
|
|
|
|
toc: str = build_toc(tocs)
|
|
|
|
context: dict[str, BaseManager[RewardCampaign] | str | BaseManager[Game]] = {
|
|
"reward_campaigns": reward_campaigns,
|
|
"games": games,
|
|
"toc": toc,
|
|
}
|
|
return TemplateResponse(request=request, template="index.html", context=context)
|
|
|
|
|
|
def game_view(request: HttpRequest) -> HttpResponse:
|
|
"""Render the game view page.
|
|
|
|
Args:
|
|
request (HttpRequest): The request object.
|
|
|
|
Returns:
|
|
HttpResponse: The response object.
|
|
"""
|
|
games: BaseManager[Game] = Game.objects.all()
|
|
|
|
context: dict[str, BaseManager[Game] | str] = {"games": games}
|
|
return TemplateResponse(request=request, template="games.html", context=context)
|
|
|
|
|
|
def reward_campaign_view(request: HttpRequest) -> HttpResponse:
|
|
"""Render the reward campaign view page.
|
|
|
|
Args:
|
|
request (HttpRequest): The request object.
|
|
|
|
Returns:
|
|
HttpResponse: The response object.
|
|
"""
|
|
reward_campaigns: BaseManager[RewardCampaign] = RewardCampaign.objects.all()
|
|
context: dict[str, BaseManager[RewardCampaign]] = {"reward_campaigns": reward_campaigns}
|
|
return TemplateResponse(request=request, template="reward_campaigns.html", context=context)
|