Use values_list with flat, yay!

2022-06-07 23:14:45 -07:00 · 2022-06-07 23:14:45 -07:00 · 7f5d47a36f
parent 3aa159bc89
commit 7f5d47a36f
1 changed files with 5 additions and 5 deletions
--- 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=""
            )