it-swarm-ru.tech

Пользовательский ввод и аргументы командной строки

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

475
Teifion

Чтобы прочитать ввод пользователя, вы можете попробовать модуль 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 ваш друг.

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

Или для Python 3:

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

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

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

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

201
steampowered

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

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

31
Dave Webb

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

18
Vhaerun

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

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'>
13
Viswesn

Используйте 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.

ПэВ

10
Simon Peverett

Если вы запускаете Python <2.7, вам необходимо 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)
9
Matt Olan

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

variantname = input()

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

x = input()
8

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

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

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

По состоянию на Python 3,2 2.7, теперь есть argparse для обработки аргументов командной строки.

6
GreenMatt

В Python 2:

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

В 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