ฉันจะทำให้โปรแกรมหลามทำงานอย่างไรเช่นเครื่องมือยูนิกซ์ที่เหมาะสม?

ฉันมีสคริปต์ Python อยู่สองสามตัวและฉันก็กำลังเขียนมันใหม่ ฉันมีปัญหาเดียวกันกับพวกเขาทั้งหมด

ฉันไม่เห็นได้ชัดว่าจะเขียนโปรแกรมอย่างไรเพื่อให้พวกเขาทำตัวเหมือนเครื่องมือยูนิกซ์ที่เหมาะสม

เพราะสิ่งนี้

$ cat characters | progname

และนี่

$ progname characters

ควรสร้างผลลัพธ์เดียวกัน

สิ่งที่ใกล้เคียงที่สุดที่ฉันสามารถพบได้ใน Python คือไลบรารี fileinput น่าเสียดายที่ฉันไม่เห็นวิธีเขียนสคริปต์ Python ของฉันอีกครั้งซึ่งทั้งหมดมีลักษณะดังนี้:

#!/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

ไลบรารี fileinput ประมวลผล 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 

มันคล้ายกับคำตอบของ Mikel ยกเว้นว่าจะใช้sysโมดูล ฉันคิดว่าพวกเขามีในนั้นมันจะต้องมีเหตุผล ...

วิธีที่ฉันชอบในการทำมันกลายเป็น ... (และนี่นำมาจากบล็อก Linux ตัวเล็ก ๆ ที่เรียกว่าHarbinger's Hollow )

#!/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ที่ lowercases และลบสำเนียงจากสตริงที่กำหนด

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

ผมคิดว่าเวลา firest ที่ผมเห็นการแก้ปัญหานี้เป็นที่นี่

หากระบบของคุณไม่มี/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:
            ...