X-Git-Url: http://git.onelab.eu/?p=sfa.git;a=blobdiff_plain;f=tools%2Freset_gids.py;h=78dece702051d18a3e7bd026a955f8d6d6ac45da;hp=f4b7da52a05b78b196039930835364e440e3f153;hb=HEAD;hpb=8a6e87c49dd654bd91158d76c0692b1597593784 diff --git a/tools/reset_gids.py b/tools/reset_gids.py index f4b7da52..78dece70 100755 --- a/tools/reset_gids.py +++ b/tools/reset_gids.py @@ -1,8 +1,6 @@ -#!/usr/bin/env python +#!/usr/bin/env python3 # -*- coding:utf-8 -*- -import types - from sfa.storage.model import * from sfa.storage.alchemy import * from sfa.trust.gid import create_uuid @@ -10,8 +8,9 @@ from sfa.trust.hierarchy import Hierarchy from sfa.util.xrn import Xrn from sfa.trust.certificate import Certificate, Keypair, convert_public_key + def fix_users(): - s=global_dbsession + s = global_dbsession hierarchy = Hierarchy() users = s.query(RegRecord).filter_by(type="user") for record in users: @@ -19,16 +18,16 @@ def fix_users(): if not record.gid: uuid = create_uuid() pkey = Keypair(create=True) - pub_key=getattr(record,'reg_keys',None) - print pub_key + pub_key = getattr(record, 'reg_keys', None) if len(pub_key) > 0: # use only first key in record - if pub_key and isinstance(pub_key, types.ListType): pub_key = pub_key[0] + if pub_key and isinstance(pub_key, list): + pub_key = pub_key[0] pub_key = pub_key.key pkey = convert_public_key(pub_key) - urn = Xrn (xrn=record.hrn, type='user').get_urn() - email=getattr(record,'email',None) - gid_object = hierarchy.create_gid(urn, uuid, pkey, email = email) + urn = Xrn(xrn=record.hrn, type='user').get_urn() + email = getattr(record, 'email', None) + gid_object = hierarchy.create_gid(urn, uuid, pkey, email=email) gid = gid_object.save_to_string(save_parents=True) record.gid = gid s.commit()