from django.db import models from events.models import Event class GameQuerySet(models.QuerySet): def enabled_for_event(self, event): return self.filter(event_games__event=event, event_games__enabled=True) class Game(models.Model): steam_app_id = models.PositiveIntegerField(null=True, blank=True) name = models.CharField(max_length=255) path = models.CharField(max_length=100) custom_path = models.BooleanField(default=False) image = models.URLField(null=True, blank=True) description = models.TextField(blank=True, default="") created_at = models.DateTimeField(auto_now_add=True) updated_at = models.DateTimeField(auto_now=True) objects = models.Manager.from_queryset(GameQuerySet)() class Meta: ordering = ["name"] def __str__(self) -> str: if self.steam_app_id: return f"{self.name} ({self.steam_app_id})" return self.name class EventGame(models.Model): event = models.ForeignKey( Event, on_delete=models.CASCADE, related_name="event_games", ) game = models.ForeignKey( Game, on_delete=models.CASCADE, related_name="event_games", ) enabled = models.BooleanField(default=True) apibot_token = models.CharField( max_length=512, blank=True, null=True, help_text=( "Bearer token for the legacy PolyLAN AAndT API (from the AAndT's " "'apibot' admin page), used to auto-sync Scopes and their assigned users." ), ) class Meta: constraints = [ models.UniqueConstraint( fields=["event", "game"], name="unique_event_game", ), ] def __str__(self) -> str: return f"{self.game.name} @ {self.event.name} ({'enabled' if self.enabled else 'disabled'})"