[videos] Add management command and fix lookup of series

This commit is contained in:
2025-11-18 10:32:54 -05:00
parent c2ba8a48ac
commit 6f7f739ca6
5 changed files with 36 additions and 6 deletions

View File

@ -0,0 +1,27 @@
from django.core.management.base import BaseCommand, CommandError
from django.core import serializers
from videos.models import Video, Series, Channel
import json
class Command(BaseCommand):
help = "Find or create a Video by ID and output it as JSON"
def add_arguments(self, parser):
parser.add_argument("video_id", type=str, help="The video ID to find or create")
def handle(self, *args, **options):
instance = Video.find_or_create(options.get("video_id", ""), overwrite=True)
data = json.loads(serializers.serialize("json", [instance]))[0]
# --- Enrich with series model ---
if instance.tv_series_id:
series_instance = instance.tv_series
series_json = json.loads(serializers.serialize("json", [series_instance]))[0]
data["series"] = series_json # new nested field
if instance.channel_id:
channel_instance = instance.channel
channel_json = json.loads(serializers.serialize("json", [channel_instance]))[0]
data["channel"] = channel_json # new nested field
self.stdout.write(json.dumps(data, indent=2))