Migrate DB schemas

Adds migration for DB to add new values.
UpstreamDataInc · Sep 11, 2024 · b624b53 · b624b53
1 parent 1a91544
commit b624b53
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/goosebit/db/migrations/models/1_20240911081506_add_image_format.py b/goosebit/db/migrations/models/1_20240911081506_add_image_format.py
@@ -0,0 +1,11 @@
+from tortoise import BaseDBAsyncClient
+
+
+async def upgrade(db: BaseDBAsyncClient) -> str:
+    return """
+        ALTER TABLE "software" ADD "image_format" SMALLINT NOT NULL  DEFAULT 0 /* SWU: 0\nRAUC: 1 */;"""
+
+
+async def downgrade(db: BaseDBAsyncClient) -> str:
+    return """
+        ALTER TABLE "software" DROP COLUMN "image_format";"""