Linuxのnlコマンドに相当するコードをPythonで実装してみた

Linuxのnlコマンドに相当するコードをPythonで実装してみました。

test.txt

AAA
BBB
CCC

nl.py

import sys
if len(sys.argv) > 1:
    try:
        f = open(sys.argv[1], 'r')
    except FileNotFoundError:
        print('Error: File {0} is not found.'.format(
            sys.argv[1]), file=sys.stderr)
        sys.exit(1)
else:
    f = sys.stdin

lines = f.readlines()
f.close()

i = 1
for line in lines:
    if line == '\n':
        print()
        continue
    print('\t' + str(i) + ' ' + line, end='')
    i += 1

出力結果

$ python nl.py test.txt
    1  AAA
    2  BBB
    3  CCC