From 7f5d47a36f765d54ad47c8b272c5b778776e8f75 Mon Sep 17 00:00:00 2001 From: Joel Bradshaw Date: Tue, 7 Jun 2022 23:14:45 -0700 Subject: [PATCH] Use values_list with flat, yay! --- .../management/commands/generate_preview_images.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/bookwyrm/management/commands/generate_preview_images.py b/bookwyrm/management/commands/generate_preview_images.py index 7952803d..9ff16c26 100644 --- a/bookwyrm/management/commands/generate_preview_images.py +++ b/bookwyrm/management/commands/generate_preview_images.py @@ -56,12 +56,12 @@ class Command(BaseCommand): self.stdout.write(" OK 🖼") # Books - book_ids = [ - values["id"] - for values in models.Book.objects.select_subclasses() + book_ids = ( + models.Book.objects.select_subclasses() .filter() - .values("id") - ] + .values_list("id", flat=True) + ) + self.stdout.write( " → Book preview images ({}): ".format(len(book_ids)), ending="" )