from __future__ import annotations
from typing import TYPE_CHECKING
from django.contrib.syndication.views import Feed
from django.urls import reverse
from django.utils.html import format_html
from twitch.models import DropCampaign, Game, Organization
if TYPE_CHECKING:
import datetime
class OrganizationFeed(Feed):
"""RSS feed for latest organizations."""
title = "TTVDrops Organizations"
link = "/organizations/"
description = "Latest organizations on TTVDrops"
def items(self) -> list[Organization]:
"""Return the latest 100 organizations."""
return list(Organization.objects.order_by("-id")[:100])
def item_title(self, item: Organization) -> str:
"""Return the organization name as the item title."""
return item.name
def item_description(self, item: Organization) -> str:
"""Return a description of the organization."""
return f"Organization {item.name}"
def item_link(self, item: Organization) -> str:
"""Return the link to the organization detail."""
return reverse("twitch:organization_detail", args=[item.pk])
class GameFeed(Feed):
"""RSS feed for latest games."""
title = "TTVDrops Games"
link = "/games/"
description = "Latest games on TTVDrops"
def items(self) -> list[Game]:
"""Return the latest 100 games."""
return list(Game.objects.order_by("-id")[:100])
def item_title(self, item: Game) -> str:
"""Return the game name as the item title."""
return str(item)
def item_description(self, item: Game) -> str:
"""Return a description of the game."""
return f"Game {item.display_name}"
def item_link(self, item: Game) -> str:
"""Return the link to the game detail."""
return reverse("twitch:game_detail", args=[item.pk])
class DropCampaignFeed(Feed):
"""RSS feed for latest drop campaigns."""
title = "Twitch Drop Campaigns"
link = "/campaigns/"
description = "Latest Twitch drop campaigns"
feed_url = "/rss/campaigns/"
feed_copyright = "Information wants to be free."
def items(self) -> list[DropCampaign]:
"""Return the latest 100 drop campaigns."""
return list(DropCampaign.objects.select_related("game").order_by("-added_at")[:100])
def item_title(self, item: DropCampaign) -> str:
"""Return the campaign name as the item title."""
return f"{item.game.display_name}: {item.clean_name}"
def item_description(self, item: DropCampaign) -> str:
"""Return a description of the campaign."""
description = ""
# Include the campaign image if available
if item.image_url:
description += format_html(
'
',
item.image_url,
item.name,
)
# Add the campaign description text
description += format_html("
{}
", item.description) if item.description else "" # Add start and end dates for clarity if item.start_at: description += f"Starts: {item.start_at.strftime('%Y-%m-%d %H:%M %Z')}
" if item.end_at: description += f"Ends: {item.end_at.strftime('%Y-%m-%d %H:%M %Z')}
" # Add information about the drops in this campaign drops = item.time_based_drops.select_related().prefetch_related("benefits").all() # type: ignore[attr-defined] if drops: description += "| Benefits | ' 'Drop Name | ' 'Requirements | ' 'Period | ' "
|---|---|---|---|
| '
for benefit in drop.benefits.all():
if benefit.image_asset_url:
description += format_html(
' | "
# Drop name
description += f'{drop.name} | ' # Requirements requirements = "" if drop.required_minutes_watched: requirements = f"{drop.required_minutes_watched} minutes watched" if drop.required_subs > 0: if requirements: requirements += f" and {drop.required_subs} subscriptions required" else: requirements = f"{drop.required_subs} subscriptions required" description += f'{requirements} | ' # Period period = "" if drop.start_at: period += drop.start_at.strftime("%Y-%m-%d %H:%M %Z") if drop.end_at: if period: period += " - " + drop.end_at.strftime("%Y-%m-%d %H:%M %Z") else: period = drop.end_at.strftime("%Y-%m-%d %H:%M %Z") description += f'{period} | ' description += "