add platform support for resources on behalf of users

Author: mikesea

fixtures/platform_create_document.yaml

@@ -0,0 +1,18 @@
+- request: !!python/object:vcr.request.Request
+    body: '{"document": {"external_id": "doc_id"}}'
+    headers: !!python/object/apply:__builtin__.frozenset
+    - - !!python/tuple [User-Agent, Swiftype-Python/1.0.0]
+      - !!python/tuple [Authorization, Bearer 6cf7fbd297f00a8e3863a0595f55ff7d141cbef2fcbe00159d0f7403649b384e]
+      - !!python/tuple [Content-Type, application/json]
+    host: localhost
+    method: POST
+    path: /api/v1/engines/myusersengine/document_types/videos/documents.json
+    port: 3000
+    protocol: http
+  response:
+    body: {string: '{"external_id":"doc_id","engine_id":"52cca3f13ae740f4af000023","document_type_id":"52cca4a43ae740f4af000024","id":"52cca54f3ae740f4af000026","_id":"52cca54f3ae740f4af000026","updated_at":"2014-01-08T01:09:35Z"}'}
+    headers: {cache-control: 'max-age=0, private, must-revalidate', connection: close,
+      content-type: application/json; charset=utf-8, etag: '"2f47ab1d3d64e1f401758e0c1adba376"',
+      server: thin 1.5.0 codename Knife, x-request-id: e3747b602f410044648906559a79ce4f,
+      x-runtime: '0.176010', x-ua-compatible: IE=Edge}
+    status: {code: 200, message: OK}

fixtures/platform_create_document_type.yaml

@@ -0,0 +1,18 @@
+- request: !!python/object:vcr.request.Request
+    body: '{"document_type": {"name": "videos"}}'
+    headers: !!python/object/apply:__builtin__.frozenset
+    - - !!python/tuple [User-Agent, Swiftype-Python/1.0.0]
+      - !!python/tuple [Authorization, Bearer 6cf7fbd297f00a8e3863a0595f55ff7d141cbef2fcbe00159d0f7403649b384e]
+      - !!python/tuple [Content-Type, application/json]
+    host: localhost
+    method: POST
+    path: /api/v1/engines/myusersengine/document_types.json
+    port: 3000
+    protocol: http
+  response:
+    body: {string: '{"id":"52cca4a43ae740f4af000024","name":"videos","slug":"videos","engine_id":"52cca3f13ae740f4af000023","updated_at":"2014-01-08T01:06:44Z","document_count":0,"field_mapping":{}}'}
+    headers: {cache-control: 'max-age=0, private, must-revalidate', connection: close,
+      content-type: application/json; charset=utf-8, etag: '"18cf4a2934702f8d1872273ee78b544c"',
+      server: thin 1.5.0 codename Knife, x-request-id: 5df4547c828f6c5f32852192d3fefe12,
+      x-runtime: '0.131149', x-ua-compatible: IE=Edge}
+    status: {code: 200, message: OK}

fixtures/platform_create_documents.yaml

@@ -0,0 +1,18 @@
+- request: !!python/object:vcr.request.Request
+    body: '{"documents": [{"external_id": "doc_id1"}, {"external_id": "doc_id2"}]}'
+    headers: !!python/object/apply:__builtin__.frozenset
+    - - !!python/tuple [User-Agent, Swiftype-Python/1.0.0]
+      - !!python/tuple [Authorization, Bearer 6cf7fbd297f00a8e3863a0595f55ff7d141cbef2fcbe00159d0f7403649b384e]
+      - !!python/tuple [Content-Type, application/json]
+    host: localhost
+    method: POST
+    path: /api/v1/engines/myusersengine/document_types/videos/documents/bulk_create.json
+    port: 3000
+    protocol: http
+  response:
+    body: {string: '[true,true]'}
+    headers: {cache-control: 'max-age=0, private, must-revalidate', connection: close,
+      content-type: application/json; charset=utf-8, etag: '"bb009c6dcb5c622766ad5aa1ea1562d9"',
+      server: thin 1.5.0 codename Knife, x-request-id: 2453d94eb997eaa1501e4afcb91b1372,
+      x-runtime: '0.143837', x-ua-compatible: IE=Edge}
+    status: {code: 200, message: OK}

fixtures/platform_engine_create.yaml

@@ -0,0 +1,18 @@
+- request: !!python/object:vcr.request.Request
+    body: '{"engine": {"name": "myusersengine"}}'
+    headers: !!python/object/apply:__builtin__.frozenset
+    - - !!python/tuple [User-Agent, Swiftype-Python/1.0.0]
+      - !!python/tuple [Authorization, Bearer 6cf7fbd297f00a8e3863a0595f55ff7d141cbef2fcbe00159d0f7403649b384e]
+      - !!python/tuple [Content-Type, application/json]
+    host: localhost
+    method: POST
+    path: /api/v1/engines.json
+    port: 3000
+    protocol: http
+  response:
+    body: {string: '{"name":"myusersengine","slug":"myusersengine","key":"wLDDEvF4GetXAkx6xiss","id":"52ccaa333ae740f4af00002b","_id":"52ccaa333ae740f4af00002b","updated_at":"2014-01-08T01:30:27Z","document_count":0}'}
+    headers: {cache-control: 'max-age=0, private, must-revalidate', connection: close,
+      content-type: application/json; charset=utf-8, etag: '"8240b4e5c212143eb3e570e548c4d8d5"',
+      server: thin 1.5.0 codename Knife, x-request-id: 7ba05f6eb9c0d05aad3ce4ca5b1d8598,
+      x-runtime: '0.146829', x-ua-compatible: IE=Edge}
+    status: {code: 200, message: OK}

fixtures/platform_engine_destroy.yaml

@@ -0,0 +1,17 @@
+- request: !!python/object:vcr.request.Request
+    body: ''
+    headers: !!python/object/apply:__builtin__.frozenset
+    - - !!python/tuple [User-Agent, Swiftype-Python/1.0.0]
+      - !!python/tuple [Authorization, Bearer 6cf7fbd297f00a8e3863a0595f55ff7d141cbef2fcbe00159d0f7403649b384e]
+      - !!python/tuple [Content-Type, application/json]
+    host: localhost
+    method: DELETE
+    path: /api/v1/engines/myuserengine.json
+    port: 3000
+    protocol: http
+  response:
+    body: {string: '{"error":"Permission denied."}'}
+    headers: {cache-control: no-cache, connection: close, content-type: application/json;
+        charset=utf-8, server: thin 1.5.0 codename Knife, x-request-id: 1ce7984333c15a3fb5c158dadddb61e8,
+      x-runtime: '0.021690', x-ua-compatible: IE=Edge}
+    status: {code: 403, message: Forbidden}

swiftype/swiftype.py

@@ -13,10 +13,10 @@
 
 class Client(object):
 
-  def __init__(self, username=None, password=None, api_key=None, client_id=None, client_secret=None, host=DEFAULT_API_HOST):
+  def __init__(self, username=None, password=None, api_key=None, access_token=None, client_id=None, client_secret=None, host=DEFAULT_API_HOST):
       self.client_id = client_id
       self.client_secret = client_secret
-      self.conn = Connection(username=username, password=password, api_key=api_key, host=host, base_path=DEFAULT_API_BASE_PATH)
+      self.conn = Connection(username=username, password=password, api_key=api_key, access_token=access_token, host=host, base_path=DEFAULT_API_BASE_PATH)
 
   def engines(self, page=None, per_page=None):
     return self.conn._get(self.__engines_path(), self.__pagination_params(page, per_page))
@@ -170,10 +170,11 @@ def __init__(self, status, msg):
 
 class Connection(object):
 
-  def __init__(self, username=None, password=None, api_key=None, host=None, base_path=None):
+  def __init__(self, username=None, password=None, api_key=None, access_token=None, host=None, base_path=None):
     self.__username = username
     self.__password = password
     self.__api_key = api_key
+    self.__access_token = access_token
     self.__host = host
     self.__base_path = base_path
 
@@ -190,7 +191,6 @@ def _put(self, path, params={}, data={}):
     return self._request('PUT', path, params=params, data=data)
 
   def _request(self, method, path, params={}, data={}):
-
     headers = {}
     headers['User-Agent'] = USER_AGENT
     headers['Content-Type'] = 'application/json'
@@ -200,6 +200,9 @@ def _request(self, method, path, params={}, data={}):
       base64_credentials = base64.encodestring(credentials)
       authorization = "Basic %s" % base64_credentials[:-1]
       headers['Authorization'] = authorization
+    elif self.__access_token is not None and self.__api_key is None:
+      params.pop('auth_token', None)
+      headers['Authorization'] = "Bearer %s" % self.__access_token
     elif self.__api_key is not None:
       params['auth_token'] = self.__api_key
     else:

tests/test_swiftype.py

@@ -302,5 +302,35 @@ def test_create_user(self):
             response = self.client.create_user()
             self.assertEqual(response['status'], 200)
 
+class TestPlatformResources(unittest.TestCase):
+
+    def setUp(self):
+        access_token = '6cf7fbd297f00a8e3863a0595f55ff7d141cbef2fcbe00159d0f7403649b384e'
+        self.engine = 'myusersengine'
+        self.document_type = 'videos'
+        self.client = swiftype.Client(access_token=access_token, host='localhost:3000')
+
+    def test_platform_engine_create(self):
+        with vcr.use_cassette('fixtures/platform_engine_create.yaml'):
+            response = self.client.create_engine(self.engine)
+            self.assertEqual(response['body']['name'], self.engine)
+
+    def test_platform_create_document_type(self):
+        with vcr.use_cassette('fixtures/platform_create_document_type.yaml'):
+            response = self.client.create_document_type(self.engine, self.document_type)
+            self.assertEqual(response['body']['slug'], self.document_type)
+
+    def test_platform_create_document(self):
+        with vcr.use_cassette('fixtures/platform_create_document.yaml'):
+            doc_id = 'doc_id'
+            id = self.client.create_document(self.engine, self.document_type, {'external_id': doc_id})['body']['external_id']
+            self.assertEqual(id, doc_id)
+
+    def test_platform_create_documents(self):
+        with vcr.use_cassette('fixtures/platform_create_documents.yaml'):
+            docs = [{'external_id': 'doc_id1'}, {'external_id': 'doc_id2'}]
+            stati = self.client.create_documents(self.engine, self.document_type, docs)['body']
+            self.assertEqual(stati, [True, True])
+
 if __name__ == '__main__':
     unittest.main()