Add support for ssl client authentication

druid-io · Jul 9, 2021 · eba905f · eba905f
1 parent 98cab4d
commit eba905f
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pydruid/client.py b/pydruid/client.py
@@ -15,6 +15,7 @@
 #
 import json
 import re
+import ssl
 import urllib
 from base64 import b64encode
 
@@ -43,6 +44,12 @@ def set_proxies(self, proxies):
         opener = urllib.request.build_opener(proxy_support)
         urllib.request.install_opener(opener)
 
+    def set_cert_chain(self, certfile, keyfile=None, password=None):
+        context = ssl.create_default_context()
+        context.load_cert_chain(certfile, keyfile, password)
+        handler = urllib.request.HTTPSHandler(context=context)
+        urllib.request.install_opener(handler)
+
     def _prepare_url_headers_and_body(self, query):
         querystr = json.dumps(query.query_dict).encode("utf-8")
         if self.url.endswith("/"):