Question

我正在重构朋友写的一些代码，最近偶然发现了这个函数：

def setup_parameters(self, data):
    '''Parse raw data to determine game settings.'''
    for line in data.split('\n'):
      line = line.strip().lower()
      if line:
        tokens = line.split()

        self.L.debug("tokens: " + str(tokens))

        key = tokens[0]
        if key == 'cols':
          self.width = int(tokens[1])
        elif key == 'rows':
          self.height = int(tokens[1])
        elif key == 'player_seed':
          random.seed(int(tokens[1]))
        elif key == 'turntime':
          self.turntime = int(tokens[1])
        elif key == 'loadtime':
          self.loadtime = int(tokens[1])
        elif key == 'viewradius2':
          self.viewradius2 = int(tokens[1])
        elif key == 'attackradius2':
          self.attackradius2 = int(tokens[1])
        elif key == 'spawnradius2':
          self.spawnradius2 = int(tokens[1])

正如你所看到的，这里有一种讨厌的切换语句，显然需要一个字典。我很想把它写成类字典，因为键是常量，但由于键映射到实例的属性（即'cols'：self.width），因此无法编译。

那么我的问题是，重构这些代码的正确方法是什么？

Answer 1

将键映射到属性的名称，并使用setattr(self, attribute_name, int(tokens[1])设置值。 E.g：

attribute_dict = dict(cols="width", rows="height", turntime="turntime", ...)
[...]
value = int(tokens[1])
if key == "player_seed":
    random.seed(value)
else:
    setattr(self, attribute_dict[key], value)

Answer 2

您可以使用键作为访问器和lambda函数来构建字典，以执行每个键的代码。

Answer 3

使用

等操作设置dict

actions = dict(cols = lambda tokens: setattr(self, "width", int(tokens[1]), ... 
               player_seed = lambda tokens: random.seed(int(tokens[1]))
              )

然后：

 actions[key](tokens)

如何重构Python“switch语句”

3 个答案: