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     if os.getenv("DAK_CONFIG"):
  43         return os.getenv("DAK_CONFIG")
  44     else:
  45         return default_config
  46
  47 class Config(Singleton):
  48     """
  49     A Config object is a singleton containing
  50     information about the DAK configuration
  51     """
  52     def __init__(self, *args, **kwargs):
  53         super(Config, self).__init__(*args, **kwargs)
  54
  55     def _readconf(self):
  56         apt_pkg.init()
  57
  58         self.Cnf = apt_pkg.newConfiguration()
  59
  60         apt_pkg.ReadConfigFileISC(self.Cnf, which_conf_file())
  61
  62         # Check whether our dak.conf was the real one or
  63         # just a pointer to our main one
  64         res = socket.gethostbyaddr(socket.gethostname())
  65         conffile = self.Cnf.get("Config::" + res[0] + "::DakConfig")
  66         if conffile:
  67             apt_pkg.ReadConfigFileISC(self.Cnf, conffile)
  68
  69         # Rebind some functions
  70         # TODO: Clean this up
  71         self.get = self.Cnf.get
  72         self.SubTree = self.Cnf.SubTree
  73         self.ValueList = self.Cnf.ValueList
  74         self.Find = self.Cnf.Find
  75         self.FindB = self.Cnf.FindB
  76
  77     def _startup(self, *args, **kwargs):
  78         self._readconf()
  79
  80     def has_key(self, name):
  81         return self.Cnf.has_key(name)
  82
  83     def __getitem__(self, name):
  84         return self.Cnf[name]
  85
  86     def __setitem__(self, name, value):
  87         self.Cnf[name] = value