X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=bin%2Fcheck-hrns.py;h=7c8bdef73caf2d19a016076c7306a7f693a52c8d;hb=20aa5b1780c2c2ae241449cf18106a61a9bb065f;hp=f92d41277f5b96709a6ba473e0a27e454b2d994a;hpb=7e2ad8541774b6c74430d465a143fd1e960d0698;p=myplc.git diff --git a/bin/check-hrns.py b/bin/check-hrns.py index f92d412..7c8bdef 100755 --- a/bin/check-hrns.py +++ b/bin/check-hrns.py @@ -2,17 +2,21 @@ import sys from optparse import OptionParser -from PLC.Namespace import hostname_to_hrn, email_to_hrn +from PLC.Namespace import hostname_to_hrn, email_to_hrn, slicename_to_hrn # (auth_hrn, email): -toplevel=api.config.PLC_HRN_ROOT +try: + from sfa.util.config import Config + toplevel=Config().SFA_INTERFACE_HRN +except: + toplevel=api.config.PLC_HRN_ROOT def handle_nodes (sites,sites_by_id, dry_run, verbose): nodes=GetNodes ({'peer_id':None},['node_id','hostname','hrn']) nodes_by_id = dict ( [ (node['node_id'],node) for node in nodes ] ) for site in sites: login_base=site['login_base'] - for node_id in site['node_ids']: + for node_id in site.get('node_ids', []): try: node=nodes_by_id[node_id] except: print 'cannot find node %s'%node_id; continue hrn=hostname_to_hrn (toplevel, login_base, node['hostname']) @@ -21,32 +25,74 @@ def handle_nodes (sites,sites_by_id, dry_run, verbose): if dry_run: continue SetNodeHrn (node['node_id'],hrn) else: - if verbose: print "host %s OK"%node['hostname'] + if verbose: print "Node %s OK"%node['hostname'] def handle_persons (sites,sites_by_id, dry_run,verbose): persons=GetPersons ({'peer_id':None},['person_id','email','hrn','site_ids']) - for person in persons: - how_many=len(person['site_ids']) - if how_many !=1: - if verbose: print "person %s not in 1 site (%s instead) -- ignored"%(person['email'],how_many) - continue - try: login_base=sites_by_id[person['site_ids'][0]]['login_base'] - except: print "Cannot handle person %s - site not found"%person['email']; continue - hrn=email_to_hrn ("%s.%s"%(toplevel,login_base),person['email']) - if person['hrn'] != hrn: - print "Person %s - current hrn %s, should be %s"%(person['email'], person['hrn'], hrn) + persons_by_id = dict ( [ (person['person_id'],person) for person in persons ] ) + for site in sites: + login_base=site['login_base'] + for person_id in site.get('person_ids', []): + try: person=persons_by_id[person_id] + except: print 'cannot find person %s'%person_id; continue + how_many=len(person['site_ids']) + if how_many !=1: + if verbose: print "Checking persons in exactly one site -- person %s in %s site(s) -- ignored"%(person['email'],how_many) + continue + + hrn=email_to_hrn ("%s.%s"%(toplevel,login_base),person['email']) + if person['hrn'] != hrn: + print "Person %s - current hrn %s, should be %s"%(person['email'], person['hrn'], hrn) + if dry_run: continue + SetPersonHrn (person['person_id'],hrn) + else: + if verbose: print "Person %s OK"%person['email'] + + +def handle_slices (sites,sites_by_id, dry_run,verbose): + slices=GetSlices ({'peer_id':None},['slice_id','name','hrn','site_id']) + slices_by_id = dict ( [ (slice['slice_id'],slice) for slice in slices ] ) + for site in sites: + login_base=site['login_base'] + for slice_id in site.get('slice_ids', []): + try: slice=slices_by_id[slice_id] + except: print 'cannot find slice %s'%slice_id; continue + hrn=slicename_to_hrn (toplevel, slice['name']) + if slice['hrn'] != hrn: + print "Slice %s - current hrn %s, should be %s"%(slice['name'], slice['hrn'], hrn) + if dry_run: continue + SetSliceHrn (slice['slice_id'],hrn) + else: + if verbose: print "Slice %s OK"%slice['name'] + + +def handle_sites (sites,sites_by_id, dry_run,verbose): + for site in sites: + hrn='.'.join([toplevel, site['login_base']]) + if site['hrn'] != hrn: + print "Site %s - current hrn %s, should be %s"%(site['name'], site['hrn'], hrn) if dry_run: continue - SetPersonHrn (person['person_id'],hrn) + SetSiteHrn (site['site_id'],hrn) else: - if verbose: print "host %s OK"%person['email'] + if verbose: print "Site %s OK"%site['name'] + def main(): - parser = OptionParser() + usage="""Usage: %prog + Checks that the hrn tags are correctly set +Example: + %prog -- -p -nv +""" + parser = OptionParser(usage=usage) parser.add_option("-p", "--person", action = "store_true", default = False, dest='persons',help="run on persons") parser.add_option("-n", "--node", action = "store_true", default = False, dest='nodes',help="run on nodes") + parser.add_option("-S", "--slice", action = "store_true", default = False, + dest='slices',help="run on slices") + parser.add_option("-t", "--site", action = "store_true", default = False, + dest='sites',help="run on sites") parser.add_option("-s", "--show", action = "store_true", default = False, dest='show', help="dry run, only show discrepencies") parser.add_option("-v", "--verbose", action = "store_true", default = False, @@ -57,17 +103,24 @@ def main(): parser.print_help() sys.exit(1) # if neither -p nor -n, run both - if not options.nodes and not options.persons: + if not options.nodes and not options.persons and not options.slices and not options.sites: options.nodes=True options.persons=True - + options.slices=True + options.sites=True + dry_run=options.show verbose=options.verbose # optimizing : we compute the set of sites only once - sites = GetSites({'peer_id':None},['site_id','login_base','node_ids','person_ids']) + sites = GetSites({'peer_id':None},['site_id','login_base','node_ids','person_ids','name','hrn']) + # remove external sites created through SFA + sites = [site for site in sites if not site['name'].startswith('sfa:')] sites_by_id = dict ( [ (site['site_id'], site) for site in sites ] ) + if options.nodes: handle_nodes(sites,sites_by_id,dry_run,verbose) if options.persons: handle_persons(sites,sites_by_id,dry_run,verbose) + if options.slices: handle_slices(sites,sites_by_id,dry_run,verbose) + if options.sites: handle_sites(sites,sites_by_id,dry_run,verbose) if __name__ == "__main__": main()