From 820279166af24dd8051f4829822fe44528c88b3f Mon Sep 17 00:00:00 2001
From: Mouse Reeve <mousereeve@riseup.net>
Date: Wed, 16 Mar 2022 12:39:49 -0700
Subject: [PATCH] Adds update script

---
 .../management/commands/instance_version.py   | 54 +++++++++++++++++++
 .../migrations/0145_sitesettings_version.py   | 18 +++++++
 bookwyrm/models/site.py                       |  1 +
 3 files changed, 73 insertions(+)
 create mode 100644 bookwyrm/management/commands/instance_version.py
 create mode 100644 bookwyrm/migrations/0145_sitesettings_version.py

diff --git a/bookwyrm/management/commands/instance_version.py b/bookwyrm/management/commands/instance_version.py
new file mode 100644
index 00000000..ca150d64
--- /dev/null
+++ b/bookwyrm/management/commands/instance_version.py
@@ -0,0 +1,54 @@
+""" Get your admin code to allow install """
+from django.core.management.base import BaseCommand
+
+from bookwyrm import models
+from bookwyrm.settings import VERSION
+
+
+# pylint: disable=no-self-use
+class Command(BaseCommand):
+    """command-line options"""
+
+    help = "What version is this?"
+
+    def add_arguments(self, parser):
+        """specify which function to run"""
+        parser.add_argument(
+            "--current",
+            action="store_true",
+            help="Version stored in database",
+        )
+        parser.add_argument(
+            "--target",
+            action="store_true",
+            help="Version stored in settings",
+        )
+        parser.add_argument(
+            "--update",
+            action="store_true",
+            help="Update database version",
+        )
+
+    # pylint: disable=unused-argument
+    def handle(self, *args, **options):
+        """execute init"""
+        site = models.SiteSettings.objects.get()
+        current = site.version or "0.0.1"
+        target = VERSION
+        if options.get("current"):
+            print(current)
+            return
+
+        if options.get("target"):
+            print(target)
+            return
+
+        if options.get("update"):
+            site.version = target
+            site.save()
+            return
+
+        if current != target:
+            print(f"{current}/{target}")
+        else:
+            print(current)
diff --git a/bookwyrm/migrations/0145_sitesettings_version.py b/bookwyrm/migrations/0145_sitesettings_version.py
new file mode 100644
index 00000000..649f90ab
--- /dev/null
+++ b/bookwyrm/migrations/0145_sitesettings_version.py
@@ -0,0 +1,18 @@
+# Generated by Django 3.2.12 on 2022-03-16 18:10
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("bookwyrm", "0144_alter_announcement_display_type"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="sitesettings",
+            name="version",
+            field=models.CharField(blank=True, max_length=10, null=True),
+        ),
+    ]
diff --git a/bookwyrm/models/site.py b/bookwyrm/models/site.py
index c6c53f76..cbad6c4b 100644
--- a/bookwyrm/models/site.py
+++ b/bookwyrm/models/site.py
@@ -27,6 +27,7 @@ class SiteSettings(models.Model):
     default_theme = models.ForeignKey(
         "Theme", null=True, blank=True, on_delete=models.SET_NULL
     )
+    version = models.CharField(null=True, blank=True, max_length=10)
 
     # admin setup options
     install_mode = models.BooleanField(default=False)