如果在python中替换elif块

时间:2011-03-11 05:28:15

标签: python conditional

事实证明,这个条件块在我的代码中不断重复。还有其他方法让我的生活更轻松吗?当然,为条件执行的正文不同。

if self.datatype == "string":
    t = "z"
elif self.datatype == "double":
    t = "d"
elif self.datatype == "number":
    t = "i"
elif self.datatype == "blob":
    t = "z"
else:
    raise EntParEx("Unknown datatype" + self.datatype)

......更多代码使用相同的条件

def emit_cpp_def(self):
    s = ""
    e = ""
    if self.datatype == "string":
        s += "static const int " + self.lenvar + " = " + self.length + ";"
        s += "\nchar"
        e += "[" + self.lenvar + " + 2" + "]"
    elif self.datatype == "double":
        s += "double"
    elif self.datatype == "number":
        s += "int"
    elif self.datatype == "blob":
        s += "char*"
    else:
        raise EntParEx("Unknown datatype" + self.datatype)

    s += " " + self.cpp_membername;
    s += e
    s += ";"
    return s;

def emit_cursor_def_code(self):
    if self.datatype == "blob":
        return ""

    ret = "nvl(" + self.db_fieldname + ", "
    #TODO: Add default value loading!
    if self.datatype == "string":
        ret += "\' \'"
    elif self.datatype == "double":
        ret += "-1.0"
    elif self.datatype == "number":
        ret += "-1"
    else:
        raise EntParEx("Unknown datatype" + self.datatype)
    ret += "), "
    return ret

编辑: 我认为我需要的是为每种类型运行特定功能。不幸的是,我不熟悉python。可以这样做吗?即。

switch_datatype(function_string(), function_integer(), ...etc)

这更糟吗?

3 个答案:

答案 0 :(得分:15)

如果它是完全相同的条件,请将其粘贴在方法中并在需要的地方调用它。否则,在某处定义字典并使用您需要的任何字典。

datatypes = {'string': 'z', 'double': 'd', 'number': 'i', 'blob': 'z'}
t = datatypes[self.datatype]

您可能会捕获KeyError并引发域异常。

答案 1 :(得分:4)

在@ jleedev的回答的基础上,如果你想要一个自定义异常,你真的会这么做,

class EntParEx(KeyError): pass
class DataMapping(dict):
    def __missing__(self, key):
        raise EntParEx("unknown datatype {}".format(key))


>>> datatypes = DataMapping(string='z', double='d', number='i', blob='z')
>>> datatypes['string']
'z'
>>> datatypes['other']
Traceback (most recent call last):
  File "<pyshell#13>", line 1, in <module>
    datatypes['other']
  File "<pyshell#8>", line 3, in __missing__
    raise EntParEx("unknown datatype {}".format(key))
EntParEx: 'unknown datatype other'

编辑以添加扩展示例的代码:

>>> datatypes = DataMapping(string='static const int {lenvar} = {length};\nchar {cpp_membername}[{lenvar}+2];',
        double='double {cpp_membername};',
        number='int {cpp_membername};',
        blob='char* {cpp_membername};')
>>> inst = C()
>>> inst.lenvar = 'a'
>>> inst.length = 5
>>> inst.cpp_membername='member'
>>> datatypes['number'].format(**vars(inst))
'int member;'
>>> datatypes['string'].format(**vars(inst))
'static const int a = 5;\nchar member[a+2];'

答案 2 :(得分:1)

你也可以做一些更有活力的事情。

type(self.datatype).__name__

返回“str”,“float”,“int”等。你可以取第一个字符。