andersk / sql.git / blame
| Commit | Line | Data |
|---|---|---|
| e4fd5e1b JP |
1 | """ |
| 2 | Global configuration objects. | |
| 3 | ||
| 4 | Joe Presbrey <presbrey@mit.edu> | |
| 5 | """ | |
| 6 | ||
| 7 | from ConfigParser import ConfigParser, NoSectionError, NoOptionError | |
| 8 | import os | |
| 9 | ||
| 10 | _ENV_BASE='_SQL_MIT_EDU' | |
| 11 | ||
| 12 | class Section(object): | |
| 13 | """Configuration section namespace.""" | |
| 14 | def __init__(self, d): | |
| 15 | self.items = d | |
| 16 | ||
| 17 | def __getattr__(self, k): | |
| 18 | v = self.items.get(k, None) | |
| 19 | try: | |
| 20 | if str(int(v)) == str(v): v = int(v) | |
| 21 | except TypeError: pass | |
| 22 | except ValueError: pass | |
| 23 | try: | |
| 24 | if str(v).lower() == 'true': v = True | |
| 25 | if str(v).lower() == 'false': v = False | |
| 26 | except TypeError: pass | |
| 27 | except ValueError: pass | |
| 28 | return v | |
| 29 | ||
| 30 | def __iter__(self): | |
| 31 | return iter(self.items) | |
| 32 | ||
| 33 | def __str__(self): | |
| 34 | return str(self.items) | |
| 35 | ||
| 36 | class Config(object): | |
| 37 | """Base configuration namespace.""" | |
| 38 | def __init__(self, filename, *args, **kwargs): | |
| 39 | self._cp = ConfigParser(*args, **kwargs) | |
| 40 | self.read(filename) | |
| 41 | self._none = Section({}) | |
| 42 | ||
| 43 | def _samefile(self, f1, f2): | |
| 44 | try: | |
| 45 | return os.path.samefile(f1, f2) | |
| 46 | except OSError, e: | |
| 47 | return False | |
| 48 | ||
| 49 | def read(self, filename): | |
| 50 | config_path = [os.path.join(x, filename) for x in [ | |
| 51 | os.path.dirname(__file__), | |
| 52 | '/etc']] | |
| 53 | if _ENV_BASE in os.environ: | |
| 54 | config_path.insert(1, os.path.join(os.environ[_ENV_BASE], 'etc', filename)) | |
| 55 | def append(path): | |
| 56 | r = os.path.join(path, filename) | |
| 57 | if not r in config_path: | |
| 58 | config_path.append(r) | |
| 59 | if 'HOME' in os.environ: | |
| 60 | append(os.environ['HOME']) | |
| 61 | if 'PWD' in os.environ: | |
| 62 | append(os.environ['PWD']) | |
| 63 | self._cp.read(config_path) | |
| 64 | self._sections = dict(map(lambda x: (x, Section(dict(self._cp.items(x)))), | |
| 65 | self._cp.sections())) | |
| 66 | ||
| 67 | def get(self, *av, **kw): | |
| 68 | try: return self._cp.get(*av, **kw) | |
| 69 | except NoSectionError: return None | |
| 70 | except NoOptionError: return None | |
| 71 | ||
| 72 | def __getattr__(self, k): | |
| 73 | return self._sections.get(k, self._none) | |
| 74 | ||
| 75 | def __str__(self): | |
| 76 | d = {} | |
| 77 | for x in self._cp.sections(): | |
| 78 | d[x] = dict(self._cp.items(x)) | |
| 79 | return str(d) | |
| 80 | ||
| 81 | if __name__ == '__main__': | |
| 82 | import mitsql | |
| 83 | print mitsql.config |