allow cert_reqs to be a string and convert it to the appropriate SSL constant.

1 files changed, 10 insertions, 0 deletions

diff --git a/redis/connection.py b/redis/connection.py
index 5607508..c209750 100755
--- a/redis/connection.py
+++ b/redis/connection.py
@@ -570,6 +570,16 @@ class SSLConnection(Connection):
         self.certfile = certfile
         if cert_reqs is None:
             cert_reqs = ssl.CERT_NONE
+        elif isinstance(cert_reqs, basestring):
+            CERT_REQS = {
+                'none': ssl.CERT_NONE,
+                'optional': ssl.CERT_OPTIONAL,
+                'required': ssl.CERT_REQUIRED
+            }
+            if cert_reqs not in CERT_REQS:
+                raise RedisError(
+                    "Invalid SSL Certificate Required Flag: %s" % cert_reqs)
+            cert_reqs = CERT_REQS[cert_reqs]
         self.cert_reqs = cert_reqs
         self.ca_certs = ca_certs