파이썬 프로그램을 적절한 유닉스 도구처럼 동작하게하려면 어떻게해야합니까?

몇 가지 Python 스크립트가 있으며 다시 작성하려고합니다. 나는 그들 모두와 같은 문제가 있습니다.

적절한 유닉스 도구처럼 동작하도록 프로그램을 작성하는 방법은 분명하지 않습니다.

왜냐하면 이건

$ cat characters | progname

이

$ progname characters

동일한 출력을 생성해야합니다.

파이썬에서 찾을 수있는 가장 가까운 것은 파일 입력 라이브러리였습니다. 불행히도 파이썬 스크립트를 다시 작성하는 방법을 실제로 보지 못합니다. 모두 다음과 같습니다.

#!/usr/bin/env python 
# coding=UTF-8

import sys, re

for file in sys.argv[1:]:
    f = open(file)
    fs = f.read()
    regexnl = re.compile('[^\s\w.,?!:;-]')
    rstuff = regexnl.sub('', fs)
    f.close()
    print rstuff

파일 입력 라이브러리는 stdin이 있으면 stdin을 처리하고 파일이 있으면 파일을 처리합니다. 그러나 단일 라인을 반복합니다.

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

나는 정말로 그것을 얻지 못한다. 작은 파일을 다루거나 파일을 많이 사용하지 않으면 분명히 보일 수 있습니다. 그러나 내 목적으로는 위와 같이 단순히 전체 파일을 열고 문자열로 읽는 것보다 훨씬 느립니다.

현재 위와 같은 스크립트를 실행합니다.

$ pythonscript textfilename1 > textfilename2

그러나 나는 파이프 (그리고 그 형제들)를 파이프에서 실행할 수 있기를 원합니다.

$ grep pattern textfile1 | pythonscript | pythonscript | pythonscript > textfile2

왜 그냥

files = sys.argv[1:]
if not files:
    files = ["/dev/stdin"]

for file in files:
    f = open(file)
    ...

파일 이름이 인수로 제공되는지 확인하거나에서 읽습니다 sys.stdin.

이 같은:

if sys.argv[1]:
   f = open(sys.argv[1])
else:
   f = sys.stdin 

sys모듈을 사용한다는 점을 제외하면 Mikel의 답변과 비슷 합니다. 나는 그들이 거기에 있다면 그것이 이유가 있어야한다고 생각합니다 ...

내가 선호하는 방법은 다음과 같습니다. (그리고 이것은 Harbinger 's Hollow 라는 멋진 작은 Linux 블로그에서 가져 왔습니다 )

#!/usr/bin/env python

import argparse, sys

parser = argparse.ArgumentParser()
parser.add_argument('filename', nargs='?')
args = parser.parse_args()
if args.filename:
    string = open(args.filename).read()
elif not sys.stdin.isatty():
    string = sys.stdin.read()
else:
    parser.print_help()

내가 이것을 가장 좋아하는 이유는 블로거가 말했듯이 실수로 입력하지 않고 호출하면 바보 같은 메시지를 출력하기 때문입니다. 또한 기존의 모든 Python 스크립트에 아주 잘 들어가서 모두 포함하도록 수정했습니다.

files=sys.argv[1:]

for f in files or [sys.stdin]:
   if isinstance(f, file):
      txt = f.read()
   else:
      txt = open(f).read()

   process(txt)

이 솔루션을 사용하고 있으며 매력처럼 작동합니다. 실제로 나는 주어진 문자열에서 소문자를 제거하고 악센트를 제거 하는 스크립트 calle unaccent 를 사용 하고 있습니다.

argument = sys.argv[1:] if len(sys.argv) > 1 else sys.stdin.read()

이 솔루션을 본 가장 강력한 시점은 여기에있는 것 같습니다 .

시스템에이 없거나 /dev/stdin더 일반적인 솔루션을 원한다면 다음과 같이 더 복잡한 것을 시도해보십시오.

class Stdin(object):
    def __getattr__(self, attr):
        return getattr(sys.stdin, attr)

    def __enter__(self):
        return self

def myopen(path):
    if path == "-":
        return Stdin()
    return open(path)

for n in sys.argv[1:] or ["-"]:
    with myopen(n) as f:
            ...