如何引发自定义词典KeyError消息

时间:2011-10-17 18:28:12

标签: python exception-handling dictionary python-2.4

我正在编写一些将由非python用户使用的脚本。我有一个包含字典的Config类,我希望能够为KeyError引发一个自定义异常。除了编写一个在检查字典中的值时引发异常的方法之外,还有一种优雅的方法吗?

以下是一个例子:

class ConfigError(Exception): pass

class Config:
    def __init__(self):
        self.cars = {'BMW': 'M5', 'Honda': 'Civic'}

    def get_car(self, name):
        try:
            return self.cars[name]
        except KeyError, e:
            raise ConfigError("Car %s doesn't exist!!!" % name)

conf = Config()
print 'Car: %s' % conf.cars['BMW']
print 'Car: %s' % conf.cars['Mercedes']

这是输出:

Car: M5 
Traceback (most recent call last):
  File "test.py", line 17, in ?
    print 'Car: %s ' % conf.cars['Mercedes']
KeyError: 'Mercedes'

我知道我可以编写一个像上面的Config.get_car()这样的方法并引发一个自定义异常,但是我希望能够在直接尝试访问Config.cars字典时引发自定义异常。我想这样做是因为配置中实际上有更多的字典,但我只需要为其中一个字典引发自定义异常,并希望保持数据访问的一致性(全部作为字典)。

注意:这是使用Python 2.4.4

1 个答案:

答案 0 :(得分:5)

考虑这样做。

class MyConfigDict( dict ):
    def __getitem__( self, key ):
        try:
            return super( MyConfigDict, self ).__getitem__( key )
        except KeyError as e:
            raise MyConfigError( e.message )

这样的事情可能允许你在你自己的扩展异常中包装每个访问。