python - Python 2.7 中的 sys.stdin 文件扩展名

Question

我正在尝试查看我使用 sys.stdin 输入的文件是否具有 .gz 文件扩展名。通常我只是直接使用文件路径，但是当我执行 sys.stdin 时，它会自动将其打开到读取对象中。

有没有什么方法可以在不执行 os.getcwd() 或获取完整文件路径的情况下从 stdin 获取文件名？

我正在尝试 sys.stdin.endswith('.gz') 但它不起作用（显然 b/c 它不是字符串）但是我可以用 sys.stdin 对象做些什么来在我之前抓取扩展名继续处理吗？

import sys
file = sys.stdin
if file.endswith('.gz'):
    print 'yup'

Answer 1

试试这样

import sys
file = sys.stdin
if file.name.endswith('.gz'):
    print 'yup'

更新：

file = raw_input("Enter filename:")
sys.stdin = open(file, 'r')
if sys.stdin.name.endswith('.gz'):
    print 'yup'