Рефакторинг дублирования кода внутри функции - PullRequest
Новая
       23

Рефакторинг дублирования кода внутри функции

0 голосов
/ 17 октября 2018

Я хочу реорганизовать функцию 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])

1 Ответ

0 голосов
/ 17 октября 2018

Поскольку вы распаковываете список, полученный при разбиении c_node при передаче его на nuke.createNode, нет никаких оснований делать особый случай, если длина списка равна 1.

Вы можете изменить: 

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)
...