Merge pull request #114 from twneale/master

Allow StartTLS response to be ascii or bytes (py3k)
vertica · May 3, 2016 · f11429c · f11429c
2 parents e0a9fa5 + b039dc8
commit f11429c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vertica_python/vertica/connection.py b/vertica_python/vertica/connection.py
@@ -113,7 +113,7 @@ def _socket(self):
             from ssl import CertificateError, SSLError
             raw_socket.sendall(messages.SslRequest().to_bytes())
             response = raw_socket.recv(1)
-            if response == 'S':
+            if response in ('S', b'S'):
                 try:
                     if isinstance(ssl_options, ssl.SSLContext):
                         raw_socket = ssl_options.wrap_socket(raw_socket, server_hostname=host)