Python параметр командной строки не поднимая ошибку, если аргумент ошибочно используется

Question

У меня есть следующий код Python, который имеет 1 командной строки необязательный параметр (c), который имеет аргумент и 2 варианта (a и b), которые не имеют аргумент:

import sys, getopt 

def main(argv): 
    inputfile = '' 
    outputfile = '' 
    try: 
     opts, args = getopt.getopt(argv,"abc:",["csvfile="]) 
    except getopt.GetoptError: 
     print 'Error in usage - a does not require an argument' 
     sys.exit(2) 
    for opt, arg in opts: 
     print "Raw input is: {}" .format(opt) 
     if opt in ("-c", "--csvfile"): 
     outputfile = arg 
     print 'Output file is {}' .format(outputfile) 
     elif opt == '-a': 
     print 'Alpha' 
     elif opt == '-b': 
     print 'Beta' 
     print 'User choice is {}' .format(opt.lstrip('-')) 

if __name__ == "__main__": 
    main(sys.argv[1:]) 

Когда я вхожу python readwritestore.py -a я получаю:

Raw input is: -a 
Alpha 
User choice is a 

Это то, что я надеялся, если аргумент командной строки -a. Однако, если я вхожу python readwritestore.py -a csvfile_name, то я получаю:

Raw input is: -a 
Alpha 
User choice is a 

Это не то, что я предназначен. В этой функции c является единственным вариантом, который запрашивает аргумент. Если я вхожу a с аргументом, код должен дать сообщение об ошибке, что я настроить

Error in usage - a does not require an argument 

Это не происходит для a или b. Это позволяет вводить аргумент без повышения ошибки.

Если параметры, которые не требуют аргумента, вводятся с аргументом, то я хотел бы, чтобы он вызывал ошибку. python readwritestore.py -a text и python readwritestore.py -b text следует поднять с ошибкой Error in usage - a does not require an argument.

Есть ли способ указать это? Правильно ли это делается getopt()?

Дополнительная информация:

Я только хочу python readwritestore.py -c text работать с аргументом. Для остальных 2 опций, a и b, код должен поднять ошибку.

Answer 1

проверка размера sys.argv (список аргумента, переданный при вызове сценария) может помочь вам проверить, что:

Я знаю, что это не то, что вы просили (argparse), но вот как ты мог бы сделать это с помощью argparse:

from argparse import * 

def main(): 
    parser = ArgumentParser() 
    parser.add_argument('-c', '--csvfile', help='do smth with cvsfile') 
    parser.add_argument(
     '-a', '--Alpha', help='Alpha', action='store_true') 
    parser.add_argument(
     '-b', '--Beta', help='beta smth', action='store_true') 
    if args.csvfile: 
     print 'Output file is {}' .format(args.csvfile) 
    if args.Alpha: 
     print 'Alpha' 
    if args.Beta: 
     print 'Beta' 

if __name__ == "__main__": 
    main() 

Это вызовет ошибку для многих аргументов. (также python readwritestore.py -h отобразит справку точно так же, как человек в unix)