add mixin for both connection

Author: jerrycheng9

shotgun_api3/shotgun.py

@@ -190,30 +190,37 @@ def _set_socket_keepalive(sock) -> None:
             LOG.debug("Unable to set %s on socket." % option_name, exc_info=True)
 
 
-class KeepaliveHTTPConnection(HTTPConnectionWithTimeout):
+class _KeepaliveConnectionMixin(object):
     """
-    httplib2 HTTP connection that enables TCP keepalive once connected.
+    Mixin that enables TCP keepalive once the connection is established.
 
-    Passed to ``httplib2.Http.request()`` as its ``connection_type`` so that the
-    bundled httplib2 does not need to be modified.
+    Must be listed before the httplib2 connection class so that this
+    ``connect()`` runs and delegates to the real one. ``self.sock`` is the
+    SSL-wrapped socket for HTTPS, which delegates ``setsockopt`` to the socket
+    underneath.
     """
 
     def connect(self) -> None:
         super().connect()
         _set_socket_keepalive(self.sock)
 
 
-class KeepaliveHTTPSConnection(HTTPSConnectionWithTimeout):
+class KeepaliveHTTPConnection(_KeepaliveConnectionMixin, HTTPConnectionWithTimeout):
     """
-    httplib2 HTTPS connection that enables TCP keepalive once connected.
+    httplib2 HTTP connection that enables TCP keepalive once connected.
 
     Passed to ``httplib2.Http.request()`` as its ``connection_type`` so that the
     bundled httplib2 does not need to be modified.
     """
 
-    def connect(self) -> None:
-        super().connect()
-        _set_socket_keepalive(self.sock)
+
+class KeepaliveHTTPSConnection(_KeepaliveConnectionMixin, HTTPSConnectionWithTimeout):
+    """
+    httplib2 HTTPS connection that enables TCP keepalive once connected.
+
+    Passed to ``httplib2.Http.request()`` as its ``connection_type`` so that the
+    bundled httplib2 does not need to be modified.
+    """
 
 
 # ----------------------------------------------------------------------------