Python 2.7：如何傳遞獨立/模塊代碼的選項

Question

我已經編寫了一個python模塊mymod.py，它也可以用作命令行中的獨立程序。

在mymod.py中，我定義了幾個函數（其中默認值是使用關鍵字設置的）， 和if __name__=="__main__"塊將模塊用作獨立程序。

我想覆蓋一些默認選項的可能性，因此在主程序I import argparse中並使用它來分析選項。我使用字典來存儲 的默認值，這樣如果有一天我需要更改默認值，我可以很容易地將其修改爲只在一個地方的值 。

它的工作原理，但我發現代碼不是「乾淨」，並認爲可能我沒有以適當的pythonic方式做它。

這是一個玩具爲例，說明我做的：

#!/usr/bin/env python 
#mymod.py 

__default_options__={ 
    "f1_x":10, 
    "f2_x":10 
} 

def f1(x=__default_options__["f1_x"]): 
    return x**2 

def f2(x=__default_options__["f2_x"]): 
    return x**4 

# this function is the "core" function which uses f1 and f2 
# to produce the main task of the program 
def f(x=__default_options__["f1_x"],y=__default_options__["f2_x"]): 
    return f1(x)+f2(y) 

if __name__=="__main__": 
    import argparse 
    parser = argparse.ArgumentParser(description = "A toy application") 
    parser.add_argument("--f1-x",help="the parameter passed to f1", 
    default=__default_options__["f1_x"], type = float,dest = "x") 
    parser.add_argument("--f2-x",help="the parameter passed to f2", 
    default=__default_options__["f2_x"], type = float, dest = "y") 

    options= parser.parse_args() 
    print f(options.x,options.y) 

像我這樣做是有點麻煩，可能針對的Python和argparse都精神傳遞的默認值。

這段代碼如何改進以使其變得更加pythonic，並且最好使用argparse？

Answer 1

可以使用`ArgumentParser.set_defaults方法，通過以下方式

default_options={ 
    "x":10, 
    "y":10 
} 

def f1(**kwargs): 
    x=kwargs.get('x', defalut_options['x']) 
    return x**2 

def f2(**kwargs): 
    y=kwargs.get('y', defalut_options['y']) 
    return x**4 

def f(**kwargs): 
    x=kwargs.get('x', defalut_options['x']) 
    y=kwargs.get('y', defalut_options['y']) 
    return f1(x=x, y=y) 

if __name__=="__main__": 
    import argparse 
    parser = argparse.ArgumentParser(description = "A toy application", formatter_class=argparse.ArgumentDefaultsHelpFormatter) 
    parser.add_argument("--f1-x",help="the parameter passed to f1", 
     type = float,dest = "x") 
    parser.add_argument("--f2-x",help="the parameter passed to f2", 
     type = float, dest = "y") 

    parser.set_defaults(**default_options) 

    options= parser.parse_args() 
    print f(options.x,options.y) 

我花了一段時間，使其工作，因爲我沒有注意到你正在使用dest在add_argument（我從來沒有使用它）。如果未提供此關鍵字，argparse將默認值dest設置爲參數的長名稱（在此例中爲f1_x和f2_x，因爲它用-替換爲_）。爲了說明問題：如果您想提供默認字典，則需要匹配dest（如果提供）。此外，請注意parser.set_defaults只是爲解析器添加參數，所以如果您的解析器中沒有某些條目，它將被添加到Namespace。

--Edited仿製kwargs添加到functions--

Answer 2

由於@Francesco在評論中寫道，您的默認詞典將無法正常工作，你可能預期：該功能將保留他們在加載默認值該模塊，而不管以後對字典的更改。以下是如何讓他們跟蹤字典的當前值：

_default_options = { 
    "f1_x":10, 
    "f2_x":10 
} 

def f1(x=None): 
    if x == None: 
     x = _default_options["f1_x"] 
    ... 

您可以然後通過ArgumentParser修改_default_options，或以任何其他方式，並f1()會使用它，如果使用任何參數調用。

這要求None永遠不會成爲x的有意義的值;如果情況並非如此，請選擇一個合適的不可能值。