sfa/client/client_helper.py

   1
   2 def pg_users_arg(records):
   3     users = []
   4     for record in records:
   5         if record['type'] != 'user':
   6             continue
   7         user = {'urn': record['geni_urn'],
   8                 'keys': record['keys']}
   9         users.append(user)
  10     return users
  11
  12 def sfa_users_arg(records, slice_record):
  13     users = []
  14     for record in records:
  15         if record['type'] != 'user':
  16             continue
  17         try:
  18             user = {'urn': record['geni_urn'], #
  19                 'keys': record['keys'],
  20                 'email': record['email'], # needed for MyPLC
  21                 'person_id': record['person_id'], # needed for MyPLC
  22                 'first_name': record['first_name'], # needed for MyPLC
  23                 'last_name': record['last_name'], # needed for MyPLC
  24                 'slice_record': slice_record, # needed for legacy refresh peer
  25                 'key_ids': record['key_ids'] # needed for legacy refresh peer
  26                 }
  27         except:
  28             # handle NITOS user args
  29             user = {'urn': record['geni_urn'],
  30                 'keys': record['keys'],
  31                 'email': record['email'],
  32                 'user_id': record['user_id'],
  33                 'slice_record': slice_record,
  34                 }
  35
  36         users.append(user)
  37
  38     return users
  39
  40 def sfa_to_pg_users_arg(users):
  41
  42     new_users = []
  43     fields = ['urn', 'keys']
  44     for user in users:
  45         new_user = dict([item for item in user.items() \
  46           if item[0] in fields])
  47         new_users.append(new_user)
  48     return new_users