Merge pull request #49 from teners/add-stopwords

Add stopwords to the client

Author: kishorenc

src/typesense/client.py

@@ -7,6 +7,7 @@
 from .configuration import Configuration
 from .api_call import ApiCall
 from .analytics import Analytics
+from .stopwords import Stopwords
 
 class Client(object):
     def __init__(self, config_dict):
@@ -19,3 +20,4 @@ def __init__(self, config_dict):
         self.analytics = Analytics(self.api_call)
         self.operations = Operations(self.api_call)
         self.debug = Debug(self.api_call)
+        self.stopwords = Stopwords(self.api_call)

src/typesense/stopwords.py

@@ -0,0 +1,21 @@
+from .stopwords_set import StopwordsSet
+
+
+class Stopwords(object):
+    RESOURCE_PATH = '/stopwords'
+
+    def __init__(self, api_call):
+        self.api_call = api_call
+        self.stopwords_sets = {}
+
+    def __getitem__(self, stopwords_set_id):
+        if stopwords_set_id not in self.stopwords_sets:
+            self.stopwords_sets[stopwords_set_id] = StopwordsSet(self.api_call, stopwords_set_id)
+
+        return self.stopwords_sets.get(stopwords_set_id)
+
+    def upsert(self, stopwords_set_id, stopwords_set):
+        return self.api_call.put('{}/{}'.format(Stopwords.RESOURCE_PATH, stopwords_set_id), stopwords_set)
+
+    def retrieve(self):
+        return self.api_call.get('{0}'.format(Stopwords.RESOURCE_PATH))

src/typesense/stopwords_set.py

@@ -0,0 +1,14 @@
+class StopwordsSet(object):
+    def __init__(self, api_call, stopwords_set_id):
+        self.stopwords_set_id = stopwords_set_id
+        self.api_call = api_call
+
+    def _endpoint_path(self):
+        from .stopwords import Stopwords
+        return u"{0}/{1}".format(Stopwords.RESOURCE_PATH, self.stopwords_set_id)
+
+    def retrieve(self):
+        return self.api_call.get(self._endpoint_path())
+
+    def delete(self):
+        return self.api_call.delete(self._endpoint_path())