From 6b3447761f2f27c1c2d09a545fff1edbb6421ec4 Mon Sep 17 00:00:00 2001
From: Mouse Reeve <mousereeve@riseup.net>
Date: Thu, 20 May 2021 18:16:35 -0700
Subject: [PATCH] Adds book format field with choices

---
 ...l_format_edition_physical_format_detail.py | 35 +++++++++++++++++++
 bookwyrm/models/book.py                       | 12 ++++++-
 2 files changed, 46 insertions(+), 1 deletion(-)
 create mode 100644 bookwyrm/migrations/0076_rename_physical_format_edition_physical_format_detail.py

diff --git a/bookwyrm/migrations/0076_rename_physical_format_edition_physical_format_detail.py b/bookwyrm/migrations/0076_rename_physical_format_edition_physical_format_detail.py
new file mode 100644
index 00000000..05ad1a2b
--- /dev/null
+++ b/bookwyrm/migrations/0076_rename_physical_format_edition_physical_format_detail.py
@@ -0,0 +1,35 @@
+# Generated by Django 3.2 on 2021-05-21 00:17
+
+from django.db import migrations
+import bookwyrm
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("bookwyrm", "0075_announcement"),
+    ]
+
+    operations = [
+        migrations.RenameField(
+            model_name="edition",
+            old_name="physical_format",
+            new_name="physical_format_detail",
+        ),
+        migrations.AddField(
+            model_name="edition",
+            name="physical_format",
+            field=bookwyrm.models.fields.CharField(
+                blank=True,
+                choices=[
+                    ("AudiobookFormat", "Audiobookformat"),
+                    ("EBook", "Ebook"),
+                    ("GraphicNovel", "Graphicnovel"),
+                    ("Hardcover", "Hardcover"),
+                    ("Paperback", "Paperback"),
+                ],
+                max_length=255,
+                null=True,
+            ),
+        ),
+    ]
diff --git a/bookwyrm/models/book.py b/bookwyrm/models/book.py
index 869ff04d..ae79223d 100644
--- a/bookwyrm/models/book.py
+++ b/bookwyrm/models/book.py
@@ -169,6 +169,13 @@ class Work(OrderedCollectionPageMixin, Book):
     deserialize_reverse_fields = [("editions", "editions")]
 
 
+# https://schema.org/BookFormatType
+FormatChoices = models.TextChoices(
+    "FormatChoices",
+    "AudiobookFormat EBook GraphicNovel Hardcover Paperback",
+)
+
+
 class Edition(Book):
     """an edition of a book"""
 
@@ -186,7 +193,10 @@ class Edition(Book):
         max_length=255, blank=True, null=True, deduplication_field=True
     )
     pages = fields.IntegerField(blank=True, null=True)
-    physical_format = fields.CharField(max_length=255, blank=True, null=True)
+    physical_format = fields.CharField(
+        max_length=255, choices=FormatChoices.choices, null=True, blank=True
+    )
+    physical_format_detail = fields.CharField(max_length=255, blank=True, null=True)
     publishers = fields.ArrayField(
         models.CharField(max_length=255), blank=True, default=list
     )