can grab tags from a build-dir/tags-file, or use the ones provided on
[sface.git] / 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()