54 lines
2.5 KiB
Python
54 lines
2.5 KiB
Python
|
"""Run the thumbnailer synchronously and stream progress to the console.
|
||
|
|
|
||
|
|
python manage.py generate_thumbnails 1 2 # specific projects
|
||
|
|
python manage.py generate_thumbnails --all # every project with a clip
|
||
|
|
python manage.py generate_thumbnails --failed # only failed/stuck ones
|
||
|
|
|
||
|
|
Unlike the upload path (which spawns a daemon thread and logs), this runs in the
|
||
|
|
foreground so you can watch each ffmpeg pass and the final status without tailing
|
||
|
|
app logs.
|
||
|
|
"""
|
||
|
|
from django.core.management.base import BaseCommand, CommandError
|
||
|
|
|
||
|
|
from scenes.models import Project
|
||
|
|
from scenes.thumbnails import generate_project_thumbnails
|
||
|
|
|
||
|
|
|
||
|
|
class Command(BaseCommand):
|
||
|
|
help = "Generate thumbnails for projects, streaming progress to stdout."
|
||
|
|
|
||
|
|
def add_arguments(self, parser):
|
||
|
|
parser.add_argument("project_ids", nargs="*", type=int,
|
||
|
|
help="Project ids to (re)generate.")
|
||
|
|
parser.add_argument("--all", action="store_true",
|
||
|
|
help="Generate for every project that has a clip.")
|
||
|
|
parser.add_argument("--failed", action="store_true",
|
||
|
|
help="Generate for projects whose status is failed/running/pending.")
|
||
|
|
|
||
|
|
def handle(self, *args, **opts):
|
||
|
|
ids = list(opts["project_ids"])
|
||
|
|
if opts["all"]:
|
||
|
|
ids += list(Project.objects.exclude(clip="").values_list("pk", flat=True))
|
||
|
|
if opts["failed"]:
|
||
|
|
ids += list(Project.objects
|
||
|
|
.filter(thumbnail_status__in=["failed", "running", "pending"])
|
||
|
|
.values_list("pk", flat=True))
|
||
|
|
ids = sorted(set(ids))
|
||
|
|
if not ids:
|
||
|
|
raise CommandError("no projects selected (pass ids, --all, or --failed)")
|
||
|
|
|
||
|
|
for pk in ids:
|
||
|
|
self.stdout.write(self.style.MIGRATE_HEADING(f"project {pk}"))
|
||
|
|
generate_project_thumbnails(pk, log=lambda msg: self.stdout.write(f" {msg}"))
|
||
|
|
|
||
|
|
# report the resulting state so a sweep ends with a clear summary
|
||
|
|
self.stdout.write("")
|
||
|
|
for project in Project.objects.filter(pk__in=ids).order_by("pk"):
|
||
|
|
line = f"project {project.pk} [{project.thumbnail_status}] {project.name}"
|
||
|
|
style = (self.style.SUCCESS if project.thumbnail_status == "ready"
|
||
|
|
else self.style.ERROR if project.thumbnail_status == "failed"
|
||
|
|
else self.style.WARNING)
|
||
|
|
self.stdout.write(style(line))
|
||
|
|
if project.thumbnail_error:
|
||
|
|
self.stdout.write(f" error: {project.thumbnail_error[:500]}")
|