1 files changed, 22 insertions, 0 deletions

diff --git a/bot/migrations/0001-add_exif.py b/bot/migrations/0001-add_exif.py
new file mode 100644
index 0000000..0eff797
--- /dev/null
+++ b/bot/migrations/0001-add_exif.py
@@ -0,0 +1,22 @@
+import os
+import sys
+from playhouse.migrate import *
+
+
+if os.getenv("DB_DRIVER") == "psql":
+    db = PostgresqlDatabase(os.getenv("DB_NAME"), user=os.getenv("DB_USER"), password=os.getenv("DB_PASS"),
+                            host=os.getenv("DB_HOST"), port=os.getenv("DB_PORT"))
+    migrator = PostgresqlMigrator(db)
+elif os.getenv("DB_DRIVER") == "sqlite3":
+    db = SqliteDatabase("database/photos.db")
+    migrator = SqliteMigrator(db)
+else:
+    print("No DB_DRIVER specified")
+    sys.exit(1)
+
+model = CharField(max_length=256, default="")
+
+# Run the migration, specifying the database table, field name and field.
+migrate(
+    migrator.add_column('photo', 'model', model),
+)