python - 如何在 Python 中读取整个文件？在命令行中普遍工作

Question

如何在 Python 中读取整个文件？我希望我的脚本能够工作，但是它被称为

• script.py log.txt
• script.py < log2.txt
• python script.py < log2.txt
• python -i script.py logs/yesterday.txt

你明白了。

---

我试过了

import fileinput
from bs4 import BeautifulSoup
f = fileinput.input()
soup = BeautifulSoup(f.read())

但是我明白了

Traceback (most recent call last):
  File "visual-studio-extension-load-times.py", line 5, in <module>
    soup = BeautifulSoup(f.read())
AttributeError: FileInput instance has no attribute 'read'

Answer 1

不用fileinput，直接自己打开文件:

import sys
try:
    fileobj = open(sys.argv[1], 'r')
except IndexError:
    fileobj = sys.stdin

with fileobj:
    data = fileobj.read()