Allow using a proxy.

pgoapi/pgoapi.py

@@ -49,6 +49,7 @@ def __init__(self):
 
         self._auth_provider = None
         self._api_endpoint = 'https://pgorelease.nianticlabs.com/plfe/rpc'
+        self._proxy = None
 
         self._position_lat = None
         self._position_lng = None
@@ -58,6 +59,9 @@ def __init__(self):
 
     def set_logger(self, logger = None):
         self.log = logger or logging.getLogger(__name__)
+
+    def set_proxy(self, proxy_config):
+        self._proxy = proxy_config
 
     def get_api_endpoint(self):
         return self._api_endpoint
@@ -73,7 +77,8 @@ def set_position(self, lat, lng, alt):
         self._position_alt = alt
 
     def create_request(self):    
-        request = PGoApiRequest(self._api_endpoint, self._auth_provider, self._position_lat, self._position_lng, self._position_alt)
+        request = PGoApiRequest(self._api_endpoint, self._auth_provider, self._position_lat,
+                                self._position_lng, self._position_alt, self._proxy)
         return request
 
     def __getattr__(self, func):
@@ -150,7 +155,7 @@ def login(self, provider, username, password, lat = None, lng = None, alt = None
 
 
 class PGoApiRequest:
-    def __init__(self, api_endpoint, auth_provider, position_lat, position_lng, position_alt):
+    def __init__(self, api_endpoint, auth_provider, position_lat, position_lng, position_alt, proxy=None):
         self.log = logging.getLogger(__name__)
 
         """ Inherit necessary parameters """
@@ -161,6 +166,8 @@ def __init__(self, api_endpoint, auth_provider, position_lat, position_lng, posi
         self._position_lng = position_lng
         self._position_alt = position_alt
 
+        self._proxy = proxy
+
         self._req_method_list = []
 
     def call(self):
@@ -174,7 +181,7 @@ def call(self):
             self.log.info('Not logged in')
             return NotLoggedInException()
 
-        request = RpcApi(self._auth_provider)
+        request = RpcApi(self._auth_provider, self._proxy)
 
         self.log.info('Execution of RPC')
         response = None

pgoapi/rpc_api.py

@@ -30,6 +30,7 @@
 import random
 import logging
 import requests
+import requesocks
 import subprocess
 
 from google.protobuf import message
@@ -49,11 +50,16 @@ class RpcApi:
 
     RPC_ID = 0
 
-    def __init__(self, auth_provider):
+    def __init__(self, auth_provider, proxy_config=None):
 
         self.log = logging.getLogger(__name__)
 
-        self._session = requests.session()
+        if proxy_config is None:
+            self._session = requests.session()
+        else:
+            self._session = requesocks.session()
+            self._session.proxies = proxy_config
+
         self._session.headers.update({'User-Agent': 'Niantic App'})
         self._session.verify = True
 
@@ -90,7 +96,7 @@ def _make_rpc(self, endpoint, request_proto_plain):
         request_proto_serialized = request_proto_plain.SerializeToString()
         try:
             http_response = self._session.post(endpoint, data=request_proto_serialized)
-        except requests.exceptions.ConnectionError as e:
+        except (requests.exceptions.ConnectionError, requesocks.exceptions.ConnectionError) as e:
             raise ServerBusyOrOfflineException
 
         return http_response

requirements.txt

@@ -4,3 +4,5 @@ requests==2.10.0
 s2sphere==0.2.4
 gpsoauth==0.3.0
 six
+stem==1.4.0
+git+git://github.com/dvska/requesocks.git@master#egg=requesocks