Como acesso argumentos de linha de comando no Python?

Eu uso o python para criar as configurações do meu projeto, mas preciso de ajuda para obter os argumentos da linha de comando.

Eu tentei isso no terminal:

$python myfile.py var1 var2 var3

No meu arquivo Python, quero usar todas as variáveis ​​que são inseridas.

O tutorial do Python explica :

import sys

print(sys.argv)

Mais especificamente, se você executar python example.py one two three:

>>> import sys
>>> print(sys.argv)
['example.py', 'one', 'two', 'three']

import sys

sys.argv[1:]

fornecerá uma lista de argumentos (sem incluir o nome do arquivo python)

Eu recomendo o argparseque vem com o Python 2.7 e posterior.

O argparsemódulo reduz o código da placa da caldeira e torna seu código mais robusto, porque o módulo lida com todos os casos de uso padrão (incluindo subcomandos), gera a ajuda e o uso para você, verifica e higieniza a entrada do usuário - tudo o que você precisa se preocupar quando precisar estão usando sys.argvabordagem. E é de graça (embutido).

Aqui está um pequeno exemplo:

import argparse

parser = argparse.ArgumentParser("simple_example")
parser.add_argument("counter", help="An integer will be increased by 1 and printed.", type=int)
args = parser.parse_args()
print(args.counter + 1)

e a saída para python prog.py -h

usage: simple_example [-h] counter

positional arguments:
  counter     counter will be increased by 1 and printed.

optional arguments:
  -h, --help  show this help message and exit

e python prog.py 1como você esperaria:

2

Código Python:

import sys

# main
param_1= sys.argv[1] 
param_2= sys.argv[2] 
param_3= sys.argv[3]  
print 'Params=', param_1, param_2, param_3

Invocação:

$python myfile.py var1 var2 var3

Resultado:

Params= var1 var2 var3 

Você pode usar sys.argvpara obter os argumentos como uma lista.

Se você precisar acessar elementos individuais, poderá usar

sys.argv[i]  

onde iestá index, 0fornece o nome do arquivo python que está sendo executado. Qualquer índice depois disso são os argumentos passados.

Se você chamar assim: $ python myfile.py var1 var2 var3

import sys

var1 = sys.argv[1]
var2 = sys.argv[2]
var3 = sys.argv[3]

Semelhante às matrizes que você também possui, sys.argv[0]que é sempre o diretório de trabalho atual.

Algumas coisas adicionais em que consigo pensar.

Como o @allsyed disse, o sys.argv fornece uma lista de componentes (incluindo o nome do programa); portanto, se você quiser saber o número de elementos transmitidos pela linha de comando, use len () para determiná-lo. Com base nisso, você pode criar mensagens de exceção / erro se o usuário não passar um número específico de parâmetros.

Além disso, se você estiver procurando uma maneira melhor de lidar com argumentos de linha de comando, sugiro que você consulte https://docs.python.org/2/howto/argparse.html