def cancel_order(self, order_id, order_type=None):
"""
Method executes cancelling_order according to it's order type.
return response.
"""
order = False
if not 'session_token' in self.__dict__:
raise ApiValueError(
"Please invoke 'generate_session2_fa' function first")
if order_type is None or order_type == 'O':
order = ks_api_client.OrderApi(self.api_client).cancel_order(
self.consumer_key, self.session_token, order_id)
elif order_type == 'NO':
order = ks_api_client.NormalOrderApi(
self.api_client).cancel_normal_order(self.consumer_key,
self.session_token,
order_id)
elif order_type == 'SMO':
order = ks_api_client.SuperMultipleOrderApi(
self.api_client).cancel_sm_order(self.consumer_key,
self.session_token, order_id)
elif order_type == 'SORO':
order = ks_api_client.SmartOrderRoutingApi(self.api_client).cancel_sor_order(self.consumer_key,\
self.session_token, order)
elif order_type == 'MTFO':
order = ks_api_client.MarginTradingApi(self.api_client).cancel_mtf_order(self.consumer_key, \
self.session_token, order)
else:
raise TypeError("Provided order type is invalid.")
return order
def __setattr__(self, name, value):
object.__setattr__(self, name, value)
if name == 'disabled_client_side_validations':
s = set(filter(None, value.split(',')))
for v in s:
if v not in JSON_SCHEMA_VALIDATION_KEYWORDS:
raise ApiValueError("Invalid keyword: '{0}''".format(v))
self._disabled_client_side_validations = s
def request(self, method, url, query_params=None, headers=None,
post_params=None, body=None, _preload_content=True,
_request_timeout=None):
"""Makes the HTTP request using RESTClient."""
if method == "GET":
return self.rest_client.GET(url,
query_params=query_params,
_preload_content=_preload_content,
_request_timeout=_request_timeout,
headers=headers)
elif method == "HEAD":
return self.rest_client.HEAD(url,
query_params=query_params,
_preload_content=_preload_content,
_request_timeout=_request_timeout,
headers=headers)
elif method == "OPTIONS":
return self.rest_client.OPTIONS(url,
query_params=query_params,
headers=headers,
_preload_content=_preload_content,
_request_timeout=_request_timeout)
elif method == "POST":
return self.rest_client.POST(url,
query_params=query_params,
headers=headers,
post_params=post_params,
_preload_content=_preload_content,
_request_timeout=_request_timeout,
body=body)
elif method == "PUT":
return self.rest_client.PUT(url,
query_params=query_params,
headers=headers,
post_params=post_params,
_preload_content=_preload_content,
_request_timeout=_request_timeout,
body=body)
elif method == "PATCH":
return self.rest_client.PATCH(url,
query_params=query_params,
headers=headers,
post_params=post_params,
_preload_content=_preload_content,
_request_timeout=_request_timeout,
body=body)
elif method == "DELETE":
return self.rest_client.DELETE(url,
query_params=query_params,
headers=headers,
_preload_content=_preload_content,
_request_timeout=_request_timeout,
body=body)
else:
raise ApiValueError(
"http method must be `GET`, `HEAD`, `OPTIONS`,"
" `POST`, `PATCH`, `PUT` or `DELETE`."
)
def margin_required(self, transaction_type, order_info):
req_margin = ReqMargin(transactionType=transaction_type,
orderInfo=self.convertArray(order_info))
if not isinstance(req_margin, ReqMargin):
raise ApiValueError("ReqMargin with type ", type(req_margin),
" is not valid.")
margin_required = ks_api_client.MarginApi(
self.api_client).margin_required(self.consumer_key,
self.session_token,
ReqMargin=req_margin)
return margin_required
def place_order(self, order_type, instrument_token, transaction_type, \
quantity, price, disclosed_quantity = 0,trigger_price = 0 , tag = "string", \
validity ="GFD", variety = "REGULAR", product = None, smart_order_routing = None):
"""
Method executes placing_orders according to it's order type.
return response.
"""
place_order = None
if not 'session_token' in self.__dict__:
raise ApiValueError("Please invoke 'session_2fa' function first")
if order_type == "O":
order = NewOrder(instrumentToken = instrument_token, tag = tag, transactionType = transaction_type,\
variety = variety, quantity = quantity, price = price, disclosedQuantity = disclosed_quantity,\
validity = validity, triggerPrice = trigger_price, product = product,smartOrderRouting = smart_order_routing)
place_order = ks_api_client.OrderApi(
self.api_client).place_new_order(self.consumer_key,
self.session_token, order)
elif order_type == 'N':
order = NewNormalOrder(instrumentToken = instrument_token, tag = tag, transactionType = transaction_type,\
variety = variety, quantity = quantity, price = price, disclosedQuantity = disclosed_quantity,\
validity = validity, triggerPrice = trigger_price)
place_order = ks_api_client.NormalOrderApi(
self.api_client).place_new_normal_order(
self.consumer_key, self.session_token, order)
elif order_type == 'SM':
order = NewSMOrder( instrumentToken = instrument_token, tag = tag, transactionType = transaction_type,\
variety = variety, quantity = quantity, price = price, disclosedQuantity = disclosed_quantity, \
validity = validity, triggerPrice = trigger_price)
place_order = ks_api_client.SuperMultipleOrderApi(self.api_client).place_new_sm_order(self.consumer_key,\
self.session_token, order)
elif order_type == 'SOR':
order = NewSOROrder( instrumentToken = instrument_token, tag = tag, transactionType = transaction_type,\
variety = variety, quantity = quantity, price = price, validity = validity, disclosedQuantity = disclosed_quantity,triggerPrice = trigger_price )
place_order = ks_api_client.SmartOrderRoutingApi(self.api_client).place_new_sor_order(self.consumer_key,\
self.session_token, order)
elif order_type == 'MTF':
order = NewMTFOrder(instrumentToken = instrument_token, tag = tag, transactionType = transaction_type,\
variety = variety, quantity = quantity, price = price, disclosedQuantity = disclosed_quantity, \
validity = validity, triggerPrice = trigger_price)
place_order = ks_api_client.MarginTradingApi(self.api_client).place_new_mtf_order(self.consumer_key, \
self.session_token, order)
elif order_type == 'MIS':
order = NewMISOrder(instrumentToken = instrument_token, tag = tag, transactionType = transaction_type,\
variety = variety, quantity = quantity, price = price, disclosedQuantity = disclosed_quantity,\
validity = validity, triggerPrice = trigger_price)
place_order = ks_api_client.MISOrderApi(
self.api_client).place_new_mis_order(self.consumer_key,
self.session_token, order)
else:
raise TypeError(
"Provided order type is invalid, use either of O(Order), N(Normal Order), SM(Super Multiple Order), SOR(Smart Order Routing), MTF(Margin Tading Facility) or MIS"
)
return place_order
def cancel_order(self, order_id):
"""
Method executes cancelling_order
return response.
"""
order = False
if not 'session_token' in self.__dict__:
raise ApiValueError("Please invoke 'session_2fa' function first")
order = ks_api_client.OrderApi(self.api_client).cancel_order(
self.consumer_key, self.session_token, order_id)
return order
def history(self, resource, json_input):
if not 'session_token' in self.__dict__:
raise ApiValueError("Please invoke 'session_2fa' function first")
if (resource == 'historicalprices'):
if (json_input.keys() != {
"exchange": "",
"cocode": "",
"fromdate": "",
"todate": ""
}.keys()):
raise ApiValueError(
"exchange,cocode,fromdate,todate fields are required.")
elif (resource == 'historicalprices-unadjusted'):
if (json_input.keys() != {
"exchange": "",
"co_code": "",
"date": ""
}.keys()):
raise ApiValueError(
"exchange,co_code,date fields are required.")
elif (resource == 'NSEFNO_HistoricalContinuousChart'):
if (json_input.keys() != {"symbol": "", "expiry type": ""}.keys()):
raise ApiValueError("symbol,expiry type fields are required.")
elif (resource == 'LiveorEODHistorical'):
if (json_input.keys() != {
"exchange": "",
"co_code": "",
"period": "",
"cnt": ""
}.keys()):
raise ApiValueError(
"exchange,co_code,period,cnt fields are required.")
encoded_json = base64.urlsafe_b64encode(
json.dumps(json_input).encode()).decode()
data = ks_api_client.HistoricalApi(self.api_client).get_resource(
resource, encoded_json)
return data
def generate_session2_fa(self, access_code):
user_details = UserDetails(userid=self.userid, accessCode=access_code)
if self.one_time_token:
generate_session_res = ks_api_client.SessionApi(self.api_client).generate_session2_fa(self.one_time_token, \
self.consumer_key, self.ip, self.app_id, UserDetails = user_details)
generate_session = ''
if (generate_session_res.get("Success")):
generate_session = generate_session_res.get("Success")
elif (generate_session_res.get("success")):
generate_session = generate_session_res.get("success")
self.session_token = generate_session['sessionToken']
return generate_session
else:
raise ApiValueError(
"Please invoke 'session_login_api' function first")
def modify_order(self, order_id, price, quantity, disclosed_quantity,
trigger_price, validity):
"""
Method executes modifying_orders
return response.
"""
if not 'session_token' in self.__dict__:
raise ApiValueError("Please invoke 'session_2fa' function first")
modify_order = ExistingOrder(orderId = order_id, disclosedQuantity = disclosed_quantity, price = price,\
quantity = quantity, triggerPrice = trigger_price,validity = validity)
modified_order_res = ks_api_client.OrderApi(
self.api_client).modify_order(self.consumer_key,
self.session_token, modify_order)
return modified_order_res
def _apply_auth_params(self, headers, querys, auth_setting):
"""Updates the request parameters based on a single auth_setting
:param headers: Header parameters dict to be updated.
:param querys: Query parameters tuple list to be updated.
:param auth_setting: auth settings for the endpoint
"""
if auth_setting['in'] == 'cookie':
headers['Cookie'] = auth_setting['value']
elif auth_setting['in'] == 'header':
headers[auth_setting['key']] = auth_setting['value']
elif auth_setting['in'] == 'query':
querys.append((auth_setting['key'], auth_setting['value']))
else:
raise ApiValueError(
'Authentication token must be in `query` or `header`')
def modify_order(self, order_id, disclosed_quantity = 0, price = 0,\
quantity = 1, trigger_price = 0,order_type = None):
"""
Method executes modifying_orders according to it's order type.
return response.
"""
if not 'session_token' in self.__dict__:
raise ApiValueError(
"Please invoke 'generate_session2_fa' function first")
if order_type is None or order_type == 'O':
modify_order = ExistingOrder(orderId = order_id, disclosedQuantity = disclosed_quantity, price = price,\
quantity = quantity, triggerPrice = trigger_price)
modified_order_res = ks_api_client.OrderApi(
self.api_client).modify_order(self.consumer_key,
self.session_token, modify_order)
elif order_type == 'NO':
modify_order = ExistingNormalOrder(orderId = order_id, disclosedQuantity = disclosed_quantity, price = price,\
quantity = quantity, triggerPrice = trigger_price)
modified_order_res = ks_api_client.NormalOrderApi(
self.api_client).modify_normal_order(self.consumer_key,
self.session_token,
modify_order)
elif order_type == 'SMO':
modify_order = ExistingSMOrder(orderId = order_id, disclosedQuantity = disclosed_quantity, price = price,\
quantity = quantity, triggerPrice = trigger_price)
modified_order_res = ks_api_client.SuperMultipleOrderApi(
self.api_client).modify_sm_order(self.consumer_key,
self.session_token,
modify_order)
elif order_type == 'SORO':
modify_order = ExistingSOROrder(orderId = order_id, disclosedQuantity = disclosed_quantity, price = price,\
quantity = quantity, triggerPrice = trigger_price)
modified_order_res = ks_api_client.SmartOrderRoutingApi(self.api_client).modify_sor_order(self.consumer_key,\
self.session_token, modify_order)
elif order_type == 'MTFO':
modify_order = ExistingMTFOrder(orderId = order_id, disclosedQuantity = disclosed_quantity, price = price,\
quantity = quantity, triggerPrice = trigger_price)
modified_order_res = ks_api_client.MarginTradingApi(self.api_client).modify_mtf_order(self.consumer_key, \
self.session_token, modify_order)
else:
raise TypeError("Provided order type is invalid.")
return modified_order_res
def cancel_mtf_order_with_http_info(self, consumerKey, sessionToken,
orderId, **kwargs): # noqa: E501
"""Cancel an order # noqa: E501
This method makes a synchronous HTTP request by default. To make an
asynchronous HTTP request, please pass async_req=True
>>> thread = api.cancel_mtf_order_with_http_info(consumerKey, sessionToken, orderId, async_req=True)
>>> result = thread.get()
:param consumerKey: (required)
:type consumerKey: str
:param sessionToken: (required)
:type sessionToken: str
:param orderId: Order ID to cancel. (required)
:type orderId: str
:param async_req: Whether to execute the request asynchronously.
:type async_req: bool, optional
:param _return_http_data_only: response data without head status code
and headers
:type _return_http_data_only: bool, optional
:param _preload_content: if False, the urllib3.HTTPResponse object will
be returned without reading/decoding response
data. Default is True.
:type _preload_content: bool, optional
:param _request_timeout: timeout setting for this request. If one
number provided, it will be total request
timeout. It can also be a pair (tuple) of
(connection, read) timeouts.
:param _request_auth: set to override the auth_settings for an a single
request; this effectively ignores the authentication
in the spec for a single request.
:type _request_auth: dict, optional
:return: Returns the result object.
If the method is called asynchronously,
returns the request thread.
:rtype: tuple(object, status_code(int), headers(HTTPHeaderDict))
"""
local_var_params = locals()
all_params = ['consumerKey', 'sessionToken', 'orderId']
all_params.extend([
'async_req', '_return_http_data_only', '_preload_content',
'_request_timeout', '_request_auth'
])
for key, val in six.iteritems(local_var_params['kwargs']):
if key not in all_params:
raise ApiTypeError("Got an unexpected keyword argument '%s'"
" to method cancel_mtf_order" % key)
local_var_params[key] = val
del local_var_params['kwargs']
# verify the required parameter 'consumerKey' is set
if self.api_client.client_side_validation and (
'consumerKey' not in local_var_params or # noqa: E501
local_var_params['consumerKey'] is None): # noqa: E501
raise ApiValueError(
"Missing the required parameter `consumerKey` when calling `cancel_mtf_order`"
) # noqa: E501
# verify the required parameter 'sessionToken' is set
if self.api_client.client_side_validation and (
'sessionToken' not in local_var_params or # noqa: E501
local_var_params['sessionToken'] is None): # noqa: E501
raise ApiValueError(
"Missing the required parameter `sessionToken` when calling `cancel_mtf_order`"
) # noqa: E501
# verify the required parameter 'orderId' is set
if self.api_client.client_side_validation and (
'orderId' not in local_var_params or # noqa: E501
local_var_params['orderId'] is None): # noqa: E501
raise ApiValueError(
"Missing the required parameter `orderId` when calling `cancel_mtf_order`"
) # noqa: E501
collection_formats = {}
path_params = {}
if 'orderId' in local_var_params:
path_params['orderId'] = local_var_params['orderId'] # noqa: E501
query_params = []
header_params = {}
if 'consumerKey' in local_var_params:
header_params['consumerKey'] = local_var_params[
'consumerKey'] # noqa: E501
if 'sessionToken' in local_var_params:
header_params['sessionToken'] = local_var_params[
'sessionToken'] # noqa: E501
form_params = []
local_var_files = {}
body_params = None
# HTTP header `Accept`
header_params['Accept'] = self.api_client.select_header_accept(
['application/json']) # noqa: E501
# Authentication setting
auth_settings = ['bearerAuth'] # noqa: E501
return self.api_client.call_api(
'/orders/1.0/order/mtf/{orderId}',
'DELETE',
path_params,
query_params,
header_params,
body=body_params,
post_params=form_params,
files=local_var_files,
response_type='object', # noqa: E501
auth_settings=auth_settings,
async_req=local_var_params.get('async_req'),
_return_http_data_only=local_var_params.get(
'_return_http_data_only'), # noqa: E501
_preload_content=local_var_params.get('_preload_content', True),
_request_timeout=local_var_params.get('_request_timeout'),
collection_formats=collection_formats,
_request_auth=local_var_params.get('_request_auth'))
def login_with_user_id_with_http_info(self, consumerKey, ip, appId,
**kwargs): # noqa: E501
"""Login using Userid # noqa: E501
Authenticate userid and password to gnerrated one time token # noqa: E501
This method makes a synchronous HTTP request by default. To make an
asynchronous HTTP request, please pass async_req=True
>>> thread = api.login_with_user_id_with_http_info(consumerKey, ip, appId, async_req=True)
>>> result = thread.get()
:param consumerKey: (required)
:type consumerKey: str
:param ip: (required)
:type ip: str
:param appId: (required)
:type appId: str
:param UserCredentials:
:type UserCredentials: UserCredentials
:param async_req: Whether to execute the request asynchronously.
:type async_req: bool, optional
:param _return_http_data_only: response data without head status code
and headers
:type _return_http_data_only: bool, optional
:param _preload_content: if False, the urllib3.HTTPResponse object will
be returned without reading/decoding response
data. Default is True.
:type _preload_content: bool, optional
:param _request_timeout: timeout setting for this request. If one
number provided, it will be total request
timeout. It can also be a pair (tuple) of
(connection, read) timeouts.
:param _request_auth: set to override the auth_settings for an a single
request; this effectively ignores the authentication
in the spec for a single request.
:type _request_auth: dict, optional
:return: Returns the result object.
If the method is called asynchronously,
returns the request thread.
:rtype: tuple(object, status_code(int), headers(HTTPHeaderDict))
"""
local_var_params = locals()
all_params = ['consumerKey', 'ip', 'appId', 'UserCredentials']
all_params.extend([
'async_req', '_return_http_data_only', '_preload_content',
'_request_timeout', '_request_auth'
])
for key, val in six.iteritems(local_var_params['kwargs']):
if key not in all_params:
raise ApiTypeError("Got an unexpected keyword argument '%s'"
" to method login_with_user_id" % key)
local_var_params[key] = val
del local_var_params['kwargs']
# verify the required parameter 'consumerKey' is set
if self.api_client.client_side_validation and (
'consumerKey' not in local_var_params or # noqa: E501
local_var_params['consumerKey'] is None): # noqa: E501
raise ApiValueError(
"Missing the required parameter `consumerKey` when calling `login_with_user_id`"
) # noqa: E501
# verify the required parameter 'ip' is set
if self.api_client.client_side_validation and (
'ip' not in local_var_params or # noqa: E501
local_var_params['ip'] is None): # noqa: E501
raise ApiValueError(
"Missing the required parameter `ip` when calling `login_with_user_id`"
) # noqa: E501
# verify the required parameter 'appId' is set
if self.api_client.client_side_validation and (
'appId' not in local_var_params or # noqa: E501
local_var_params['appId'] is None): # noqa: E501
raise ApiValueError(
"Missing the required parameter `appId` when calling `login_with_user_id`"
) # noqa: E501
collection_formats = {}
path_params = {}
query_params = []
header_params = {}
if 'consumerKey' in local_var_params:
header_params['consumerKey'] = local_var_params[
'consumerKey'] # noqa: E501
if 'ip' in local_var_params:
header_params['ip'] = local_var_params['ip'] # noqa: E501
if 'appId' in local_var_params:
header_params['appId'] = local_var_params['appId'] # noqa: E501
form_params = []
local_var_files = {}
body_params = None
if 'UserCredentials' in local_var_params:
body_params = local_var_params['UserCredentials']
# HTTP header `Accept`
header_params['Accept'] = self.api_client.select_header_accept(
['application/json']) # noqa: E501
# HTTP header `Content-Type`
header_params[
'Content-Type'] = self.api_client.select_header_content_type( # noqa: E501
['application/json']) # noqa: E501
# Authentication setting
auth_settings = ['bearerAuth'] # noqa: E501
return self.api_client.call_api(
'/session/1.0/session/login/userid',
'POST',
path_params,
query_params,
header_params,
body=body_params,
post_params=form_params,
files=local_var_files,
response_type='object', # noqa: E501
auth_settings=auth_settings,
async_req=local_var_params.get('async_req'),
_return_http_data_only=local_var_params.get(
'_return_http_data_only'), # noqa: E501
_preload_content=local_var_params.get('_preload_content', True),
_request_timeout=local_var_params.get('_request_timeout'),
collection_formats=collection_formats,
_request_auth=local_var_params.get('_request_auth'))
def get_resource_with_http_info(self, resource, input,
**kwargs): # noqa: E501
"""Get historical data # noqa: E501
Get Historical data # noqa: E501
This method makes a synchronous HTTP request by default. To make an
asynchronous HTTP request, please pass async_req=True
>>> thread = api.get_resource_with_http_info(resource, input, async_req=True)
>>> result = thread.get()
:param resource: (required)
:type resource: str
:param input: (required)
:type input: str
:param async_req: Whether to execute the request asynchronously.
:type async_req: bool, optional
:param _return_http_data_only: response data without head status code
and headers
:type _return_http_data_only: bool, optional
:param _preload_content: if False, the urllib3.HTTPResponse object will
be returned without reading/decoding response
data. Default is True.
:type _preload_content: bool, optional
:param _request_timeout: timeout setting for this request. If one
number provided, it will be total request
timeout. It can also be a pair (tuple) of
(connection, read) timeouts.
:param _request_auth: set to override the auth_settings for an a single
request; this effectively ignores the authentication
in the spec for a single request.
:type _request_auth: dict, optional
:return: Returns the result object.
If the method is called asynchronously,
returns the request thread.
:rtype: tuple(object, status_code(int), headers(HTTPHeaderDict))
"""
local_var_params = locals()
all_params = ['resource', 'input']
all_params.extend([
'async_req', '_return_http_data_only', '_preload_content',
'_request_timeout', '_request_auth'
])
for key, val in six.iteritems(local_var_params['kwargs']):
if key not in all_params:
raise ApiTypeError("Got an unexpected keyword argument '%s'"
" to method get_resource" % key)
local_var_params[key] = val
del local_var_params['kwargs']
# verify the required parameter 'resource' is set
if self.api_client.client_side_validation and (
'resource' not in local_var_params or # noqa: E501
local_var_params['resource'] is None): # noqa: E501
raise ApiValueError(
"Missing the required parameter `resource` when calling `get_resource`"
) # noqa: E501
# verify the required parameter 'input' is set
if self.api_client.client_side_validation and (
'input' not in local_var_params or # noqa: E501
local_var_params['input'] is None): # noqa: E501
raise ApiValueError(
"Missing the required parameter `input` when calling `get_resource`"
) # noqa: E501
collection_formats = {}
path_params = {}
if 'resource' in local_var_params:
path_params['resource'] = local_var_params[
'resource'] # noqa: E501
if 'input' in local_var_params:
path_params['input'] = local_var_params['input'] # noqa: E501
query_params = []
header_params = {}
form_params = []
local_var_files = {}
body_params = None
# HTTP header `Accept`
header_params['Accept'] = self.api_client.select_header_accept(
['application/json']) # noqa: E501
# Authentication setting
auth_settings = ['bearerAuth'] # noqa: E501
return self.api_client.call_api(
'/trade/1.0.0/equity/{resource}/i/{input}',
'GET',
path_params,
query_params,
header_params,
body=body_params,
post_params=form_params,
files=local_var_files,
response_type='object', # noqa: E501
auth_settings=auth_settings,
async_req=local_var_params.get('async_req'),
_return_http_data_only=local_var_params.get(
'_return_http_data_only'), # noqa: E501
_preload_content=local_var_params.get('_preload_content', True),
_request_timeout=local_var_params.get('_request_timeout'),
collection_formats=collection_formats,
_request_auth=local_var_params.get('_request_auth'))
def request(self,
method,
url,
query_params=None,
headers=None,
body=None,
post_params=None,
_preload_content=True,
_request_timeout=None):
"""Perform requests.
:param method: http request method
:param url: http request url
:param query_params: query parameters in the url
:param headers: http request headers
:param body: request json body, for `application/json`
:param post_params: request post parameters,
`application/x-www-form-urlencoded`
and `multipart/form-data`
:param _preload_content: if False, the urllib3.HTTPResponse object will
be returned without reading/decoding response
data. Default is True.
:param _request_timeout: timeout setting for this request. If one
number provided, it will be total request
timeout. It can also be a pair (tuple) of
(connection, read) timeouts.
"""
method = method.upper()
assert method in [
'GET', 'HEAD', 'DELETE', 'POST', 'PUT', 'PATCH', 'OPTIONS'
]
if post_params and body:
raise ApiValueError(
"body parameter cannot be used with post_params parameter.")
post_params = post_params or {}
headers = headers or {}
timeout = None
if _request_timeout:
if isinstance(_request_timeout,
six.integer_types + (float, )): # noqa: E501,F821
timeout = urllib3.Timeout(total=_request_timeout)
elif (isinstance(_request_timeout, tuple)
and len(_request_timeout) == 2):
timeout = urllib3.Timeout(connect=_request_timeout[0],
read=_request_timeout[1])
if 'Content-Type' not in headers:
headers['Content-Type'] = 'application/json'
try:
# For `POST`, `PUT`, `PATCH`, `OPTIONS`, `DELETE`
if method in ['POST', 'PUT', 'PATCH', 'OPTIONS', 'DELETE']:
if query_params:
url += '?' + urlencode(query_params)
if re.search('json', headers['Content-Type'], re.IGNORECASE):
request_body = None
if body is not None:
request_body = json.dumps(body)
r = self.pool_manager.request(
method,
url,
body=request_body,
preload_content=_preload_content,
timeout=timeout,
headers=headers)
elif headers[
'Content-Type'] == 'application/x-www-form-urlencoded': # noqa: E501
r = self.pool_manager.request(
method,
url,
fields=post_params,
encode_multipart=False,
preload_content=_preload_content,
timeout=timeout,
headers=headers)
elif headers['Content-Type'] == 'multipart/form-data':
# must del headers['Content-Type'], or the correct
# Content-Type which generated by urllib3 will be
# overwritten.
del headers['Content-Type']
r = self.pool_manager.request(
method,
url,
fields=post_params,
encode_multipart=True,
preload_content=_preload_content,
timeout=timeout,
headers=headers)
# Pass a `string` parameter directly in the body to support
# other content types than Json when `body` argument is
# provided in serialized form
elif isinstance(body, str) or isinstance(body, bytes):
request_body = body
r = self.pool_manager.request(
method,
url,
body=request_body,
preload_content=_preload_content,
timeout=timeout,
headers=headers)
else:
# Cannot generate the request from given parameters
msg = """Cannot prepare a request message for provided
arguments. Please check that your arguments match
declared content type."""
raise ApiException(status=0, reason=msg)
# For `GET`, `HEAD`
else:
r = self.pool_manager.request(method,
url,
fields=query_params,
preload_content=_preload_content,
timeout=timeout,
headers=headers)
except urllib3.exceptions.SSLError as e:
msg = "{0}\n{1}".format(type(e).__name__, str(e))
raise ApiException(status=0, reason=msg)
if _preload_content:
r = RESTResponse(r)
# log response body
logger.debug("response body: %s", r.data)
if not 200 <= r.status <= 299:
raise ApiException(status=r.status, reason=r.reason, body=r.data)
return r
