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"})