Add support for configurable API base endpoint URLs

monkeylearn/__init__.py

@@ -2,23 +2,28 @@
 from __future__ import (
     print_function, unicode_literals, division, absolute_import)
 
+from monkeylearn.settings import DEFAULT_BASE_ENDPOINT
 from monkeylearn.classification import Classification
 from monkeylearn.extraction import Extraction
 from monkeylearn.pipelines import Pipelines
 
 class MonkeyLearn(object):
 
-    def __init__(self, token):
+    def __init__(self, token, base_endpoint=DEFAULT_BASE_ENDPOINT):
         self.token = token
+        self.base_endpoint = base_endpoint
 
     @property
     def classifiers(self):
-        return Classification(self.token)
+        return Classification(token=self.token,
+                              base_endpoint=self.base_endpoint)
 
     @property
     def extractors(self):
-        return Extraction(self.token)
+        return Extraction(token=self.token,
+                          base_endpoint=self.base_endpoint)
 
     @property
     def pipelines(self):
-        return Pipelines(self.token)
+        return Pipelines(token=self.token,
+                         base_endpoint=self.base_endpoint)

monkeylearn/classification.py

@@ -6,14 +6,14 @@
 from six.moves import range
 
 from monkeylearn.utils import SleepRequestsMixin, MonkeyLearnResponse, HandleErrorsMixin
-from monkeylearn.settings import CLASSIFICATION_ENDPOINT, DEFAULT_BATCH_SIZE
+from monkeylearn.settings import DEFAULT_BASE_ENDPOINT, DEFAULT_BATCH_SIZE
 
 
 class Classification(SleepRequestsMixin, HandleErrorsMixin):
 
-    def __init__(self, token):
+    def __init__(self, token, base_endpoint=DEFAULT_BASE_ENDPOINT):
         self.token = token
-        self.endpoint = CLASSIFICATION_ENDPOINT
+        self.endpoint = base_endpoint + 'classifiers/'
 
     @property
     def categories(self):

monkeylearn/extraction.py

@@ -5,13 +5,13 @@
 from six.moves import range
 
 from monkeylearn.utils import SleepRequestsMixin, MonkeyLearnResponse, HandleErrorsMixin
-from monkeylearn.settings import EXTRACTION_ENDPOINT, DEFAULT_BATCH_SIZE
+from monkeylearn.settings import DEFAULT_BASE_ENDPOINT, DEFAULT_BATCH_SIZE
 
 class Extraction(SleepRequestsMixin, HandleErrorsMixin):
 
-    def __init__(self, token):
+    def __init__(self, token, base_endpoint=DEFAULT_BASE_ENDPOINT):
         self.token = token
-        self.endpoint = EXTRACTION_ENDPOINT
+        self.endpoint = base_endpoint + 'extractors/'
 
     def extract(self, module_id, text_list, batch_size=DEFAULT_BATCH_SIZE,
                 sleep_if_throttled=True):

monkeylearn/pipelines.py

@@ -3,14 +3,14 @@
     print_function, unicode_literals, division, absolute_import)
 
 from monkeylearn.utils import SleepRequestsMixin, MonkeyLearnResponse, HandleErrorsMixin
-from monkeylearn.settings import PIPELINES_ENDPOINT
+from monkeylearn.settings import DEFAULT_BASE_ENDPOINT
 from monkeylearn.exceptions import MonkeyLearnException
 
 class Pipelines(SleepRequestsMixin, HandleErrorsMixin):
 
-    def __init__(self, token):
+    def __init__(self, token, base_endpoint=DEFAULT_BASE_ENDPOINT):
         self.token = token
-        self.endpoint = PIPELINES_ENDPOINT
+        self.endpoint = base_endpoint + 'pipelines/'
 
     def run(self, module_id, data, sandbox=False, sleep_if_throttled=True):
         if not isinstance(data, dict):

monkeylearn/settings.py

@@ -5,7 +5,4 @@
 DEFAULT_BATCH_SIZE = 200
 MAX_BATCH_SIZE = 500
 MIN_BATCH_SIZE = 100
-BASE_ENDPOINT = 'https://api.monkeylearn.com/v2/'
-CLASSIFICATION_ENDPOINT = BASE_ENDPOINT + 'classifiers/'
-EXTRACTION_ENDPOINT = BASE_ENDPOINT + 'extractors/'
-PIPELINES_ENDPOINT = BASE_ENDPOINT + 'pipelines/'
+DEFAULT_BASE_ENDPOINT = 'https://api.monkeylearn.com/v2/'