File: LP6E/AppendixB/Part5/mypkg/__main__.py

def countLines(name):
    file = open(name)
    return len(file.readlines())

def countChars(name):
    return len(open(name).read())

def test(name):                                  # Or pass file object
    return countLines(name), countChars(name)    # Or return a dictionary

if __name__ == '__main__':                       # Added: self-test code
    print(test('mymod.py'))                      # When run, not when imported



[Home page] Books Code Blog Python Author Train Find ©M.Lutz