it-swarm-pt.tech

Argumentos de entrada e de linha de comando do usuário

Como eu tenho um script Python que a) pode aceitar a entrada do usuário e como faço isso b) ler em argumentos se executado a partir da linha de comando?

465
Teifion

Para ler a entrada do usuário, você pode tentar o módulo cmdname__ para criar facilmente um mini-interpretador de linha de comando (com textos de ajuda e autocompletar) e raw_input ( inputNAME_ for Python 3+) para ler um linha de texto do usuário.

text = raw_input("Prompt")  # Python 2
text = input("Prompt")  # Python 3

Entradas de linha de comando estão em sys.argv. Tente isso no seu script:

import sys
print (sys.argv)

Existem dois módulos para analisar as opções da linha de comandos:  optparsename__ (depreciado desde o Python 2.7, use argparseNAME_ em vez disso) e getoptNAME_ . Se você quiser apenas inserir arquivos em seu script, observe o poder de fileinputNAME_ .

A Referência da biblioteca Python é sua amiga.

416
Antti Rasinen
var = raw_input("Please enter something: ")
print "you entered", var

Ou para o Python 3:

var = input("Please enter something: ")
print("You entered: " + var)
444
lbz

raw_input não está mais disponível no Python 3.x. Mas raw_input foi renomeado para input, então a mesma funcionalidade existe.

input_var = input("Enter something: ")
print ("you entered " + input_var) 

Documentação da mudança

201
steampowered

A melhor maneira de processar argumentos de linha de comando é o módulo argparse.

Use raw_input() para obter a entrada do usuário. Se você importar o readline module seus usuários terão edição de linha e histórico.

31
Dave Webb

Cuidado para não usar a função input, a menos que você saiba o que está fazendo. Ao contrário de raw_input, input aceitará qualquer expressão python, então é como eval

18
Vhaerun

Este programa simples ajuda você a entender como alimentar a entrada do usuário a partir da linha de comando e mostrar ajuda na transmissão de argumentos inválidos.

import argparse
import sys

try:
     parser = argparse.ArgumentParser()
     parser.add_argument("square", help="display a square of a given number",
                type=int)
    args = parser.parse_args()

    #print the square of user input from cmd line.
    print args.square**2

    #print all the sys argument passed from cmd line including the program name.
    print sys.argv

    #print the second argument passed from cmd line; Note it starts from ZERO
    print sys.argv[1]
except:
    e = sys.exc_info()[0]
    print e

1) Para encontrar a raiz quadrada de 5

C:\Users\Desktop>python -i emp.py 5
25
['emp.py', '5']
5

2) Passando argumento inválido diferente de número

C:\Users\bgh37516\Desktop>python -i emp.py five
usage: emp.py [-h] square
emp.py: error: argument square: invalid int value: 'five'
<type 'exceptions.SystemExit'>
13
Viswesn

Use 'raw_input' para entrada de um console/terminal.

se você quer apenas um argumento de linha de comando como um nome de arquivo ou algo assim, por exemplo.

$ python my_prog.py file_name.txt

então você pode usar sys.argv ...

import sys
print sys.argv

sys.argv é uma lista onde 0 é o nome do programa, portanto, no exemplo acima, sys.argv [1] seria "file_name.txt"

Se você quiser ter todas as opções de linha de comando, use o módulo optparse.

Pev

10
Simon Peverett

Se você estiver executando o Python <2.7, precisará optparse , que, como explica o documento, criará uma interface para os argumentos da linha de comando que são chamados quando seu aplicativo é executado.

No entanto, em Python ≥2,7, optparse foi preterido e foi substituído por argparse como mostrado acima. Um exemplo rápido dos documentos ...

O código a seguir é um programa Python que pega uma lista de inteiros e produz a soma ou o máximo:

import argparse

parser = argparse.ArgumentParser(description='Process some integers.')
parser.add_argument('integers', metavar='N', type=int, nargs='+',
                   help='an integer for the accumulator')
parser.add_argument('--sum', dest='accumulate', action='store_const',
                   const=sum, default=max,
                   help='sum the integers (default: find the max)')

args = parser.parse_args()
print args.accumulate(args.integers)
9
Matt Olan

Se é uma versão 3.x então simplesmente use:

variantname = input()

Por exemplo, você deseja inserir 8:

x = input()
8

x será igual a 8, mas será uma string, exceto se você definir de outra forma.

Então você pode usar o comando convert, como:

a = int(x) * 1.1343
print(round(a, 2)) # '9.07'
9.07
6
CorpseDead

A partir do Python 3,2 2.7, há agora argparse para processar argumentos de linha de comando.

6
GreenMatt

No Python 2:

data = raw_input('Enter something: ')
print data

Em Python 3:

data = input('Enter something: ')
print(data)
2
Mark
import six

if six.PY2:
    input = raw_input

print(input("What's your name? "))
1
Will Charlton