Support python interception for httplib

Author: pimterry

overrides/pythonpath/httplib.py

@@ -0,0 +1,46 @@
+from httptoolkit_intercept import preload_real_module
+
+preload_real_module('httplib')
+
+import httplib, os, functools
+
+# Re-export all public fields
+from httplib import *
+# Load a few extra notable private fields, for max compatibility
+from httplib import __file__, __doc__
+
+_httpProxy = os.environ['HTTP_PROXY']
+[_proxyHost, _proxyPort] = _httpProxy.split('://')[1].split(':')
+_certPath = os.environ['SSL_CERT_FILE']
+
+_http_connection_init = HTTPConnection.__init__
+@functools.wraps(_http_connection_init)
+def _new_http_connection_init(self, host, port=80, *k, **kw):
+    _http_connection_init(self, _proxyHost, _proxyPort, *k, **kw)
+    self.set_tunnel(host, port)
+HTTPConnection.__init__ = _new_http_connection_init
+
+def _build_default_context():
+    import ssl
+    context = ssl.SSLContext(ssl.PROTOCOL_SSLv23)
+    context.options |= ssl.OP_NO_SSLv2
+    context.options |= ssl.OP_NO_SSLv3
+    return context
+
+_https_connection_init = HTTPSConnection.__init__
+@functools.wraps(_https_connection_init)
+def _new_https_connection_init(self, host, port=443, *k, **kw):
+    context = None
+    if 'context' in kw:
+        context = kw.get('context')
+    elif len(k) > 7:
+        context = k[7]
+
+    if context == None:
+        context = kw['context'] = _build_default_context()
+
+    context.load_verify_locations(_certPath)
+
+    _https_connection_init(self, _proxyHost, _proxyPort, *k, **kw)
+    self.set_tunnel(host, port)
+HTTPSConnection.__init__ = _new_https_connection_init

overrides/pythonpath/httptoolkit_intercept.py

@@ -0,0 +1,25 @@
+try:
+    import importlib
+    if not hasattr(importlib, 'reload'):
+        raise Exception('wrong importlib')
+except:
+    import imp
+    importlib = imp
+
+def preload_real_module(moduleName):
+    # Re-importing the real module at the top level of an override fails after deleting it from
+    # sys.modules['httplib'] in Python 2. Some interesting issues there ofc, but doing this 
+    # instead works nicely in both Python 2 & 3.
+    import sys, os
+
+    override_path = os.path.dirname(os.path.abspath(__file__))
+    # print('override path %s' % override_path)
+    original_sys_path = list(sys.path)
+    # print('sys_path %s' % original_sys_path)
+    sys.path = [p for p in sys.path if p != override_path and p != '']
+    # print('new sys_path %s' % sys.path)
+
+    del sys.modules[moduleName]
+    __import__(moduleName)
+
+    sys.path = original_sys_path

overrides/pythonpath/stripe.py

@@ -1,27 +1,5 @@
-try:
-    import importlib
-    if not hasattr(importlib, 'reload'):
-        raise Exception('wrong importlib')
-except:
-    import imp
-    importlib = imp
-
-def _load_real_stripe():
-    # Importing this at the top level breaks after deleting sys.modules['stripe'],
-    # in Python 2. Some interesting internal issues there, but this works in Py 2 & 3.
-    import sys, os
-
-    override_path = os.path.dirname(os.path.abspath(__file__))
-
-    original_sys_path = list(sys.path)
-
-    sys.path = [p for p in sys.path if p != override_path]
-    del sys.modules['stripe']
-    import stripe
-
-    sys.path = original_sys_path
-
-_load_real_stripe()
+from httptoolkit_intercept import preload_real_module
+preload_real_module('stripe')
 
 import stripe, os
 stripe.ca_bundle_path = os.environ['SSL_CERT_FILE']