Migrations (#124)

* Migrations setup

* Add major to array migration

Co-authored-by: Lucent Fong <[email protected]>
Co-authored-by: js324 <[email protected]>

migrate.py

@@ -0,0 +1,7 @@
+import config
+from mongodb_migrations.cli import MigrationManager
+
+mgr = MigrationManager()
+mgr.config.mongo_url = config.DB_URI
+mgr.config.metastore = 'migrations'
+mgr.run()

migrations/20200820_users_major_to_array.py

@@ -0,0 +1,13 @@
+from mongodb_migrations.base import BaseMigration
+from titlecase import titlecase
+
+class Migration(BaseMigration):
+    def upgrade(self):
+        for u in self.db.users.find():
+            u['major'] = list(map(lambda s: titlecase(s.strip()), u['major'].split(';')))
+            self.db.users.save(u)
+
+    def downgrade(self):
+        for u in self.db.users.find():
+            u['major'] = u['major'].join(';')
+            self.db.users.save(u)

migrations/_20200806_example.py

@@ -0,0 +1,10 @@
+from mongodb_migrations.base import BaseMigration
+
+class Migration(BaseMigration):
+    def upgrade(self):
+        for i in self.db.some_collection.find():
+            item['new_column'] = 'new_value'
+            self.db.some_collection.save(item)
+
+    def downgrade(self):
+        self.db.some_collection.update_many({}, {"$unset": {"new_column": ""}})

requirements.txt

@@ -25,6 +25,7 @@ jsonschema==2.6.0
 lazy-object-proxy==1.3.1
 mccabe==0.6.1
 mock==4.0.2
+mongodb-migrations==0.7.0
 more-itertools==5.0.0
 oauth2client==4.1.3
 oauthlib==3.1.0
@@ -46,6 +47,7 @@ rsa==4.7
 s3transfer==0.2.1
 six==1.12.0
 sparkpost==1.3.6
+titlecase==1.1.1
 typed-ast==1.4.3
 uritemplate==3.0.0
 urllib3==1.24.2