diff --git a/bookwyrm/management/commands/generate_preview_images.py b/bookwyrm/management/commands/generate_preview_images.py index 28ea1c0cb..6ca26b2d3 100644 --- a/bookwyrm/management/commands/generate_preview_images.py +++ b/bookwyrm/management/commands/generate_preview_images.py @@ -10,48 +10,55 @@ 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.") + 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( + " | 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("šŸ§‘ā€šŸŽØ āŽØ I will only generate the instance preview image.") self.stdout.write(" | āœ§ Be right back! āœ§") # Site - self.stdout.write(" ā†’ Site preview image: ", ending='') + self.stdout.write(" ā†’ Site preview image: ", ending="") preview_images.generate_site_preview_image_task.delay() self.stdout.write(" OK šŸ–¼") - if options['all']: + if options["all"]: # Users users = models.User.objects.filter( local=True, is_active=True, ) - self.stdout.write(" ā†’ User preview images ({}): ".format(len(users)), ending='') + 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(".", ending="") self.stdout.write(" OK šŸ–¼") # Books books = models.Book.objects.select_subclasses().filter() - self.stdout.write(" ā†’ Book preview images ({}): ".format(len(books)), ending='') + 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(".", ending="") self.stdout.write(" OK šŸ–¼") self.stdout.write("šŸ§‘ā€šŸŽØ āŽØ Iā€™m all done! āœ§ Enjoy āœ§") - - diff --git a/bookwyrm/preview_images.py b/bookwyrm/preview_images.py index 60b69d0a3..7a23353c8 100644 --- a/bookwyrm/preview_images.py +++ b/bookwyrm/preview_images.py @@ -126,11 +126,15 @@ def generate_instance_layer(content_width): def generate_rating_layer(rating, content_width): - icon_star_full = Image.open(os.path.join(settings.STATIC_ROOT, "images/icons/star-full.png")) + icon_star_full = Image.open( + os.path.join(settings.STATIC_ROOT, "images/icons/star-full.png") + ) icon_star_empty = Image.open( os.path.join(settings.STATIC_ROOT, "images/icons/star-empty.png") ) - icon_star_half = Image.open(os.path.join(settings.STATIC_ROOT, "images/icons/star-half.png")) + icon_star_half = Image.open( + os.path.join(settings.STATIC_ROOT, "images/icons/star-half.png") + ) icon_size = 64 icon_margin = 10 @@ -321,7 +325,9 @@ def generate_site_preview_image_task(): "text_three": site.instance_tagline, } - image = generate_preview_image(texts=texts, picture=logo, show_instance_layer=False) + image = generate_preview_image( + texts=texts, picture=logo, show_instance_layer=False + ) save_and_cleanup(image, instance=site)