diff options
-rw-r--r-- | Lib/formatter.py | 14 |
1 files changed, 9 insertions, 5 deletions
diff --git a/Lib/formatter.py b/Lib/formatter.py index d8cca52..9338261 100644 --- a/Lib/formatter.py +++ b/Lib/formatter.py @@ -436,11 +436,15 @@ def test(file = None): fp = open(sys.argv[1]) else: fp = sys.stdin - for line in fp: - if line == '\n': - f.end_paragraph(1) - else: - f.add_flowing_data(line) + try: + for line in fp: + if line == '\n': + f.end_paragraph(1) + else: + f.add_flowing_data(line) + finally: + if fp is not sys.stdin: + fp.close() f.end_paragraph(0) |