catocli 2.0.4__py3-none-any.whl → 2.0.5__py3-none-any.whl

build/lib/graphql_client/__init__.py

@@ -0,0 +1,11 @@
+# coding: utf-8
+
+from __future__ import absolute_import
+
+# # import api into package
+from graphql_client.api.call_api import CallApi
+
+# import ApiClient
+from graphql_client.api_client_types import ApiClient
+from graphql_client.configuration import Configuration
+from graphql_client.models.no_schema import NoSchema

build/lib/graphql_client/api/__init__.py

@@ -0,0 +1,3 @@
+from __future__ import absolute_import
+
+from graphql_client.api.call_api import CallApi

build/lib/graphql_client/api/call_api.py

@@ -0,0 +1,84 @@
+# coding: utf-8
+
+from __future__ import absolute_import
+import re
+import json
+
+# python 2 and python 3 compatibility library
+import sys
+sys.path.insert(0, 'vendor')
+import six
+from graphql_client.api_client_types import ApiClient
+
+class CallApi(object):
+	def __init__(self, api_client=None):
+		if api_client is None:
+			api_client = ApiClient()
+		self.api_client = api_client
+
+	def call_api(self, body, args, **kwargs):  # noqa: E501
+		(data) = self.call_api_with_http_info(body, args, **kwargs)  # noqa: E501
+		return data
+	
+	def call_api_with_http_info(self, body, args, **kwargs):  # noqa: E501
+		all_params = ['body', 'sync_type']  # noqa: E501
+		all_params.append('async_req')
+		if args.get("v")==True:
+			all_params.append('_return_http_data_only')
+			all_params.append('_preload_content')
+			all_params.append('_request_timeout')
+
+		params = locals()
+		for key, val in six.iteritems(params['kwargs']):
+			if key not in all_params:
+				raise TypeError(
+					"Got an unexpected keyword argument '%s'"
+					" to method create_asset4" % key
+				)
+			params[key] = val
+		del params['kwargs']
+
+		if ('body' not in params or params['body'] is None):
+			raise ValueError("Missing the required parameter `body` when calling `create_asset4`")  
+
+		collection_formats = {}
+
+		header_params = {}
+
+		form_params = []
+		local_var_files = {}
+
+		body_params = None
+		if 'body' in params:
+			body_params = params['body']
+			header_params['Accept'] = self.api_client.select_header_accept(['application/json'])
+			header_params['Content-Type'] = self.api_client.select_header_content_type(['application/json'])
+			# Only add x-api-key if not using custom headers
+			using_custom_headers = hasattr(self.api_client.configuration, 'custom_headers') and self.api_client.configuration.custom_headers
+			if not using_custom_headers and 'x-api-key' in self.api_client.configuration.api_key:
+				header_params['x-api-key'] = self.api_client.configuration.api_key['x-api-key']
+			header_params['User-Agent'] = "Cato-CLI-v"+self.api_client.configuration.version
+		
+		# Add custom headers from configuration
+		if using_custom_headers:
+			header_params.update(self.api_client.configuration.custom_headers)
+
+		if args.get("v")==True:
+			print("Host: ",self.api_client.configuration.host)
+			# Create a copy of header_params with masked API key for verbose output
+			masked_headers = header_params.copy()
+			if 'x-api-key' in masked_headers:
+				masked_headers['x-api-key'] = '***MASKED***'
+			print("Request Headers:",json.dumps(masked_headers,indent=4,sort_keys=True))
+			print("Request Data:",json.dumps(body_params,indent=4,sort_keys=True),"\n\n")
+			
+		return self.api_client.call_api(
+			header_params,
+			body=body_params,
+			files=local_var_files,
+			response_type="NoSchema",  # noqa: E501 
+			async_req=params.get('async_req'),
+			_return_http_data_only=params.get('_return_http_data_only'),
+			_preload_content=params.get('_preload_content', True),
+			_request_timeout=params.get('_request_timeout'),
+			collection_formats=collection_formats)

build/lib/graphql_client/api_client.py

@@ -0,0 +1,192 @@
+# coding: utf-8
+
+from __future__ import absolute_import
+
+import io
+import json
+import logging
+import re
+import ssl
+
+# python 2 and python 3 compatibility library
+import sys
+sys.path.insert(0, 'vendor')
+import certifi
+import six
+from six.moves.urllib.parse import urlencode
+
+try:
+	import urllib3
+except ImportError:
+	raise ImportError('GraphQL python client requires urllib3.')
+
+logger = logging.getLogger(__name__)
+
+class QUERYResponse(io.IOBase):
+
+	def __init__(self, resp):
+		self.urllib3_response = resp
+		self.status = resp.status
+		self.reason = resp.reason
+		self.data = resp.data
+
+	def getheaders(self):
+		"""Returns a dictionary of the response headers."""
+		return self.urllib3_response.getheaders()
+
+	def getheader(self, name, default=None):
+		"""Returns a given response header."""
+		return self.urllib3_response.getheader(name, default)
+
+class QUERYClientObject(object):
+
+	def __init__(self, configuration, pools_size=4, maxsize=None):
+		# urllib3.PoolManager will pass all kw parameters to connectionpool
+		# https://github.com/shazow/urllib3/blob/f9409436f83aeb79fbaf090181cd81b784f1b8ce/urllib3/poolmanager.py#L75  # noqa: E501
+		# https://github.com/shazow/urllib3/blob/f9409436f83aeb79fbaf090181cd81b784f1b8ce/urllib3/connectionpool.py#L680  # noqa: E501
+		# maxsize is the number of requests to host that are allowed in parallel  # noqa: E501
+		# Custom SSL certificates and client certificates: http://urllib3.readthedocs.io/en/latest/advanced-usage.html  # noqa: E501
+
+		# cert_reqs
+		if configuration.verify_ssl:
+			cert_reqs = ssl.CERT_REQUIRED
+		else:
+			cert_reqs = ssl.CERT_NONE
+
+		# ca_certs
+		if configuration.ssl_ca_cert:
+			ca_certs = configuration.ssl_ca_cert
+		else:
+			# if not set certificate file, use Mozilla's root certificates.
+			ca_certs = certifi.where()
+
+		addition_pool_args = {}
+		if configuration.assert_hostname is not None:
+			addition_pool_args['assert_hostname'] = configuration.assert_hostname  # noqa: E501
+
+		if maxsize is None:
+			if configuration.connection_pool_maxsize is not None:
+				maxsize = configuration.connection_pool_maxsize
+			else:
+				maxsize = 4
+
+		# https pool manager
+		if configuration.proxy:
+			self.pool_manager = urllib3.ProxyManager(
+				num_pools=pools_size,
+				maxsize=maxsize,
+				cert_reqs=cert_reqs,
+				ca_certs=ca_certs,
+				cert_file=configuration.cert_file,
+				key_file=configuration.key_file,
+				proxy_url=configuration.proxy,
+				**addition_pool_args
+			)
+		else:
+			self.pool_manager = urllib3.PoolManager(
+				num_pools=pools_size,
+				maxsize=maxsize,
+				cert_reqs=cert_reqs,
+				ca_certs=ca_certs,
+				cert_file=configuration.cert_file,
+				key_file=configuration.key_file,
+				**addition_pool_args
+			)
+
+	def request(self, method, url, headers=None,
+				body=None, _preload_content=True,
+				_request_timeout=None):
+		"""Perform requests.
+		:param headers: http request headers
+		:param body: request json body, for `application/json`
+		: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 ['POST']
+		headers = headers or {}
+
+		timeout = None
+		if _request_timeout:
+			if isinstance(_request_timeout, (int, ) if six.PY3 else (int, long)):  # 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:
+			if re.search('json', headers['Content-Type'], re.IGNORECASE):
+				try:
+					r = self.pool_manager.request( method=method, url=url,
+						body=json.dumps(body),
+						preload_content=_preload_content,
+						timeout=timeout,
+						headers=headers)
+				except Exception as e:
+					# remove the restriction for raw 
+					print("ERROR: ",e)
+					exit()
+				
+			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)
+			
+		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 = QUERYResponse(r)
+
+			# log response body
+			logger.debug("response body: %s", r.data)
+
+		if not 200 <= r.status <= 299:
+			raise ApiException(http_resp=r)
+
+		return r
+
+	def POST(self, url, headers=None, body=None, _preload_content=True, _request_timeout=None):
+		return self.request("POST", url, headers=headers,
+							_preload_content=_preload_content,
+							_request_timeout=_request_timeout,
+							body=body)
+
+class ApiException(Exception):
+
+	def __init__(self, status=None, reason=None, http_resp=None):
+		if http_resp:
+			self.status = http_resp.status
+			self.reason = http_resp.reason
+			self.body = http_resp.data
+			self.headers = http_resp.getheaders()
+		else:
+			self.status = status
+			self.reason = reason
+			self.body = None
+			self.headers = None
+
+	def __str__(self):
+		"""Custom error messages for exception"""
+		if self.body:
+			# error_message += "HTTP response body:\n{}".format(self.body.decode('utf8'))
+			return self.body.decode('utf8')
+		else:
+			error_message = "({0})\n"\
+							"Reason: {1}\n".format(self.status, self.reason)
+			if self.headers:
+				error_message += "HTTP response headers: {0}\n".format(
+					self.headers)
+

build/lib/graphql_client/api_client_types.py

@@ -0,0 +1,409 @@
+# coding: utf-8
+
+from __future__ import absolute_import
+
+import datetime
+import json
+import mimetypes
+from multiprocessing.pool import ThreadPool
+import os
+import re
+import tempfile
+
+from graphql_client.configuration import Configuration
+import graphql_client.models
+from graphql_client import api_client
+
+# python 2 and python 3 compatibility library
+import sys
+sys.path.insert(0, 'vendor')
+import six
+
+class ApiClient(object):
+	PRIMITIVE_TYPES = (float, bool, bytes, six.text_type) + six.integer_types
+	NATIVE_TYPES_MAPPING = {
+		'int': int,
+		'long': int if six.PY3 else long,
+		'float': float,
+		'str': str,
+		'bool': bool,
+		'date': datetime.date,
+		'datetime': datetime.datetime,
+		'object': object,
+	}
+
+	def __init__(self, configuration=None, header_name=None, header_value=None,
+				 cookie=None):
+		if configuration is None:
+			configuration = Configuration()
+		self.configuration = configuration
+
+		self.pool = ThreadPool()
+		self.rest_client = api_client.QUERYClientObject(configuration)
+		self.default_headers = {}
+		if header_name is not None:
+			self.default_headers[header_name] = header_value
+		self.cookie = cookie
+		self.user_agent = "Cato-CLI-v"+self.configuration.version
+
+	def __del__(self):
+		self.pool.close()
+		self.pool.join()
+
+	@property
+	def user_agent(self):
+		"""User agent for this API client"""
+		return self.default_headers['User-Agent']
+
+	@user_agent.setter
+	def user_agent(self, value):
+		self.default_headers['User-Agent'] = value
+
+	def set_default_header(self, header_name, header_value):
+		self.default_headers[header_name] = header_value
+
+	def __call_api(
+			self, header_params=None, body=None,
+			files=None, response_type=None,
+			_return_http_data_only=None, collection_formats=None,
+			_preload_content=True, _request_timeout=None):
+
+		config = self.configuration
+
+		# header parameters
+		header_params = header_params or {}
+		header_params.update(self.default_headers)
+		if self.cookie:
+			header_params['Cookie'] = self.cookie
+		if header_params:
+			header_params = self.sanitize_for_serialization(header_params)
+			header_params = dict(self.parameters_to_tuples(header_params,
+														   collection_formats))
+
+		# body
+		if body:
+			body = self.sanitize_for_serialization(body)
+
+		# request url
+		url = self.configuration.host
+
+		# perform request and return response
+		response_data = self.request(
+			 url, headers=header_params, body=body,
+			_preload_content=_preload_content,
+			_request_timeout=_request_timeout)
+
+		self.last_response = response_data
+
+		return_data = response_data
+		if _preload_content:
+			# deserialize response data
+			if response_type:
+				return_data = self.deserialize(response_data, response_type)
+			else:
+				return_data = None
+
+		# Temporarily commented out to allow custom error handling
+		# if "errors" in return_data:
+		#	print(json.dumps(return_data))
+		#	exit(1)
+
+		if _return_http_data_only:
+			return (return_data)
+		else:
+			return (return_data, response_data.status,
+					response_data.getheaders())
+
+	def sanitize_for_serialization(self, obj):
+		"""Builds a JSON POST object.
+		If obj is None, return None.
+		If obj is str, int, long, float, bool, return directly.
+		If obj is datetime.datetime, datetime.date
+			convert to string in iso8601 format.
+		If obj is list, sanitize each element in the list.
+		If obj is dict, return the dict.
+		:param obj: The data to serialize.
+		:return: The serialized form of data.
+		"""
+		if obj is None:
+			return None
+		elif isinstance(obj, self.PRIMITIVE_TYPES):
+			return obj
+		elif isinstance(obj, list):
+			return [self.sanitize_for_serialization(sub_obj)
+					for sub_obj in obj]
+		elif isinstance(obj, tuple):
+			return tuple(self.sanitize_for_serialization(sub_obj)
+						 for sub_obj in obj)
+		elif isinstance(obj, (datetime.datetime, datetime.date)):
+			return obj.isoformat()
+
+		if isinstance(obj, dict):
+			obj_dict = obj
+		else:
+			obj_dict = {obj.attribute_map[attr]: getattr(obj, attr)
+						for attr, _ in six.iteritems(obj.swagger_types)
+						if getattr(obj, attr) is not None}
+
+		return {key: self.sanitize_for_serialization(val)
+				for key, val in six.iteritems(obj_dict)}
+
+	def deserialize(self, response, response_type):
+		# handle file downloading
+		# save response body into a tmp file and return the instance
+		if response_type == "file":
+			return self.__deserialize_file(response)
+
+		# fetch data from response object
+		try:
+			data = json.loads(response.data)
+		except ValueError:
+			data = response.data
+
+		return self.__deserialize(data, response_type)
+
+	def __deserialize(self, data, klass):
+		if data is None:
+			return None
+
+		if type(klass) == str:
+			if klass.startswith('list['):
+				sub_kls = re.match(r'list\[(.*)\]', klass).group(1)
+				return [self.__deserialize(sub_data, sub_kls)
+						for sub_data in data]
+
+			if klass.startswith('dict('):
+				sub_kls = re.match(r'dict\(([^,]*), (.*)\)', klass).group(2)
+				return {k: self.__deserialize(v, sub_kls)
+						for k, v in six.iteritems(data)}
+
+			# convert str to class
+			if klass in self.NATIVE_TYPES_MAPPING:
+				klass = self.NATIVE_TYPES_MAPPING[klass]
+			else:
+				klass = getattr(graphql_client.models, klass)
+
+		if klass in self.PRIMITIVE_TYPES:
+			return self.__deserialize_primitive(data, klass)
+		elif klass == object:
+			return self.__deserialize_object(data)
+		elif klass == datetime.date:
+			return self.__deserialize_date(data)
+		elif klass == datetime.datetime:
+			return self.__deserialize_datatime(data)
+		else:
+			return self.__deserialize_model(data, klass)
+
+	def call_api(self, header_params=None, body=None, files=None,
+				 response_type=None, async_req=None,
+				 _return_http_data_only=None, collection_formats=None,
+				 _preload_content=True, _request_timeout=None):
+
+		if not async_req:
+			return self.__call_api( header_params, body, files,
+								   response_type, _return_http_data_only, collection_formats,
+								   _preload_content, _request_timeout)
+		else:
+			thread = self.pool.apply_async(self.__call_api, ( header_params, 
+										   body, files, response_type, 
+										   _return_http_data_only,
+										   collection_formats,
+										   _preload_content, _request_timeout))
+		return thread
+
+	def request(self, url, headers=None, 
+			 body=None, _preload_content=True,
+			_request_timeout=None):
+
+		return self.rest_client.POST(url, headers=headers,
+									 _preload_content=_preload_content,
+									 _request_timeout=_request_timeout,
+									 body=body)
+
+	def parameters_to_tuples(self, params, collection_formats):
+		"""Get parameters as list of tuples, formatting collections.
+
+		:param params: Parameters as dict or list of two-tuples
+		:param dict collection_formats: Parameter collection formats
+		:return: Parameters as list of tuples, collections formatted
+		"""
+		new_params = []
+		if collection_formats is None:
+			collection_formats = {}
+		for k, v in six.iteritems(params) if isinstance(params, dict) else params:  # noqa: E501
+			if k in collection_formats:
+				collection_format = collection_formats[k]
+				if collection_format == 'multi':
+					new_params.extend((k, value) for value in v)
+				else:
+					if collection_format == 'ssv':
+						delimiter = ' '
+					elif collection_format == 'tsv':
+						delimiter = '\t'
+					elif collection_format == 'pipes':
+						delimiter = '|'
+					else:  # csv is the default
+						delimiter = ','
+					new_params.append(
+						(k, delimiter.join(str(value) for value in v)))
+			else:
+				new_params.append((k, v))
+		return new_params
+
+	def select_header_accept(self, accepts):
+		"""Returns `Accept` based on an array of accepts provided.
+
+		:param accepts: List of headers.
+		:return: Accept (e.g. application/json).
+		"""
+		if not accepts:
+			return
+
+		accepts = [x.lower() for x in accepts]
+
+		if 'application/json' in accepts:
+			return 'application/json'
+		else:
+			return ', '.join(accepts)
+
+	def select_header_content_type(self, content_types):
+		"""Returns `Content-Type` based on an array of content_types provided.
+
+		:param content_types: List of content-types.
+		:return: Content-Type (e.g. application/json).
+		"""
+		if not content_types:
+			return 'application/json'
+
+		content_types = [x.lower() for x in content_types]
+
+		if 'application/json' in content_types or '*/*' in content_types:
+			return 'application/json'
+		else:
+			return content_types[0]
+
+	def __deserialize_file(self, response):
+		"""Deserializes body to file
+
+		Saves response body into a file in a temporary folder,
+		using the filename from the `Content-Disposition` header if provided.
+
+		:param response:  RESTResponse.
+		:return: file path.
+		"""
+		fd, path = tempfile.mkstemp(dir=self.configuration.temp_folder_path)
+		os.close(fd)
+		os.remove(path)
+
+		content_disposition = response.getheader("Content-Disposition")
+		if content_disposition:
+			filename = re.search(r'filename=[\'"]?([^\'"\s]+)[\'"]?',
+								 content_disposition).group(1)
+			path = os.path.join(os.path.dirname(path), filename)
+			response_data = response.data
+			with open(path, "wb") as f:
+				if isinstance(response_data, str):
+					# change str to bytes so we can write it
+					response_data = response_data.encode('utf-8')
+					f.write(response_data)
+				else:
+					f.write(response_data)
+		return path
+
+	def __deserialize_primitive(self, data, klass):
+		"""Deserializes string to primitive type.
+
+		:param data: str.
+		:param klass: class literal.
+
+		:return: int, long, float, str, bool.
+		"""
+		try:
+			return klass(data)
+		except UnicodeEncodeError:
+			return six.text_type(data)
+		except TypeError:
+			return data
+
+	def __deserialize_object(self, value):
+		"""Return a original value.
+
+		:return: object.
+		"""
+		return value
+
+	def __deserialize_date(self, string):
+		"""Deserializes string to date.
+
+		:param string: str.
+		:return: date.
+		"""
+		try:
+			from dateutil.parser import parse
+			return parse(string).date()
+		except ImportError:
+			return string
+		except ValueError:
+			raise apiclient.ApiException(
+				status=0,
+				reason="Failed to parse `{0}` as date object".format(string)
+			)
+
+	def __deserialize_datatime(self, string):
+		"""Deserializes string to datetime.
+
+		The string should be in iso8601 datetime format.
+
+		:param string: str.
+		:return: datetime.
+		"""
+		try:
+			from dateutil.parser import parse
+			return parse(string)
+		except ImportError:
+			return string
+		except ValueError:
+			raise apiclient.ApiException(
+				status=0,
+				reason=(
+					"Failed to parse `{0}` as datetime object"
+					.format(string)
+				)
+			)
+
+	def __hasattr(self, object, name):
+			return name in object.__class__.__dict__
+
+	def __deserialize_model(self, data, klass):
+		"""Deserializes list or dict to model.
+
+		:param data: dict, list.
+		:param klass: class literal.
+		:return: model object.
+		"""
+
+		if not klass.swagger_types and not self.__hasattr(klass, 'get_real_child_model'):
+			return data
+
+		kwargs = {}
+		if klass.swagger_types is not None:
+			for attr, attr_type in six.iteritems(klass.swagger_types):
+				if (data is not None and
+						klass.attribute_map[attr] in data and
+						isinstance(data, (list, dict))):
+					value = data[klass.attribute_map[attr]]
+					kwargs[attr] = self.__deserialize(value, attr_type)
+
+		instance = klass(**kwargs)
+
+		if (isinstance(instance, dict) and
+				klass.swagger_types is not None and
+				isinstance(data, dict)):
+			for key, value in data.items():
+				if key not in klass.swagger_types:
+					instance[key] = value
+		if self.__hasattr(instance, 'get_real_child_model'):
+			klass_name = instance.get_real_child_model(data)
+			if klass_name:
+				instance = self.__deserialize(data, klass_name)
+		return instance