pyx/config.py

   1 # -*- coding: ISO-8859-1 -*-
   2 #
   3 #
   4 # Copyright (C) 2003-2004 Jörg Lehmann <joergl@users.sourceforge.net>
   5 # Copyright (C) 2003-2005 André Wobst <wobsta@users.sourceforge.net>
   6 #
   7 # This file is part of PyX (http://pyx.sourceforge.net/).
   8 #
   9 # PyX is free software; you can redistribute it and/or modify
  10 # it under the terms of the GNU General Public License as published by
  11 # the Free Software Foundation; either version 2 of the License, or
  12 # (at your option) any later version.
  13 #
  14 # PyX is distributed in the hope that it will be useful,
  15 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  16 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17 # GNU General Public License for more details.
  18 #
  19 # You should have received a copy of the GNU General Public License
  20 # along with PyX; if not, write to the Free Software
  21 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA
  22
  23 import ConfigParser, os.path, warnings
  24 import siteconfig
  25
  26 cflist = [os.path.join(siteconfig.pyxrcdir, "pyxrc"),  os.path.expanduser("~/.pyxrc")]
  27
  28 config = ConfigParser.ConfigParser()
  29 config.read(cflist)
  30
  31 def get(section, option, default):
  32     try:
  33         return config.get(section, option)
  34     except:
  35         return default
  36
  37 def getint(section, option, default):
  38     try:
  39         return config.getint(section, option)
  40     except:
  41         return default
  42
  43 def getfloat(section, option, default):
  44     try:
  45         return config.getfloat(section, option)
  46     except:
  47         return default
  48
  49 def getboolean(section, option, default):
  50     try:
  51         return config.getboolean(section, option)
  52     except:
  53         return default
  54
  55 def getlist(section, option, default):
  56     try:
  57         return config.get(section, option).split()
  58     except:
  59         return default
  60
  61
  62 formatWarnings = get("general", "warnings", "default")
  63 if formatWarnings not in ["default", "short", "shortest"]:
  64     raise RuntimeError("invalid config value for option 'warnings' in section 'general'")
  65 if formatWarnings != "default":
  66     def formatwarning(message, category, filename, lineno, line=None):
  67         if formatWarnings == "short":
  68             return "%s:%s: %s: %s\n" % (filename, lineno, category.__name__, message)
  69         else:
  70             return "%s\n" % message
  71     warnings.formatwarning = formatwarning