Registry client

openprocurement_client/api_base_client.py

@@ -126,6 +126,7 @@ def __init__(self,
 
     @staticmethod
     def _get_access_token(obj):
+
         return getattr(getattr(obj, 'access', ''), 'token', '')
 
     def _update_params(self, params):

openprocurement_client/registry_client.py

@@ -0,0 +1,110 @@
+import logging
+
+from .api_base_client import APIBaseClient, APITemplateClient, verify_file
+from .exceptions import InvalidResponse
+
+from iso8601 import parse_date
+from munch import munchify
+from retrying import retry
+from simplejson import loads
+
+
+class RegistryClient(APIBaseClient):
+    """ Client for validate members by EDR """
+
+    host_url = 'https://192.168.50.9'
+    api_version = '0.1'
+
+    def __init__(self,
+                 key,
+                 resource='lots',  # another possible value is 'assets'
+                 host_url=None,
+                 api_version=None,
+                 params=None,
+                 ds_client=None,
+                 user_agent=None):
+        super(RegistryClient, self).__init__(
+            key, resource, host_url, api_version, params, ds_client,
+            user_agent
+        )
+        self.host_url = host_url or self.host_url
+        self.api_version = api_version or self.api_version
+
+    def get_assets(self, extra_headers=None):
+        self.headers.update(extra_headers or {})
+        response = self.request(
+            'GET',
+            '{}/api/{}/assets'.format(self.host_url, self.api_version)
+        )
+        return process_response(response)
+
+    def get_lots(self, extra_headers=None):
+        resp = self.request('GET', 'http://localhost:20602/api/0.10/lots')
+
+        self.headers.update(extra_headers or {})
+        response = self.request(
+            'GET',
+            '{}/api/{}/lots'.format(self.host_url, self.api_version)
+        )
+        return process_response(response)
+
+    def get_asset(self, asset_id=None, extra_headers=None):
+        self.headers.update(extra_headers or {})
+
+        response = self.request(
+            'GET',
+            '{}/api/{}/assets/{}'.format(self.host_url, self.api_version, asset_id)
+        )
+        return process_response(response)
+
+    def get_lot(self, lot_id=None, extra_headers=None):
+        self.headers.update(extra_headers or {})
+
+        response = self.request(
+            'GET',
+            '{}/api/{}/lots/{}'.format(self.host_url, self.api_version, lot_id)
+        )
+        return process_response(response)
+
+    def patch_asset(self, asset):
+        return self._patch_resource_item(
+            '{}/{}'.format(self.prefix_path, asset['data']['id']),
+            payload=asset,
+            headers={'X-Access-Token': self._get_access_token(asset)}
+        )
+
+    def patch_lot(self, lot):
+        return self._patch_resource_item(
+            '{}/{}'.format(self.prefix_path, lot['data']['id']),
+            payload=lot,
+            headers={'X-Access-Token': self._get_access_token(lot)}
+        )
+
+
+class RegistryClientSync(RegistryClient):
+    def sync_item(self, params=None, extra_headers=None):
+        _params = (params or {}).copy()
+        _params['feed'] = 'changes'
+        self.headers.update(extra_headers or {})
+
+        response = self.request('GET', self.prefix_path,
+                                params_dict=_params)
+        if response.status_code == 200:
+            item_list = munchify(loads(response.text))
+            return item_list
+
+    @retry(stop_max_attempt_number=5)
+    def get_lot_by_id(self, id, extra_headers=None):
+        self.headers.update(extra_headers or {})
+        return super(RegistryClientSync, self).get_lot(id)
+
+    @retry(stop_max_attempt_number=5)
+    def get_asset_by_id(self, id, extra_headers=None):
+        self.headers.update(extra_headers or {})
+        return super(RegistryClientSync, self).get_asset(id)
+
+
+def process_response(response):
+    if response.status_code == 200:
+        return munchify(loads(response.text))
+    raise InvalidResponse(response)

openprocurement_client/tests/_server.py

@@ -1,10 +1,10 @@
-from bottle import request, response, redirect, static_file
+from bottle import request, response, redirect, static_file, run
 from munch import munchify
 from simplejson import dumps, load
 from openprocurement_client.document_service_client \
     import DocumentServiceClient
 from openprocurement_client.tests.data_dict import TEST_TENDER_KEYS, \
-    TEST_PLAN_KEYS, TEST_CONTRACT_KEYS
+    TEST_PLAN_KEYS, TEST_CONTRACT_KEYS, TEST_LOT_KEYS, TEST_ASSET_KEYS
 import magic
 import os
 
@@ -26,7 +26,9 @@
 RESOURCE_DICT = \
     {'tender':   {'sublink': 'tenders',   'data': TEST_TENDER_KEYS},
      'contract': {'sublink': 'contracts', 'data': TEST_CONTRACT_KEYS},
-     'plan':     {'sublink': 'plans',     'data': TEST_PLAN_KEYS}}
+     'plan':     {'sublink': 'plans',     'data': TEST_PLAN_KEYS},
+     'lot':     {'sublink': 'lots',     'data': TEST_LOT_KEYS},
+     'asset':   {'sublink': 'assets',   'data': TEST_ASSET_KEYS}}
 
 
 def resource_filter(resource_name):
@@ -330,6 +332,12 @@ def contract_change_patch(contract_id, change_id):
         "contract_change_patch": (API_PATH.format('contracts') + '/<contract_id>/changes/<change_id>', 'PATCH', contract_change_patch),
         "contract_patch": (API_PATH.format('<resource_name:resource_filter:contract>') + "/<resource_id>", 'PATCH', resource_patch),
         "contract_patch_credentials": (API_PATH.format('<resource_name:resource_filter:contract>') + '/<resource_id>/credentials', 'PATCH', patch_credentials),
+        "assets": (API_PATH.format('<resource_name:resource_filter:asset>'), 'GET', resource_page_get),
+        "asset": (API_PATH.format('<resource_name:resource_filter:asset>') + '/<resource_id>', 'GET', resource_page),
+        "asset_patch": (API_PATH.format('<resource_name:resource_filter:asset>') + "/<resource_id>", 'PATCH', resource_patch),
+        "lots": (API_PATH.format('<resource_name:resource_filter:lot>'), 'GET', resource_page_get),
+        "lot": (API_PATH.format('<resource_name:resource_filter:lot>') + '/<resource_id>', 'GET', resource_page),
+        "lot_patch": (API_PATH.format('<resource_name:resource_filter:lot>') + "/<resource_id>", 'PATCH', resource_patch),
         }