Merge pull request #365 from jdufresne/unclosed

Fix unclosed file warnings discovered during tests

1 files changed, 5 insertions, 2 deletions

diff --git a/sqlparse/cli.py b/sqlparse/cli.py
index ad6bc7a..f73f5d4 100755
--- a/sqlparse/cli.py
+++ b/sqlparse/cli.py
@@ -150,8 +150,11 @@ def main(args=None):
         if PY2:
             data = getreader(args.encoding)(sys.stdin).read()
         else:
-            data = TextIOWrapper(
-                sys.stdin.buffer, encoding=args.encoding).read()
+            wrapper = TextIOWrapper(sys.stdin.buffer, encoding=args.encoding)
+            try:
+                data = wrapper.read()
+            finally:
+                wrapper.detach()
     else:
         try:
             with open(args.filename, 'r', args.encoding) as f: