summaryrefslogtreecommitdiff
path: root/sqlparse/cli.py
diff options
context:
space:
mode:
authorJon Dufresne <jon.dufresne@gmail.com>2017-10-28 17:07:09 -0700
committerJon Dufresne <jon.dufresne@gmail.com>2017-10-28 17:07:09 -0700
commitacaf00cbaf32e7a14e39bcc0a5fd774a5385faa0 (patch)
tree48a8f6022b722d9f23ed0c483d5f0ebf14c0d085 /sqlparse/cli.py
parentf8825153da66289c021252f0b221c16302224a86 (diff)
downloadsqlparse-acaf00cbaf32e7a14e39bcc0a5fd774a5385faa0.tar.gz
Fix unclosed file warnings discovered during tests
Appear as: ResourceWarning: unclosed file ... Always explicitly close files or detach file wrappers.
Diffstat (limited to 'sqlparse/cli.py')
-rwxr-xr-xsqlparse/cli.py7
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: