Commit dd046a1813133dce05e665e16de4d87c8253ed3d
Committed by
OrysiaDrabych
1 parent
959a77b6
Update openprocurement_client_helper.py
To work with contracting management procedure
Showing
1 changed file
with
11 additions
and
0 deletions
| @@ -2,6 +2,7 @@ from openprocurement_client.client import Client, EDRClient | @@ -2,6 +2,7 @@ from openprocurement_client.client import Client, EDRClient | ||
| 2 | from openprocurement_client.document_service_client \ | 2 | from openprocurement_client.document_service_client \ |
| 3 | import DocumentServiceClient | 3 | import DocumentServiceClient |
| 4 | from openprocurement_client.plan import PlansClient | 4 | from openprocurement_client.plan import PlansClient |
| 5 | +from openprocurement_client.contract import ContractingClient | ||
| 5 | from openprocurement_client.exceptions import IdNotFound | 6 | from openprocurement_client.exceptions import IdNotFound |
| 6 | from restkit.errors import RequestFailed, BadStatusLine, ResourceError | 7 | from restkit.errors import RequestFailed, BadStatusLine, ResourceError |
| 7 | from retrying import retry | 8 | from retrying import retry |
| @@ -43,6 +44,16 @@ def prepare_ds_api_wrapper(ds_host_url, auth_ds): | @@ -43,6 +44,16 @@ def prepare_ds_api_wrapper(ds_host_url, auth_ds): | ||
| 43 | return StableDsClient(ds_host_url, auth_ds) | 44 | return StableDsClient(ds_host_url, auth_ds) |
| 44 | 45 | ||
| 45 | 46 | ||
| 47 | +class ContractingStableClient(ContractingClient): | ||
| 48 | + @retry(stop_max_attempt_number=100, wait_random_min=500, wait_random_max=4000, retry_on_exception=retry_if_request_failed) | ||
| 49 | + def request(self, *args, **kwargs): | ||
| 50 | + return super(ContractingStableClient, self).request(*args, **kwargs) | ||
| 51 | + | ||
| 52 | + | ||
| 53 | +def prepare_contract_api_wrapper(key, host_url, api_version): | ||
| 54 | + return ContractingStableClient(key, host_url, api_version) | ||
| 55 | + | ||
| 56 | + | ||
| 46 | class StableEDRClient(EDRClient): | 57 | class StableEDRClient(EDRClient): |
| 47 | @retry(stop_max_attempt_number=100, wait_random_min=500, | 58 | @retry(stop_max_attempt_number=100, wait_random_min=500, |
| 48 | wait_random_max=4000, retry_on_exception=retry_if_request_failed) | 59 | wait_random_max=4000, retry_on_exception=retry_if_request_failed) |
Please
register
or
login
to post a comment