Как читать из stdin или из файла, если в Python нет данных?

Question

У меня есть сценарий CLI и вы хотите, чтобы он считывал данные из файла. Он должен быть в состоянии прочитать его двумя способами:

• cat data.txt | ./my_script.py
• ./my_script.py data.txt

-a немного как grep, например.

То, что я знаю:

• sys.argv и optparse позволяют мне читать любые арг и варианты легко.
• sys.stdin позвольте мне зачитать данные по трубам в
• fileinput сделать весь процесс автоматического

К сожалению:

• с использованием fileinput использует стандартный ввод и любые арг в качестве входных данных. Поэтому я не могу использовать параметры, которые не являются именами файлов, когда он пытается их открыть.
• sys.stdin.readlines() работает отлично, но если я не трубу никаких данных, он висит, пока я не ввести Ctrl + D
• Я не знаю, как реализовать «, если ничего стандартного ввода, читать из файла в аргументах "потому что stdin всегда True в булевом контексте.

Я бы хотел, чтобы это было возможно, если это возможно.

Answer 1

Процесс ваши аргументы не имя файла, однако вы хотите, так что вы завершаете с массивом аргументов без опций, а затем передать этот массив в качестве параметра fileinput.input():

import fileinput 
for line in fileinput.input(remaining_args): 
    process(line) 

Answer 2

Не существует надежного способа определить, связано ли что sys.stdin связано с чем угодно, и не подходит ли это (например, пользователь хочет вставить данные). Определите наличие имени файла в качестве аргумента и используйте stdin, если он не найден.

Answer 3

Для UNIX/Linux вы можете обнаружить, является ли данные, который поступает в глядя на os.isatty(0)

$ date | python -c "import os;print os.isatty(0)" 
False 
$ python -c "import os;print os.isatty(0)" 
True 

Я не уверен, есть эквивалент для Windows.

редактировать Хорошо, я попробовал с python2.6 на окнах XP

C:\Python26>echo "hello" | python.exe -c "import os;print os.isatty(0)" 
False 

C:\Python26> python.exe -c "import os;print os.isatty(0)" 
True 

Так может быть, это не все безнадежно для окон

Answer 4

Argparse позволяет это сделать в достаточно легко и вы действительно должны использовать его вместо optparse, если у вас нет проблем с совместимостью.

код будет идти-то вроде этого:

import argparse 
parser = argparse.ArgumentParser() 
parser.add_argument('--input', type = argparse.FileType('r'), default = '-') 

Теперь у вас есть парсер, который будет анализировать ваши аргументы командной строки, используйте файл, если он видит один, или использовать стандартный ввод, если он не делает.

Answer 5

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

sys.stdin.isatty() 

Он возвращает значение false, если вход от трубопровода или истина в противном случае.

Answer 6

Я ноб, поэтому это может быть не очень хороший ответ, но я пытаюсь сделать то же самое (разрешить один или несколько файлов в командной строке, по умолчанию - STDIN).

Окончательные комбо я соединял:

parser = argparse.ArgumentParser() 
parser.add_argument("infiles", nargs="*") 
args = parser.parse_args() 

for line in fileinput.input(args.infiles): 
    process(line) 

Это кажется, что единственным способом получить все желаемое поведение в одном элегантном корпусе, не требуя именованные аргументы. Так же, как команды UNIX используются в качестве такового:

cat file1 file2 
wc -l < file1 

Не:

cat --file file1 --file file2 

Оценил обратную связь/подтверждение от ветерана идиоматических питонистов, чтобы убедиться, что я получил лучший ответ. Не видели этого полного решения, упоминаемого где-либо еще, только фрагменты.