Re-enable the TBA event download management command

Author: jhh

flake.nix

@@ -53,13 +53,15 @@
 
             pillowHeifOverrides = import ./lib/overrides-pillow-heif.nix { inherit pkgs; };
             psycopgOverrides = import ./lib/overrides-psycopg.nix { inherit pkgs; };
+            tbaApiOverrides = import ./lib/overrides-tba-api-v3client.nix { inherit pkgs; };
           in
           baseSet.overrideScope (
             lib.composeManyExtensions [
               pyproject-build-systems.overlays.default
               overlay
               pillowHeifOverrides
               psycopgOverrides
+              tbaApiOverrides
             ]
           );
 
@@ -88,7 +90,13 @@
             '';
           };
 
-          manage = writeShellApplication {
+          manage = import ./lib/manage.nix {
+            inherit pkgs;
+            inherit (self.packages.${system}) venv;
+            inherit (self.packages.${system}) static;
+          };
+
+          manage-old = writeShellApplication {
             name = "strykeforce-manage";
 
             text = ''

lib/manage.nix

@@ -0,0 +1,25 @@
+{
+  pkgs,
+  static,
+  venv,
+}:
+pkgs.writeShellApplication {
+  name = "strykeforce-manage";
+  text = ''
+    if [ "$UID" -ne 0 ]; then
+        echo "error: run this command as root."
+        exit 1
+    fi
+
+    # shellcheck source=/dev/null
+    source /run/agenix/stryker_website_secrets
+
+    sudo -u strykeforce env \
+      DJANGO_SETTINGS_MODULE=website.settings.production \
+      DJANGO_DATABASE_URL=postgres:///strykeforce \
+      SECRET_KEY=not-secret \
+      STATIC_ROOT=${static} \
+      TBA_READ_KEY="$TBA_READ_KEY" \
+      ${venv}/bin/strykeforce-manage "$@"
+  '';
+}

lib/overrides-tba-api-v3client.nix

@@ -0,0 +1,10 @@
+{ pkgs }:
+final: prev: {
+  tba-api-v3client = prev.tba-api-v3client.overrideAttrs (old: {
+    nativeBuildInputs =
+      old.nativeBuildInputs or [ ]
+      ++ (final.resolveBuildSystem {
+        setuptools = [ ];
+      });
+  });
+}

pyproject.toml

@@ -12,6 +12,7 @@ dependencies = [
     "psycopg[c,pool]>=3.2.3",
     "redis>=5.2.1",
     "sentry-sdk>=2.19.2",
+    "tba-api-v3client>=3.8.0",
     "wagtail>=6.3.2",
     "whitenoise[brotli]>=6.8.2",
 ]

uv.lock

@@ -2,10 +2,9 @@
 
 import tbaapiv3client
 from django.conf import settings
-from django.core.management import BaseCommand
-from django.core.management import CommandError
+from django.core.exceptions import ObjectDoesNotExist
+from django.core.management import BaseCommand, CommandError
 from django.db import transaction
-from django.db.models import Model
 from tbaapiv3client import ApiException
 
 from website.events.models import Event
@@ -62,7 +61,7 @@ def handle(self, *args, **options):
         for event in events:
             try:
                 obj = Event.objects.get(key=event.key)
-            except Model.DoesNotExist:
+            except ObjectDoesNotExist:
                 obj = Event()
 
             obj.key = event.key

website/events/management/commands/import_events.py

@@ -1,18 +1,20 @@
 from __future__ import annotations
 
 from django.db import models
-from django.db.models import CharField
-from django.db.models import DateField
-from django.db.models import FloatField
-from django.db.models import IntegerField
-from django.db.models import JSONField
-from django.db.models import URLField
+from django.db.models import (
+    CharField,
+    DateField,
+    FloatField,
+    IntegerField,
+    JSONField,
+    URLField,
+)
 from django.utils import timezone
 from wagtail.admin.panels import FieldPanel
-from wagtail.contrib.routable_page.models import path
-from wagtail.contrib.routable_page.models import RoutablePageMixin
+from wagtail.contrib.routable_page.models import RoutablePageMixin, path
 from wagtail.fields import RichTextField
 from wagtail.models import Page
+from wagtail.snippets.models import register_snippet
 
 MAX_LENGTH = 1000
 
@@ -70,6 +72,7 @@ def event_for_key(self, request, key):
     ]
 
 
+@register_snippet
 class Event(models.Model):
     key = CharField(unique=True, default="", max_length=MAX_LENGTH)
     name = CharField(default="", max_length=MAX_LENGTH)