Argparse：兩個位置參數與nargs ='+'

Question

我試圖做圖像之間的數學運算。我已經定義（簡化了我真正的代碼版本）：

parser = argparse.ArgumentParser(description='Arithmetic operations on images') 
parser.add_argument("input1", metavar='input1', action='store', 
     help='list of input images to operate with', nargs="+", type=str) 
parser.add_argument("operation", metavar='operation', action='store', type=str, 
     help='type of operation (+,-,*,/) to be done', nargs=1) 
parser.add_argument("input2",metavar='input2', action='store', nargs="+", 
     type=str, help='image (or value) with which to perform the operation on input1') 

此代碼，生產：

arith.py -h 
usage: arith.py [-h] input1 [input1 ...] operation input2 [input2 ...] 

它確實是這樣理解的是輸入1可以包含一個或多個元素，操作將是一個單一的一個，而input2可以是任意數量的元素。

問題當然是，有兩個位置參數與未確定數量的元素，argparse混淆了什麼是什麼。我嘗試在'操作'中添加選項= [「+」，「 - 」，「*」，「/」]，以便它知道在哪裏做分離，但似乎argparse無法做到。實際上，argparse文檔中，談到NARGS =「*」你可以閱讀：

需要注意的是，一般沒有太大的意義，有NARGS =不止一個位置參數「*」

我以爲我可以將args.input1，args.operation和args.input2加在一起，並將自己分開尋找「+」，「 - 」，「/」，「*」，但在做一件非常醜陋的事情之前，我會想想要挖掘集體思想。

Answer 1

如果您使用「必需的可選」參數，您可以使這項工作。如果需要命令行上的選項來顯示從左操作數切換到操作時更改爲正確的操作數。例如。其餘的 -

import argparse 

parser = argparse.ArgumentParser(description='Process some integers.') 
parser.add_argument("--left-operands", "-l", nargs="+", required=True) 
parser.add_argument("--right-operands", "-r", nargs="+", required=True) 
parser.add_argument("--operation", "-o", choices=["+", "-", "*", "/"], required=True) 

argline = "-l 1 2 -o + -r 3 4" 

print(parser.parse_args(argline.split(" "))) 

Answer 2

當分配字符串positionals，解析器只有那些用一個前綴字符（「」例如）開始之間進行區分。它無法區分表示「數字」的字符串和表示「操作」的字符串。實際上，它執行此正則表達式操作：

re.match('(A+)(A)(A+)','AAAAAAAAA') 

這將產生(AAAAAA),(A),(A)。它爲最後兩組分配足夠的字符串以滿足其規格，並將其餘的分配給第一組。

所以你需要某種'標誌'來標記第一個列表的結束。

這是，我認爲，最接近你會得到​​：

parser.add_argument("input1", nargs="+", type=int) 
parser.add_argument("-o", "--operation", choices=['+','minus','*','/']) 
parser.add_argument("input2", nargs="+", type=int) 

應該把

PROG 1 3 4 -o + 5 6 7 
PROG 1 3 4 -o+ 5 6 7 
PROG 1 3 4 --operation=+ 5 6 7 

到（我認爲）

namespace(input1=[1,3,4], operation='+', input2=[5,6,7]) 

注意choices的列表不包括' - '。這是因爲解析器將它視爲prefix_char。可能有偷偷摸摸的方式作爲參數價值，但我不會花時間來找到它。

我在解析器中將input1的值轉換爲整數。之後你可以做到這一點。當然，製作花車代替。

我省略了默認參數，如type=str,action='store'。

---

但也許更好的解決方案是將所有的值作爲1列表，並自己分裂。至少有了這3個參數，你沒有太多的使用權力。

alist = ['1','2','3','+','4','5','6'] 
i = <find index of '+-/*'> 
input1 = alist[:i] 
operations = alist[i] 
input2 = alsits[i+1:]