forked from mirrors/bookwyrm
64 lines
2.3 KiB
Python
64 lines
2.3 KiB
Python
""" Generate preview images """
|
||
import sys
|
||
|
||
from django.core.management.base import BaseCommand
|
||
|
||
from bookwyrm import activitystreams, models, settings, preview_images
|
||
|
||
|
||
class Command(BaseCommand):
|
||
help = "Generate preview images"
|
||
|
||
def add_arguments(self, parser):
|
||
parser.add_argument(
|
||
"--all",
|
||
"-a",
|
||
action="store_true",
|
||
help="Generates images for ALL types: site, users and books. Can use a lot of computing power.",
|
||
)
|
||
|
||
# pylint: disable=no-self-use,unused-argument
|
||
def handle(self, *args, **options):
|
||
"""generate preview images"""
|
||
self.stdout.write(
|
||
" | Hello! I will be generating preview images for your instance."
|
||
)
|
||
if options["all"]:
|
||
self.stdout.write(
|
||
"🧑🎨 ⎨ This might take quite long if your instance has a lot of books and users."
|
||
)
|
||
self.stdout.write(" | ✧ Thank you for your patience ✧")
|
||
else:
|
||
self.stdout.write("🧑🎨 ⎨ I will only generate the instance preview image.")
|
||
self.stdout.write(" | ✧ Be right back! ✧")
|
||
|
||
# Site
|
||
self.stdout.write(" → Site preview image: ", ending="")
|
||
preview_images.generate_site_preview_image_task.delay()
|
||
self.stdout.write(" OK 🖼")
|
||
|
||
if options["all"]:
|
||
# Users
|
||
users = models.User.objects.filter(
|
||
local=True,
|
||
is_active=True,
|
||
)
|
||
self.stdout.write(
|
||
" → User preview images ({}): ".format(len(users)), ending=""
|
||
)
|
||
for user in users:
|
||
preview_images.generate_user_preview_image_task.delay(user.id)
|
||
self.stdout.write(".", ending="")
|
||
self.stdout.write(" OK 🖼")
|
||
|
||
# Books
|
||
books = models.Book.objects.select_subclasses().filter()
|
||
self.stdout.write(
|
||
" → Book preview images ({}): ".format(len(books)), ending=""
|
||
)
|
||
for book in books:
|
||
preview_images.generate_edition_preview_image_task.delay(book.id)
|
||
self.stdout.write(".", ending="")
|
||
self.stdout.write(" OK 🖼")
|
||
|
||
self.stdout.write("🧑🎨 ⎨ I’m all done! ✧ Enjoy ✧")
|