Changeset 17c118f

Timestamp:

Jun 5, 2013, 6:42:54 AM (12 years ago)

Author:

Alex Dehnert <adehnert@…>

Branches:

master, stable, stage

Children:

d983e4f, 53e624e

Parents:

ecf87c8

git-author:

Alex Dehnert <adehnert@…> (06/05/13 06:33:23)

git-committer:

Alex Dehnert <adehnert@…> (06/05/13 06:42:54)

Message:

Create rooms if missing while setting lock_type

Apparently 50-020 is in some, but not all, instances of the DB. This makes sure
that it will get created, rather than failing the migration. (We could just
leave it missing and ignore the error, but it seems nice to include all access
type info we can get.)

File:

• asadb/space/migrations/0006_lock_types_setup.py (modified) (1 diff)

asadb/space/migrations/0006_lock_types_setup.py

@@ -88 +88 @@
         spaces.filter(number__startswith='50-').update(lock_type=lock_types.get(slug='unknown'))
         spaces.filter(number__startswith='N52-').update(lock_type=lock_types.get(slug='semo'))
+        created_rooms = []
         for room, lock_type in room_lock_types:
             print "Setting room=%s's lock_type to %s" % (room, lock_type, )
-            room_obj = spaces.get(number=room)
-            room_obj.lock_type = lock_types.get(slug=lock_type)
+            lt_obj = lock_types.get(slug=lock_type)
+            room_obj, created = spaces.get_or_create(number=room, defaults=dict(lock_type=lt_obj))
+            if created: created_rooms.append(room_obj)
+            room_obj.lock_type = lt_obj
             room_obj.save()
+
+        for room_obj in created_rooms:
+            print "Created %s while setting lock_type" % (room_obj.number, )