Allow migrations to be run by pattern

run_migrations.py

@@ -3,8 +3,8 @@
 """
 import imp
 import os
+import re
 import sys
-import pymongo
 from os.path import join
 import logging
 from backdrop.core.database import Database
@@ -34,10 +34,16 @@ def get_database(config):
         config.MONGO_HOSTS, config.MONGO_PORT, config.DATABASE_NAME)
 
 
-def get_migrations():
+def match_migration_path(path, pattern):
+    if not path.endswith('.py'):
+        return False
+    return pattern is None or pattern.search(path)
+
+
+def get_migrations(pattern):
     migrations_path = join(ROOT_PATH, 'migrations')
     for migration_file in os.listdir(migrations_path):
-        if migration_file.endswith('.py'):
+        if match_migration_path(migration_file, pattern):
             migration_path = join(migrations_path, migration_file)
 
             yield imp.load_source('migration', migration_path)
@@ -48,6 +54,10 @@ def get_migrations():
     config = load_config(os.getenv('GOVUK_ENV', 'development'))
     database = get_database(config)
 
-    for migration in get_migrations():
+    migration_pattern = None
+    if len(sys.argv) > 1:
+        migration_pattern = re.compile(sys.argv[1])
+
+    for migration in get_migrations(migration_pattern):
         log.info("Running migration %s" % migration)
         migration.up(database)