Add optional support for SOCKS proxies via PySocks.

Author: slmnhq

python/setup.py

@@ -33,6 +33,9 @@
 if platform == 'darwin':
     install_requires += ['readline']
 
+# Optional PySocks support
+extras_require = dict(Socks=['PySocks >= 1.5.0'])
+
 setup(
   name = 'cm_api',
   version = '11.0.0',    # Compatible with API v11 (CM 5.5)
@@ -43,6 +46,7 @@
   # Project uses simplejson, so ensure that it gets installed or upgraded
   # on the target machine
   install_requires = install_requires,
+  extras_require = extras_require,
 
   author = 'Cloudera, Inc.',
   author_email = 'scm-users@cloudera.org',

python/src/cm_api/http_client.py

@@ -14,11 +14,24 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import os
 import cookielib
 import logging
 import posixpath
 import types
 import urllib
+
+try:
+  import socks
+  import socket
+  socks_server = os.environ.get("SOCKS_SERVER", None)
+  if socks_server:
+    host, port = socks_server.split(":")
+    socks.setdefaultproxy(socks.PROXY_TYPE_SOCKS5, host, int(port))
+    socket.socket = socks.socksocket
+except ImportError:
+  pass
+
 import urllib2
 
 __docformat__ = "epytext"

python/src/cm_api/resource.py

@@ -14,6 +14,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import os
 try:
   import json
 except ImportError:
@@ -22,6 +23,15 @@
 import posixpath
 import time
 import socket
+try:
+  import socks
+  socks_server = os.environ.get("SOCKS_SERVER", None)
+  if socks_server:
+    host, port = socks_server.split(":")
+    socks.setdefaultproxy(socks.PROXY_TYPE_SOCKS5, host, int(port))
+    socket.socket = socks.socksocket
+except ImportError:
+  pass
 import urllib2
 
 LOG = logging.getLogger(__name__)