Joakim Hellsén
8f4e851fb9
- Deleted all test files in accounts and twitch apps to clean up the codebase. - Updated the DropCampaign, Game, Organization, DropBenefit, TimeBasedDrop, and DropBenefitEdge models to include database indexing for improved query performance. - Modified the DropCampaignListView and GameDetailView to remove unnecessary status filtering and streamline campaign retrieval logic. - Enhanced the campaign detail template to properly format campaign descriptions. - Adjusted the import_drop_campaign management command to increase default worker and batch sizes for improved performance. - Cleaned up the admin configuration for DropCampaign and TimeBasedDrop models.
197 lines
7 KiB
Python
197 lines
7 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import ClassVar
|
|
|
|
from django.db import models
|
|
from django.utils import timezone
|
|
|
|
|
|
class Game(models.Model):
|
|
"""Represents a game on Twitch."""
|
|
|
|
id = models.TextField(primary_key=True)
|
|
slug = models.TextField(blank=True, default="", db_index=True)
|
|
display_name = models.TextField(db_index=True)
|
|
|
|
class Meta:
|
|
indexes: ClassVar[list] = [
|
|
models.Index(fields=["slug"]),
|
|
models.Index(fields=["display_name"]),
|
|
]
|
|
|
|
def __str__(self) -> str:
|
|
"""Return a string representation of the game."""
|
|
return self.display_name
|
|
|
|
|
|
class Organization(models.Model):
|
|
"""Represents an organization on Twitch that can own drop campaigns."""
|
|
|
|
id = models.TextField(primary_key=True)
|
|
name = models.TextField(db_index=True)
|
|
|
|
class Meta:
|
|
indexes: ClassVar[list] = [
|
|
models.Index(fields=["name"]),
|
|
]
|
|
|
|
def __str__(self) -> str:
|
|
"""Return a string representation of the organization."""
|
|
return self.name
|
|
|
|
|
|
class DropCampaign(models.Model):
|
|
"""Represents a Twitch drop campaign."""
|
|
|
|
id = models.TextField(primary_key=True)
|
|
name = models.TextField(db_index=True)
|
|
description = models.TextField(blank=True)
|
|
details_url = models.URLField(max_length=500, blank=True, default="")
|
|
account_link_url = models.URLField(max_length=500, blank=True, default="")
|
|
image_url = models.URLField(max_length=500, blank=True, default="")
|
|
start_at = models.DateTimeField(db_index=True)
|
|
end_at = models.DateTimeField(db_index=True)
|
|
is_account_connected = models.BooleanField(default=False)
|
|
|
|
# Foreign keys
|
|
game = models.ForeignKey(Game, on_delete=models.CASCADE, related_name="drop_campaigns", db_index=True)
|
|
owner = models.ForeignKey(Organization, on_delete=models.CASCADE, related_name="drop_campaigns", db_index=True)
|
|
|
|
# Tracking fields
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
updated_at = models.DateTimeField(auto_now=True)
|
|
|
|
class Meta:
|
|
indexes: ClassVar[list] = [
|
|
models.Index(fields=["name"]),
|
|
models.Index(fields=["start_at", "end_at"]),
|
|
models.Index(fields=["game"]),
|
|
models.Index(fields=["owner"]),
|
|
]
|
|
|
|
def __str__(self) -> str:
|
|
"""Return a string representation of the drop campaign."""
|
|
return self.name
|
|
|
|
@property
|
|
def is_active(self) -> bool:
|
|
"""Check if the campaign is currently active."""
|
|
now = timezone.now()
|
|
return self.start_at <= now <= self.end_at
|
|
|
|
@property
|
|
def clean_name(self) -> str:
|
|
"""Return the campaign name without the game name prefix.
|
|
|
|
Examples:
|
|
"Ravendawn - July 2" -> "July 2"
|
|
"Party Animals Twitch Drop" -> "Twitch Drop"
|
|
"Skull & Bones - Closed Beta" -> "Closed Beta" (& is replaced with "and")
|
|
"""
|
|
if not self.game or not self.game.display_name:
|
|
return self.name
|
|
|
|
# Try different variations of the game name
|
|
game_variations = [self.game.display_name]
|
|
|
|
# Add & to "and" conversion
|
|
if "&" in self.game.display_name:
|
|
game_variations.append(self.game.display_name.replace("&", "and"))
|
|
|
|
# Add "and" to & conversion
|
|
if "and" in self.game.display_name:
|
|
game_variations.append(self.game.display_name.replace("and", "&"))
|
|
|
|
# Check each variation
|
|
for game_name in game_variations:
|
|
if not self.name.startswith(game_name):
|
|
continue
|
|
|
|
# Check if it's followed by a separator like " - "
|
|
if self.name[len(game_name) :].startswith(" - "):
|
|
return self.name[len(game_name) + 3 :].strip()
|
|
|
|
# Or just remove the game name if it's followed by a space
|
|
if len(self.name) > len(game_name) and self.name[len(game_name)] == " ":
|
|
return self.name[len(game_name) + 1 :].strip()
|
|
|
|
return self.name
|
|
|
|
|
|
class DropBenefit(models.Model):
|
|
"""Represents a benefit that can be earned from a drop."""
|
|
|
|
DISTRIBUTION_TYPES: ClassVar[list[tuple[str, str]]] = [
|
|
("DIRECT_ENTITLEMENT", "Direct Entitlement"),
|
|
("CODE", "Code"),
|
|
]
|
|
|
|
id = models.TextField(primary_key=True)
|
|
name = models.TextField(db_index=True)
|
|
image_asset_url = models.URLField(max_length=500, blank=True, default="")
|
|
created_at = models.DateTimeField(db_index=True)
|
|
entitlement_limit = models.PositiveIntegerField(default=1)
|
|
is_ios_available = models.BooleanField(default=False)
|
|
distribution_type = models.TextField(choices=DISTRIBUTION_TYPES, db_index=True)
|
|
|
|
# Foreign keys
|
|
game = models.ForeignKey(Game, on_delete=models.CASCADE, related_name="drop_benefits", db_index=True)
|
|
owner_organization = models.ForeignKey(Organization, on_delete=models.CASCADE, related_name="drop_benefits", db_index=True)
|
|
|
|
class Meta:
|
|
indexes: ClassVar[list] = [
|
|
models.Index(fields=["name"]),
|
|
models.Index(fields=["created_at"]),
|
|
models.Index(fields=["distribution_type"]),
|
|
models.Index(fields=["game"]),
|
|
models.Index(fields=["owner_organization"]),
|
|
]
|
|
|
|
def __str__(self) -> str:
|
|
"""Return a string representation of the drop benefit."""
|
|
return self.name
|
|
|
|
|
|
class TimeBasedDrop(models.Model):
|
|
"""Represents a time-based drop in a drop campaign."""
|
|
|
|
id = models.TextField(primary_key=True)
|
|
name = models.TextField(db_index=True)
|
|
required_minutes_watched = models.PositiveIntegerField(db_index=True)
|
|
required_subs = models.PositiveIntegerField(default=0)
|
|
start_at = models.DateTimeField(db_index=True)
|
|
end_at = models.DateTimeField(db_index=True)
|
|
|
|
# Foreign keys
|
|
campaign = models.ForeignKey(DropCampaign, on_delete=models.CASCADE, related_name="time_based_drops", db_index=True)
|
|
benefits = models.ManyToManyField(DropBenefit, through="DropBenefitEdge", related_name="drops")
|
|
|
|
class Meta:
|
|
indexes: ClassVar[list] = [
|
|
models.Index(fields=["name"]),
|
|
models.Index(fields=["start_at", "end_at"]),
|
|
models.Index(fields=["campaign"]),
|
|
models.Index(fields=["required_minutes_watched"]),
|
|
]
|
|
|
|
def __str__(self) -> str:
|
|
"""Return a string representation of the time-based drop."""
|
|
return self.name
|
|
|
|
|
|
class DropBenefitEdge(models.Model):
|
|
"""Represents the relationship between a TimeBasedDrop and a DropBenefit."""
|
|
|
|
drop = models.ForeignKey(TimeBasedDrop, on_delete=models.CASCADE, db_index=True)
|
|
benefit = models.ForeignKey(DropBenefit, on_delete=models.CASCADE, db_index=True)
|
|
entitlement_limit = models.PositiveIntegerField(default=1)
|
|
|
|
class Meta:
|
|
unique_together = ("drop", "benefit")
|
|
indexes: ClassVar[list] = [
|
|
models.Index(fields=["drop", "benefit"]),
|
|
]
|
|
|
|
def __str__(self) -> str:
|
|
"""Return a string representation of the drop benefit edge."""
|
|
return f"{self.drop.name} - {self.benefit.name}"
|