+#!/usr/bin/env plcsh
+import sys
+from optparse import OptionParser
+
+try:
+ vsys_tag_type=GetSliceTags({'tagname':'vsys'})[0]
+except:
+ print "Can't find tag vsys - exiting"
+ sys.exit(1)
+
+def add_value (slice, value, options):
+ (slice_id, slice_name ) = (slice['slice_id'], slice['name'])
+ if options.dry_run:
+ print "Would add vsys=%s to slice %s (%d)"%(value,slice_name,slice_id)
+ return
+ if options.verbose:
+ print "Adding vsys=%s to slice %s (%d)"%(value,slice_name,slice_id)
+ AddSliceTag (slice_id, 'vsys', value)
+
+
+def check (options):
+ # retrieve applicable slices
+ filter={}
+ if options.pattern: filter['name']=options.pattern
+ if not options.all: filter['peer_id']=None
+ slices=GetSlices(filter)
+ # find list of values
+ if options.tags:
+ values=options.tags
+ else:
+ config_tags=api.config.PLC_VSYS_DEFAULTS
+ values= [ value for value in config_tags.split(',') if value ]
+ # let's go
+ for value in values:
+ slice_tags=GetSliceTags({'tagname':'vsys','value':value})
+ names_with_tag = [ st['name'] for st in slice_tags ]
+ counter=0
+ for slice in slices:
+ if slice['name'] not in names_with_tag:
+ add_value (slice,value,options)
+ counter+=1
+ if options.verbose:
+ print "Found %d slices for which %s is missing"%(counter,value)
+
+def main ():
+ usage="""Usage: %prog
+ Checks that a set of slices has a set of vsys tags set
+Example:
+ %prog -- -p 'inria*' -t promisc -t fd_tuntap -vn
+"""
+ parser = OptionParser(usage=usage)
+ parser.add_option("-t","--tag",action='append',default=[],dest='tags',
+ help="ignore PLC config and provide tags on the command line - can be repeated")
+ parser.add_option("-a","--all",action='store_true',default=False,
+ dest='all',help="Apply on foreign slices as well (default is local only)")
+ parser.add_option("-p","--pattern",action='store',default=None,
+ dest='pattern',help="Apply on slices whose name matches pattern")
+ parser.add_option("-v", "--verbose", action = "store_true", default = False,
+ dest='verbose', help="be verbose")
+ parser.add_option("-n", "--dry-run", action = "store_true", default = False,
+ dest='dry_run', help="don't actually do it")
+ (options, args) = parser.parse_args()
+
+ if args:
+ parser.print_help()
+ sys.exit(1)
+
+ check (options)
+
+if __name__ == "__main__":
+ main()