Python 在线切换

Question

我正在寻找一种更优雅的方式来执行以下操作:

py_loglevel = logging.DEBUG
if self.loglevel == "INFO":
   py_loglevel = logging.INFO
elif self.loglevel == "WARNING":
   py_loglevel = logging.WARNING
elif self.loglevel == "ERROR":
   py_loglevel = logging.ERROR

这当然可行，但不知何故让我很恼火，因为它使我的代码变得困惑并且吸引了我的注意力，尽管它确实微不足道。这可以在 oneliner 中完成吗？

Answer 1

使用字典:

levels = {
    'INFO': logging.INFO, 
    'WARNING': logging.WARNING,
    'ERROR': logging.ERROR,
}
py_loglevel = levels.get(self.loglevel, logging.DEBUG)

或者，查找名称作为模块的属性:

py_loglevel = getattr(logging, self.loglevel, logging.DEBUG)

前提是名称与模块上定义的名称相匹配。