challenge1-2 nl.py

nl.py

import sys
import os

args = sys.argv

if len(args) < 2:
    print("usage python nl.py path/to/file")
    exit()

path = args[1]
if os.path.exists(path) == False:
    print("file not found.")
    exit()

i = 1
f = open(path, 'r')
while True:
    s = f.readline()
    if not s: break
    if len(s) == 1:
        print("      \t")
    else:
        print("{:6d}\t{}".format(i, s[:-1]))
        i += 1

test_nl.sh

#!/bin/bash

echo "test nl.py"
diff <(nl fizz_buzz.py) <(python nl.py fizz_buzz.py) | wc -l | awk '{$0 == 0 ? m = "ok" : m = "ng"} END {print m}'