daklib/config.py

   1 #!/usr/bin/env python
   2
   3 """
   4 Config access class
   5
   6 @contact: Debian FTPMaster <ftpmaster@debian.org>
   7 @copyright: 2008  Mark Hymers <mhy@debian.org>
   8 @license: GNU General Public License version 2 or later
   9 """
  10
  11 # This program is free software; you can redistribute it and/or modify
  12 # it under the terms of the GNU General Public License as published by
  13 # the Free Software Foundation; either version 2 of the License, or
  14 # (at your option) any later version.
  15
  16 # This program is distributed in the hope that it will be useful,
  17 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  18 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19 # GNU General Public License for more details.
  20
  21 # You should have received a copy of the GNU General Public License
  22 # along with this program; if not, write to the Free Software
  23 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  24
  25 ################################################################################
  26
  27 # <NCommander> mhy, how about "Now with 20% more monty python references"
  28
  29 ################################################################################
  30
  31 import os
  32 import apt_pkg
  33 import socket
  34
  35 from singleton import Singleton
  36
  37 ################################################################################
  38
  39 default_config = "/etc/dak/dak.conf" #: default dak config, defines host properties
  40
  41 def which_conf_file():
  42     return os.getenv("DAK_CONFIG", default_config)
  43
  44 class Config(Singleton):
  45     """
  46     A Config object is a singleton containing
  47     information about the DAK configuration
  48     """
  49     def __init__(self, *args, **kwargs):
  50         super(Config, self).__init__(*args, **kwargs)
  51
  52     def _readconf(self):
  53         apt_pkg.init()
  54
  55         self.Cnf = apt_pkg.newConfiguration()
  56
  57         apt_pkg.ReadConfigFileISC(self.Cnf, which_conf_file())
  58
  59         # Check whether our dak.conf was the real one or
  60         # just a pointer to our main one
  61         res = socket.gethostbyaddr(socket.gethostname())
  62         conffile = self.Cnf.get("Config::" + res[0] + "::DakConfig")
  63         if conffile:
  64             apt_pkg.ReadConfigFileISC(self.Cnf, conffile)
  65
  66         # Rebind some functions
  67         # TODO: Clean this up
  68         self.get = self.Cnf.get
  69         self.SubTree = self.Cnf.SubTree
  70         self.ValueList = self.Cnf.ValueList
  71         self.Find = self.Cnf.Find
  72         self.FindB = self.Cnf.FindB
  73
  74     def _startup(self, *args, **kwargs):
  75         self._readconf()
  76
  77     def has_key(self, name):
  78         return self.Cnf.has_key(name)
  79
  80     def __getitem__(self, name):
  81         return self.Cnf[name]
  82
  83     def __setitem__(self, name, value):
  84         self.Cnf[name] = value