diff options
Diffstat (limited to 'sqlparse')
| -rwxr-xr-x | sqlparse/cli.py | 7 |
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: |
