import sys, itertools def open_arg(_arg0, filename = "-"): # Default of - means stdin if filename == "-": return sys.stdin return open(filename, 'r') def head(file): for line in file.readlines()[:10]: print(line, end = "") if __name__ == '__main__': try: file = open_arg(*sys.argv) head(file) except FileNotFoundError: print("File not found!", file=sys.stderr) sys.exit(1) except: print("Some other error occured", file=sys.stderr)