Pass аргументов из ЦМДА в питон скрипт

Question

я пишу свои скрипты питона и запускать их с ЦМД, набрав:

C:\> python script.py 

Некоторые из моих сценариев содержат отдельные алгоритмы и методы, которые вызываются на основе флага. Теперь я хотел бы передать флаг через ЦМД непосредственно вместо того, чтобы идти в сценарий и изменить флаг перед бежать, я хочу что-то подобное:

C:\> python script.py -algorithm=2 

Я прочитал, что люди используют sys.argv для почти подобных целей, но, читая руководства и форумы, я не мог понять, как это работает.

Answer 1

Есть несколько модулей, специализирующихся на разборе аргументов командной строки: , optparse и argparse. optparse устарел, и getopt менее эффективен, чем argparse, поэтому я советую вам использовать последнее, это будет более полезно в долгосрочной перспективе.

Вот краткий пример:

import argparse 
# Define the parser 
parser = argparse.ArgumentParser(description='Short sample app') 
# Declare an argument (`--algo`), telling that the corresponding value should be stored in the `algo` field, and using a default value if the argument isn't given 
parser.add_argument('--algo', action="store", dest='algo', default=0) 
# Now, parse the command line arguments and store the values in the `args` variable 
args = parser.parse_args() 
# Individual arguments can be accessed as attributes... 
print args.algo 

Это должно вам начать. В худшем случае есть много документации, доступной в режиме онлайн (скажем, this one, например) ...

Answer 2

Пробуйте использовать модуль getopt. Он может обрабатывать как короткие и длинные параметры командной строки и осуществляется аналогичным образом на других языках (C, оболочки сценариев и т.д.):

import sys, getopt 


def main(argv): 

    # default algorithm: 
    algorithm = 1 

    # parse command line options: 
    try: 
     opts, args = getopt.getopt(argv,"a:",["algorithm="]) 
    except getopt.GetoptError: 
     <print usage> 
     sys.exit(2) 

    for opt, arg in opts: 
     if opt in ("-a", "--algorithm"): 
      # use alternative algorithm: 
      algorithm = arg 

    print "Using algorithm: ", algorithm 

    # Positional command line arguments (i.e. non optional ones) are 
    # still available via 'args': 
    print "Positional args: ", args 

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

Вы можете затем передать указать другой алгоритм, используя -a или --algorithm= варианты:

python <scriptname> -a2    # use algorithm 2 
python <scriptname> --algorithm=2 # ditto 

См: getopt documentation