sfa/client/common.py

   1 # a few utilities common to sfi and sfaadmin
   2
   3 def optparse_listvalue_callback(option, opt, value, parser):
   4     former=getattr(parser.values,option.dest)
   5     if not former: former=[]
   6     # support for using e.g. sfi update -t slice -x the.slice.hrn -r none
   7     # instead of -r '' which is painful and does not pass well through ssh
   8     if value.lower()=='none':
   9         newvalue=former
  10     else:
  11         newvalue=former+value.split(',')
  12     setattr(parser.values, option.dest, newvalue)
  13
  14 def optparse_dictvalue_callback (option, option_string, value, parser):
  15     try:
  16         (k,v)=value.split('=',1)
  17         d=getattr(parser.values, option.dest)
  18         d[k]=v
  19     except:
  20         parser.print_help()
  21         sys.exit(1)
  22
  23 # a code fragment that could be helpful for argparse which unfortunately is
  24 # available with 2.7 only, so this feels like too strong a requirement for the client side
  25 #class ExtraArgAction  (argparse.Action):
  26 #    def __call__ (self, parser, namespace, values, option_string=None):
  27 # would need a try/except of course
  28 #        (k,v)=values.split('=')
  29 #        d=getattr(namespace,self.dest)
  30 #        d[k]=v
  31 #####
  32 #parser.add_argument ("-X","--extra",dest='extras', default={}, action=ExtraArgAction,
  33 #                     help="set extra flags, testbed dependent, e.g. --extra enabled=true")
  34
  35 ##############################
  36 # these are not needed from the outside
  37 def terminal_render_plural (how_many, name,names=None):
  38     if not names: names="%ss"%name
  39     if how_many<=0: return "No %s"%name
  40     elif how_many==1: return "1 %s"%name
  41     else: return "%d %s"%(how_many,names)
  42
  43 def terminal_render_default (record,options):
  44     print "%s (%s)" % (record['hrn'], record['type'])
  45 def terminal_render_user (record, options):
  46     print "%s (User)"%record['hrn'],
  47     if record.get('reg-pi-authorities',None): print " [PI at %s]"%(" and ".join(record['reg-pi-authorities'])),
  48     if record.get('reg-slices',None): print " [IN slices %s]"%(" and ".join(record['reg-slices'])),
  49     user_keys=record.get('reg-keys',[])
  50     if not options.verbose:
  51         print " [has %s]"%(terminal_render_plural(len(user_keys),"key"))
  52     else:
  53         print ""
  54         for key in user_keys: print 8*' ',key.strip("\n")
  55
  56 def terminal_render_slice (record, options):
  57     print "%s (Slice)"%record['hrn'],
  58     if record.get('reg-researchers',None): print " [USERS %s]"%(" and ".join(record['reg-researchers'])),
  59 #    print record.keys()
  60     print ""
  61 def terminal_render_authority (record, options):
  62     print "%s (Authority)"%record['hrn'],
  63     if record.get('reg-pis',None): print " [PIS %s]"%(" and ".join(record['reg-pis'])),
  64     print ""
  65 def terminal_render_node (record, options):
  66     print "%s (Node)"%record['hrn']
  67
  68
  69 ### used in sfi list
  70 def terminal_render (records, options):
  71     # sort records by type
  72     grouped_by_type = {}
  73     for record in records:
  74         type = record['type']
  75         if type not in grouped_by_type:
  76             grouped_by_type[type]=[]
  77         grouped_by_type[type].append(record)
  78     group_types = grouped_by_type.keys()
  79     group_types.sort()
  80     for type in group_types:
  81         group = grouped_by_type[type]
  82 #        print 20 * '-', type
  83         try:    renderer = eval('terminal_render_' + type)
  84         except: renderer = terminal_render_default
  85         for record in group:
  86             renderer(record, options)
  87
  88 ####################
  89 def filter_records(type, records):
  90     filtered_records = []
  91     for record in records:
  92         if (record['type'] == type) or (type == "all"):
  93             filtered_records.append(record)
  94     return filtered_records
  95
  96