- record = GeniRecord(dict = record_dict)
- type = record.get_type()
- self.api.auth.verify_object_permission(record.get_name())
- auth_name = self.api.auth.get_authority(record.get_name())
- if not auth_name:
- auth_name = record.get_name()
- table = self.api.auth.get_auth_table(auth_name)
-
- # make sure the record exists
- existing_record_list = table.resolve(type, record.get_name())
- if not existing_record_list:
- raise RecordNotFound(record.get_name())
- existing_record = existing_record_list[0]
-
- # Update_membership needs the membership lists in the existing record
- # filled in, so it can see if members were added or removed
- self.api.fill_record_info(existing_record)
-
- # Use the pointer from the existing record, not the one that the user
- # gave us. This prevents the user from inserting a forged pointer
- pointer = existing_record.get_pointer()
-
- # update the PLC information that was specified with the record
-
- if (type == "authority"):
- self.api.plshell.UpdateSite(self.api.plauth, pointer, record)
-
- elif type == "slice":
- hrn=record.get_name()
- pl_record=self.api.geni_fields_to_pl_fields(type, hrn, record)
- self.api.plshell.UpdateSlice(self.api.plauth, pointer, pl_record)
-
- elif type == "user":
- # SMBAKER: UpdatePerson only allows a limited set of fields to be
- # updated. Ideally we should have a more generic way of doing
- # this. I copied the field names from UpdatePerson.py...
- update_fields = {}
- all_fields = record
- for key in all_fields.keys():
- if key in ['first_name', 'last_name', 'title', 'email',
- 'password', 'phone', 'url', 'bio', 'accepted_aup',
- 'enabled']:
- update_fields[key] = all_fields[key]
- self.api.plshell.UpdatePerson(self.api.plauth, pointer, update_fields)
-
- if 'key' in record and record['key']:
- # must check this key against the previous one if it exists
- persons = self.api.plshell.GetPersons(self.api.plauth, [pointer], ['key_ids'])
- person = persons[0]
- keys = person['key_ids']
- keys = self.api.plshell.GetKeys(self.api.plauth, person['key_ids'])
- key_exists = False
- if isinstance(record['key'], list):
- new_key = record['key'][0]
- else:
- new_key = record['key']
-
- # Delete all stale keys
- for key in keys:
- if record['key'] != key['key']:
- self.api.plshell.DeleteKey(self.api.plauth, key['key_id'])
- else:
- key_exists = True
- if not key_exists:
- self.api.plshell.AddPersonKey(self.api.plauth, pointer, {'key_type': 'ssh', 'key': new_key})
-
- # find the existing geni record
- hrn = record['hrn']
- auth_name = self.api.auth.get_authority(hrn)
- auth_info = self.api.auth.get_auth_info(auth_name)
- table = self.api.auth.get_auth_table(auth_name)
- person_records = table.resolve('user', hrn)
- person_record = person_records[0]
-
- # update the openssl key and gid
- pkey = convert_public_key(new_key)
- uuid = create_uuid()
- gid_object = self.api.auth.hierarchy.create_gid(hrn, uuid, pkey)
- gid = gid_object.save_to_string(save_parents=True)
- record['gid'] = gid
- record.set_gid(gid)
- table.update(record)
-
- elif type == "node":
- self.api.plshell.UpdateNode(self.api.plauth, pointer, record)
-
- else:
- raise UnknownGeniType(type)
-
- # update membership for researchers, pis, owners, operators
- self.api.update_membership(existing_record, record)