sface/sficonfig.py

   1 # config for sface
   2 # uses in this order
   3 # command-line args
   4 # ~/.sfi/sfi_config
   5 ###
   6
   7 import os
   8 from optparse import OptionParser
   9
  10
  11 class SfiConfig:
  12
  13     d_registry= "http://www.planet-lab.org:12345"
  14     d_slicemgr= "http://www.planet-lab.org:12347"
  15     d_aggmgr=   "http://www.planet-lab.org:12346"
  16
  17     supported = [
  18         # local code name, config variable name, default
  19         ('slice',       'SFI_SLICE' ,   None,           '-s','--slice',         "slice HRN"),
  20         ('user',        'SFI_USER',     None,           '-u','--user',          "user HRN"),
  21         ('authority',   'SFI_AUTH',     None,           '-t','--auth',          "users's authority HRN"),
  22         ('registry',    'SFI_REGISTRY', d_registry,     '-r','--registry',      "registry's URL"),
  23         ('slicemgr',    'SFI_SM' ,      d_slicemgr,     '-m','--slicemgr',      "slice manager's URL"),
  24         ('aggmgr',      'SFI_AM',       d_aggmgr,       '-a','--aggregate',     "aggregate manager's URL"),
  25         ('verbose',     'SFACE_VERBOSE',False,          '-v','--verbose',       "UI verbosity"),
  26         ('debug',       'SFACE_DEBUG',  False,          '-d','--debug',         "UI debug flag"),
  27         ]
  28
  29     def fields (self):
  30         return [ tup[0] for tup in SfiConfig.supported ]
  31
  32     def field_labels (self):
  33         return [ (tup[0],tup[5]) for tup in SfiConfig.supported ]
  34
  35     def sfi_field (self, sfi):
  36         for tuple in SfiConfig.supported:
  37             if tuple[1]==sfi: return tuple[0]
  38         return None
  39
  40     def field_default (self, field):
  41         for tuple in SfiConfig.supported:
  42             if tuple[0]==field: return tuple[2]
  43         return None
  44
  45     # xxx todo - need for validators - not even sure this is still useful
  46     def define_accessors (self):
  47         for (field,sfi,default,_,__,___) in SfiConfig.supported:
  48             self.define_accessor (field,sfi,default)
  49
  50     def define_accessor (self,field,sfi,default):
  51         get_name="get" + field.capitalize();
  52         if not hasattr(SfiConfig,get_name):
  53             def get_call (self): return getattr(self,field)
  54             setattr (SfiConfig, get_name, get_call)
  55         set_name="set" + field.capitalize();
  56         if not hasattr(SfiConfig,set_name):
  57             def set_call (self, newvalue): setattr (self, field, newvalue)
  58             setattr (SfiConfig, set_name, set_call)
  59
  60     # the generic form of accessors
  61     def get(self,field): return getattr(self,field)
  62     def set(self,field,value): setattr(self,field,value)
  63
  64     def __init__(self):
  65         self.read_config()
  66
  67     def filename (self):
  68         return os.path.expanduser("~/.sfi/sfi_config")
  69
  70     def read_config(self):
  71         tmp={}
  72         try:
  73             execfile(self.filename(), tmp)
  74         except:
  75             print "Warning - no config file found %s"%self.filename()
  76             pass
  77         for (field,sfi,default,_,__,___) in SfiConfig.supported:
  78             if tmp.has_key(sfi):setattr(self,field,tmp[sfi])
  79             else:               setattr(self,field,default)
  80         self.display("After reading config from %s"%self.filename())
  81
  82     def display (self, msg):
  83         if self.debug:
  84             print msg
  85             for k in self.fields():
  86                 print "%-20s: %r"%(k,getattr(self,k))
  87
  88     def save_config(self):
  89         configfile = self.filename()
  90         tmpfile = configfile + ".tmp"
  91
  92         out = open(tmpfile, "w")
  93         lineno=0
  94         fields=self.fields()
  95         for line in open(configfile):
  96             lineno += 1
  97             try:
  98                 sfi, val = line.split('=')
  99                 sfi = sfi.strip()
 100                 val = val.strip()
 101                 field=self.sfi_field(sfi)
 102                 if field:
 103                     line = "%s = '%s'\n" % (sfi, getattr(self, field))
 104             except:
 105                 if self.debug:
 106                     import traceback
 107                     print 'line',lineno,'ignored',line
 108             out.write(line)
 109         out.close()
 110
 111         os.unlink(configfile)
 112         os.rename(tmpfile, configfile)
 113
 114     def add_options_to_OptionParser (self, parser):
 115         for (field,_,default,short,long,msg) in SfiConfig.supported:
 116             if default==True or default==False:
 117                 parser.add_option(short,long,dest=field,action="store_true",help=msg)
 118             else:
 119                 parser.add_option(short,long,dest=field,action="store",default=None, help=msg)
 120
 121     def update_from_OptionParser (self, optparse_options):
 122         for field in self.fields():
 123             if not hasattr(optparse_options,field) : continue
 124             value=getattr(optparse_options,field)
 125             if value is not None:
 126                 setattr(self,field,getattr(optparse_options,field))
 127
 128 #    def setUser(self, user):
 129 #        SfiConfig.SFI_USER = user
 130 #
 131 #        # Should probably get authority from user record instead...
 132 #        a = user.split('.')
 133 #        SfiConfig.SFI_AUTH = '.'.join(a[:len(a)-1])
 134
 135     def getSliceRSpecFile(self):
 136         return os.path.expanduser("~/.sfi/%s.rspec" % self.getSlice())
 137
 138
 139 # configuration singleton
 140 config = SfiConfig()
 141 config.define_accessors()