python/file_reader_with_context_manager.py at Advance-Python-Assignment-Solution · sreekanthpogula/python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
from contextlib import contextmanager

@contextmanager
def open_file(filename):
    try:
        f = open(filename, 'r')
        yield f
    finally:
        f.close()

def read_lines(filename):
    with open_file(filename) as f:
        for line in f:
            yield line.strip()

# Example usage
for line in read_lines('testfile.txt'):
    print(line)