summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--Lib/formatter.py14
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)