Чтобы использовать инструкцию python with, можно использовать следующий код:
with
import sys with open(sys.argv[1], 'r') if len(sys.argv) > 1 else sys.stdin as f: # read data using f # ......