4098430f1abefc0d7bf8d3fdb07d532e80e45a3f
[sfa.git] / 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                 'email': record['email']}
10         users.append(user)
11     return users    
12
13 def sfa_users_arg(records, slice_record):
14     users = []
15     for record in records:
16         if record['type'] != 'user': 
17             continue
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         users.append(user)
28     return users        
29
30 def sfa_to_pg_users_arg(users):
31
32     new_users = []
33     fields = ['urn', 'keys']
34     for user in users:
35         new_user = dict([item for item in user.items() \
36           if item[0] in fields])
37         new_users.append(new_user)
38     return new_users