unix `nl` command like code with python

nl.py

# example: python nl.py fizz_buzz.py
import sys


def display_lines_with_number(lines):
    i = 1
    for line in lines:
        # 改行コードの場合は、行番号を振らずに表示する
        if line == "\n":
            print(line, end="")
            continue
        print(f"    {str(i)}  {line}", end="")
        i += 1


argc = len(sys.argv)
if argc == 1:
    f = sys.stdin
elif argc == 2:
    try:
        f = open(sys.argv[1], "r")
    except FileNotFoundError:
        sys.exit("wc: No such file or directory: " + sys.argv[1])
else:
    sys.exit("usage: wc [file]")

display_lines_with_number(f.readlines())