Как преобразовать эту функцию python, чтобы взять sys.argv [1] вместо sys.stdin? [Дубликат]

Мне нравится общая идиома использования диспетчера контекста, но (слишком) тривиальное решение заканчивается закрытием sys.stdin, когда вы находитесь вне оператора with, которого я хочу избежать.

Заимствование из этого ответа , это обходное решение:

import sys
import contextlib

@contextlib.contextmanager
def _smart_open(filename, mode='Ur'):
    if filename == '-':
        if mode is None or mode == '' or 'r' in mode:
            fh = sys.stdin
        else:
            fh = sys.stdout
    else:
        fh = open(filename, mode)
    try:
        yield fh
    finally:
        if filename is not '-':
            fh.close()

if __name__ == '__main__':
    args = sys.argv[1:]
    if args == []:
        args = ['-']
    for filearg in args:
        with _smart_open(filearg) as handle:
            do_stuff(handle)

Предположим, вы могли бы достичь чего-то подобного с os.dup() , но код I приготовленный до этого, оказался более сложным и более волшебным, тогда как выше было несколько неуклюже, но очень просто.

Что-то вроде:

if input_from_file:
    f = open(file_name, "rt")
else:
    f = sys.stdin
inL = f.readline()
while inL:
    print inL.rstrip()
    inL = f.readline()

Не прямой ответ, но связанный.

Обычно, когда вы пишете скрипт python, вы можете использовать пакет argparse. ЕСЛИ это случай, который вы можете использовать:

parser = argparse.ArgumentParser()
parser.add_argument('infile', nargs='?', type=argparse.FileType('r'), default=sys.stdin)

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

, и мы установили по умолчанию sys.stdin;

, так что если есть файл он будет читать, а если нет, то он будет принимать входные данные из stdin «Примечание: мы используем позиционный аргумент в приведенном выше примере»

для более подробной информации: https: // docs. python.org/2/library/argparse.html#nargs

Я предпочитаю использовать «-» в качестве индикатора, который вы должны прочитать из stdin, более явным:

import sys
with open(sys.argv[1], 'r') if sys.argv[1] is not "-" else sys.stdin as f:
    pass # do something here

Чтобы использовать инструкцию python with, можно использовать следующий код:

import sys
with open(sys.argv[1], 'r') if len(sys.argv) > 1 else sys.stdin as f:
    # read data using f
    # ......

Модуль fileinput может делать то, что вы хотите - если аргументы non-option находятся в args, тогда:

import fileinput
for line in fileinput.input(args):
    print line

Если args пуст, тогда fileinput.input() будет читать от stdin; в противном случае он читает из каждого файла по очереди, аналогично Perl while(<>).