Question

我想重构assign_hotkey函数，因为我不喜欢if语句中的重复代码。

代码：

n_path_i = 0
c_node_i = 1
h_key_i = 2

hotkeys = [
        ['Filter/Blur', 'Blur, size 20 label "hello world"', 'A'],
        ['Draw/LightWrap', 'LightWrap', 'B'],
        ['Draw/Grain', 'Grain2', 'X']
        ]

def assign_hotkey(n_path, c_node, h_key):
    c_node_splitted = c_node.split(',')
    if len(c_node_splitted) > 1:
        menu.addCommand(n_path, 
                        lambda: nuke.createNode(*c_node_splitted),
                        h_key)
    else:
        menu.addCommand(n_path, 
                        lambda: nuke.createNode(c_node),
                        h_key)

for i in hotkeys:
    assign_hotkey(i[n_path_i], i[c_node_i], i[h_key_i])

Answer 1

由于要解压缩将c_node传递给nuke.createNode时获得的列表，因此完全没有理由对列表的长度进行特殊处理，因为列表的长度为1。 / p>

您可以更改：

if len(c_node_splitted) > 1:
    menu.addCommand(n_path, 
                    lambda: nuke.createNode(*c_node_splitted),
                    h_key)
else:
    menu.addCommand(n_path, 
                    lambda: nuke.createNode(c_node),
                    h_key)

收件人：

menu.addCommand(n_path, 
                lambda: nuke.createNode(*c_node_splitted),
                h_key)

重构函数内部的代码重复

1 个答案: