Only prevent running if both buckets exist

alphagov · Oct 7, 2013 · 63b3c0d · 63b3c0d
1 parent 6f2529d
commit 63b3c0d
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/migrations/004_load_seed_file.py b/migrations/004_load_seed_file.py
@@ -12,12 +12,8 @@
 def up(db):
     names = db.collection_names()
 
-    if "users" in names:
-        log.info("users collection already created")
-        return
-
-    if "buckets" in names:
-        log.info("buckets collection already created")
+    if "users" in names and "buckets" in names:
+        log.info("both users and buckets collections already exist")
         return
 
     invoke = os.path.join(os.path.dirname(sys.executable), "invoke")