Пользовательский ввод и аргументы командной строки - PullRequest
471 голосов
/ 16 сентября 2008

Как у меня есть скрипт Python, который а) может принимать пользовательский ввод и как мне сделать его б) читать аргументы, если он запускается из командной строки?

Ответы [ 12 ]

447 голосов
/ 16 сентября 2008
var = raw_input("Please enter something: ")
print "you entered", var

Или для Python 3:

var = input("Please enter something: ")
print("You entered: " + var)
423 голосов
/ 16 сентября 2008

Для чтения пользовательского ввода вы можете попробовать модуль cmd для простого создания интерпретатора мини-командной строки (с текстами справки и автозаполнением) и raw_input (input для Python 3+) для чтения строки текста от пользователя.

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

Входные данные командной строки находятся в sys.argv. Попробуйте это в вашем скрипте:

import sys
print (sys.argv)

Существует два модуля для разбора параметров командной строки: optparse (устарело с Python 2.7, вместо этого используйте argparse) и getopt. Если вы просто хотите ввести файлы в свой скрипт, вот сила fileinput.

Справочник по библиотеке Python - ваш друг.

201 голосов
/ 01 декабря 2011

raw_input больше не доступен в Python 3.x. Но raw_input был переименован в input, поэтому такая же функциональность существует.

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

Документация об изменении

31 голосов
/ 16 сентября 2008

Лучший способ обработки аргументов командной строки - это модуль argparse.

Используйте raw_input(), чтобы получить пользовательский ввод. Если вы импортируете readline module, ваши пользователи будут иметь редактирование строк и историю.

18 голосов
/ 16 сентября 2008

Осторожно, не используйте функцию input, если вы не знаете, что делаете. В отличие от raw_input, input примет любое выражение Python, поэтому оно похоже на eval

13 голосов
/ 20 мая 2015

Эта простая программа поможет вам понять, как вводить пользовательский ввод из командной строки и показать справку о передаче неверного аргумента.

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) Чтобы найти квадратный корень из 5

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

2) Передача неверного аргумента, отличного от числа

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'>
10 голосов
/ 16 сентября 2008

Используйте raw_input для ввода с консоли / терминала.

если вам нужен аргумент командной строки, например имя файла или что-то подобное, например

$ python my_prog.py file_name.txt

тогда вы можете использовать sys.argv ...

import sys
print sys.argv

sys.argv - это список, где 0 - это имя программы, поэтому в приведенном выше примере sys.argv [1] будет "file_name.txt"

Если вы хотите иметь полные параметры командной строки, используйте модуль optparse.

п

9 голосов
/ 26 октября 2012

Если вы используете Python <2.7, вам нужно <a href="http://docs.python.org/library/optparse.html" rel="noreferrer"> optparse , который, как объясняет документ, создаст интерфейс с аргументами командной строки, которые вызываются при запуске приложения.

Однако в Python ≥2.7 функция optparse устарела и была заменена на argparse , как показано выше. Быстрый пример из документов ...

Следующий код представляет собой программу на Python, которая принимает список целых чисел и выдает либо сумму, либо максимум:

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)
6 голосов
/ 17 февраля 2017

Если это версия 3.x, просто используйте:

variantname = input()

Например, вы хотите ввести 8:

x = input()
8

x будет равно 8, но это будет строка, если вы не определите ее иначе.

Таким образом, вы можете использовать команду конвертирования, например:

a = int(x) * 1.1343
print(round(a, 2)) # '9.07'
9.07
6 голосов
/ 07 августа 2010

Начиная с Python 3.2 2.7, теперь есть argparse для обработки аргументов командной строки.

...