Update wizards/affiliation_institute to add protection.

controllers/wizards.py

@@ -50,14 +50,13 @@ def affiliation_institute():
     # extract keys defining the affiliation
     # u and v are the main keys use in inspirehep and cds
     # b is uses by some note in Atlas
-    key_1 = dict(key_u=record[u"110"]["u"], key_v="")
-    key_2 = dict(key_u=record[u"110"]["t"], key_v="")
-    key_3 = dict(key_u=record[u"110"]["b"], key_v="")
+    keys = (record[u"110"][k] for k in ("u", "t", "b") if k in record[u"110"])
+    keys = (dict(key_u=key, key_v="") for key in keys)
 
     # check that the rules does not exist
     # load new rules
     is_key_add = False
-    for key in (key_1, key_2, key_3):
+    for key in keys:
         if get_id(db.affiliation_keys, **key) is None:
             db.affiliation_keys[0] = key
             is_key_add = True