sface/config.py

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