Limit explicit UTF-8 encoding to Python < 3.

1 files changed, 4 insertions, 2 deletions

diff --git a/bin/sqlformat b/bin/sqlformat
index 6a2496a..fcee452 100755
--- a/bin/sqlformat
+++ b/bin/sqlformat
@@ -98,8 +98,10 @@ def main():
         err = sys.exc_info()[1]  # Python 2.5 compatibility
         _error('Invalid options: %s' % err, exit_=1)
 
-    stream.write(sqlparse.format(data, **formatter_opts).encode('utf-8',
-                                                                'replace'))
+    s = sqlparse.format(data, **formatter_opts)
+    if sys.version_info < (3,):
+        s = s.encode('utf-8', 'replace')
+    stream.write(s)
     stream.flush()