Migration for quotas added

2023-11-25 18:05:12 +01:00
parent 0f423166f1
commit 0043abdbad
4 changed files with 47 additions and 0 deletions
--- a/migrations/202311251700.py
+++ b/migrations/202311251700.py
@@ -0,0 +1,9 @@
+from mongodb_migrations.base import BaseMigration
+
+
+class Migration(BaseMigration):
+    def upgrade(self):
+        self.db.users.update_many({}, {"$set": {"quota": None}})
+
+    def downgrade(self):
+        self.db.test_collection.update_many({}, {"$unset": "quota"})