airflow_client/client/api/dag_source_api.py - airflow-client-python - Git at Google

 """
     Airflow API

     Airflow API. All endpoints located under ``/api/v2`` can be used safely, are stable and backward compatible. Endpoints located under ``/ui`` are dedicated to the UI and are subject to breaking change depending on the need of the frontend. Users should not rely on those but use the public ones instead.  **Filtering with pattern parameters.** Many list endpoints accept ``*_pattern`` and ``*_prefix_pattern`` query parameters. Unless a parameter's own description says otherwise, ``*_pattern`` is a case-insensitive substring match (SQL ``ILIKE '%term%'``) where ``%`` matches any sequence and ``_`` matches any single character (e.g. ``%customer_%``) — convenient, but it cannot use B-tree indexes, so it is slow on large tables. ``*_prefix_pattern`` matches the start of the value, is case-sensitive and index-friendly (prefer it at scale); there ``%`` and ``_`` are literal and trailing non-alphanumeric characters are stripped so the range scan stays index-compatible under locale-aware collations (e.g. ``test_`` matches values starting with ``test``, and ``s3://`` matches ``s3``). In both, ``|`` means OR (e.g. ``dag1|dag2``) and ``~`` matches everything. Regular expressions are not supported by these parameters; regex-capable endpoints expose a separate parameter.

     The version of the OpenAPI document: 2
     Generated by OpenAPI Generator (https://openapi-generator.tech)

     Do not edit the class manually.
 """  # noqa: E501


 import warnings
 from pydantic import validate_call, Field, StrictFloat, StrictStr, StrictInt
 from typing import Any, Dict, List, Optional, Tuple, Union
 from typing_extensions import Annotated

 from pydantic import StrictInt, StrictStr, field_validator
 from typing import Optional
 from airflow_client.client.models.dag_source_response import DAGSourceResponse

 from airflow_client.client.api_client import ApiClient, RequestSerialized
 from airflow_client.client.api_response import ApiResponse
 from airflow_client.client.rest import RESTResponseType


 class DagSourceApi:
     """NOTE: This class is auto generated by OpenAPI Generator
     Ref: https://openapi-generator.tech

     Do not edit the class manually.
     """

     def __init__(self, api_client=None) -> None:
         if api_client is None:
             api_client = ApiClient.get_default()
         self.api_client = api_client


     @validate_call
     def get_dag_source(
         self,
         dag_id: StrictStr,
         version_number: Optional[StrictInt] = None,
         accept: Optional[StrictStr] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
             Tuple[
                 Annotated[StrictFloat, Field(gt=0)],
                 Annotated[StrictFloat, Field(gt=0)]
             ]
         ] = None,
         _request_auth: Optional[Dict[StrictStr, Any]] = None,
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
     ) -> DAGSourceResponse:
         """Get Dag Source

         Get source code using file token.

         :param dag_id: (required)
         :type dag_id: str
         :param version_number:
         :type version_number: int
         :param accept:
         :type accept: str
         :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.
         :type _request_timeout: int, tuple(int, int), optional
         :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
         :param _content_type: force content-type for the request.
         :type _content_type: str, Optional
         :param _headers: set to override the headers for a single
                          request; this effectively ignores the headers
                          in the spec for a single request.
         :type _headers: dict, optional
         :param _host_index: set to override the host_index for a single
                             request; this effectively ignores the host_index
                             in the spec for a single request.
         :type _host_index: int, optional
         :return: Returns the result object.
         """ # noqa: E501

         _param = self._get_dag_source_serialize(
             dag_id=dag_id,
             version_number=version_number,
             accept=accept,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
             _host_index=_host_index
         )

         _response_types_map: Dict[str, Optional[str]] = {
             '200': "DAGSourceResponse",
             '400': "HTTPExceptionResponse",
             '401': "HTTPExceptionResponse",
             '403': "HTTPExceptionResponse",
             '404': "HTTPExceptionResponse",
             '406': "HTTPExceptionResponse",
             '422': "HTTPValidationError",
         }
         response_data = self.api_client.call_api(
             *_param,
             _request_timeout=_request_timeout
         )
         response_data.read()
         return self.api_client.response_deserialize(
             response_data=response_data,
             response_types_map=_response_types_map,
         ).data


     @validate_call
     def get_dag_source_with_http_info(
         self,
         dag_id: StrictStr,
         version_number: Optional[StrictInt] = None,
         accept: Optional[StrictStr] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
             Tuple[
                 Annotated[StrictFloat, Field(gt=0)],
                 Annotated[StrictFloat, Field(gt=0)]
             ]
         ] = None,
         _request_auth: Optional[Dict[StrictStr, Any]] = None,
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
     ) -> ApiResponse[DAGSourceResponse]:
         """Get Dag Source

         Get source code using file token.

         :param dag_id: (required)
         :type dag_id: str
         :param version_number:
         :type version_number: int
         :param accept:
         :type accept: str
         :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.
         :type _request_timeout: int, tuple(int, int), optional
         :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
         :param _content_type: force content-type for the request.
         :type _content_type: str, Optional
         :param _headers: set to override the headers for a single
                          request; this effectively ignores the headers
                          in the spec for a single request.
         :type _headers: dict, optional
         :param _host_index: set to override the host_index for a single
                             request; this effectively ignores the host_index
                             in the spec for a single request.
         :type _host_index: int, optional
         :return: Returns the result object.
         """ # noqa: E501

         _param = self._get_dag_source_serialize(
             dag_id=dag_id,
             version_number=version_number,
             accept=accept,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
             _host_index=_host_index
         )

         _response_types_map: Dict[str, Optional[str]] = {
             '200': "DAGSourceResponse",
             '400': "HTTPExceptionResponse",
             '401': "HTTPExceptionResponse",
             '403': "HTTPExceptionResponse",
             '404': "HTTPExceptionResponse",
             '406': "HTTPExceptionResponse",
             '422': "HTTPValidationError",
         }
         response_data = self.api_client.call_api(
             *_param,
             _request_timeout=_request_timeout
         )
         response_data.read()
         return self.api_client.response_deserialize(
             response_data=response_data,
             response_types_map=_response_types_map,
         )


     @validate_call
     def get_dag_source_without_preload_content(
         self,
         dag_id: StrictStr,
         version_number: Optional[StrictInt] = None,
         accept: Optional[StrictStr] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
             Tuple[
                 Annotated[StrictFloat, Field(gt=0)],
                 Annotated[StrictFloat, Field(gt=0)]
             ]
         ] = None,
         _request_auth: Optional[Dict[StrictStr, Any]] = None,
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
     ) -> RESTResponseType:
         """Get Dag Source

         Get source code using file token.

         :param dag_id: (required)
         :type dag_id: str
         :param version_number:
         :type version_number: int
         :param accept:
         :type accept: str
         :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.
         :type _request_timeout: int, tuple(int, int), optional
         :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
         :param _content_type: force content-type for the request.
         :type _content_type: str, Optional
         :param _headers: set to override the headers for a single
                          request; this effectively ignores the headers
                          in the spec for a single request.
         :type _headers: dict, optional
         :param _host_index: set to override the host_index for a single
                             request; this effectively ignores the host_index
                             in the spec for a single request.
         :type _host_index: int, optional
         :return: Returns the result object.
         """ # noqa: E501

         _param = self._get_dag_source_serialize(
             dag_id=dag_id,
             version_number=version_number,
             accept=accept,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
             _host_index=_host_index
         )

         _response_types_map: Dict[str, Optional[str]] = {
             '200': "DAGSourceResponse",
             '400': "HTTPExceptionResponse",
             '401': "HTTPExceptionResponse",
             '403': "HTTPExceptionResponse",
             '404': "HTTPExceptionResponse",
             '406': "HTTPExceptionResponse",
             '422': "HTTPValidationError",
         }
         response_data = self.api_client.call_api(
             *_param,
             _request_timeout=_request_timeout
         )
         return response_data.response


     def _get_dag_source_serialize(
         self,
         dag_id,
         version_number,
         accept,
         _request_auth,
         _content_type,
         _headers,
         _host_index,
     ) -> RequestSerialized:

         _host = None

         _collection_formats: Dict[str, str] = {
         }

         _path_params: Dict[str, str] = {}
         _query_params: List[Tuple[str, str]] = []
         _header_params: Dict[str, Optional[str]] = _headers or {}
         _form_params: List[Tuple[str, str]] = []
         _files: Dict[
             str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
         ] = {}
         _body_params: Optional[bytes] = None

         # process the path parameters
         if dag_id is not None:
             _path_params['dag_id'] = dag_id
         # process the query parameters
         if version_number is not None:

             _query_params.append(('version_number', version_number))

         # process the header parameters
         if accept is not None:
             _header_params['accept'] = accept
         # process the form parameters
         # process the body parameter


         # set the HTTP header `Accept`
         if 'Accept' not in _header_params:
             _header_params['Accept'] = self.api_client.select_header_accept(
                 [
                     'application/json',
                     'text/plain'
                 ]
             )


         # authentication setting
         _auth_settings: List[str] = [
             'OAuth2PasswordBearer',
             'HTTPBearer'
         ]

         return self.api_client.param_serialize(
             method='GET',
             resource_path='/api/v2/dagSources/{dag_id}',
             path_params=_path_params,
             query_params=_query_params,
             header_params=_header_params,
             body=_body_params,
             post_params=_form_params,
             files=_files,
             auth_settings=_auth_settings,
             collection_formats=_collection_formats,
             _host=_host,
             _request_auth=_request_auth
         )
	"""
	Airflow API

	Airflow API. All endpoints located under ``/api/v2`` can be used safely, are stable and backward compatible. Endpoints located under ``/ui`` are dedicated to the UI and are subject to breaking change depending on the need of the frontend. Users should not rely on those but use the public ones instead. Filtering with pattern parameters. Many list endpoints accept ``_pattern`` and ``_prefix_pattern`` query parameters. Unless a parameter's own description says otherwise, ``_pattern`` is a case-insensitive substring match (SQL ``ILIKE '%term%'``) where ``%`` matches any sequence and ``_`` matches any single character (e.g. ``%customer_%``) — convenient, but it cannot use B-tree indexes, so it is slow on large tables. ``_prefix_pattern`` matches the start of the value, is case-sensitive and index-friendly (prefer it at scale); there ``%`` and ``_`` are literal and trailing non-alphanumeric characters are stripped so the range scan stays index-compatible under locale-aware collations (e.g. ``test_`` matches values starting with ``test``, and ``s3://`` matches ``s3``). In both, ``\|`` means OR (e.g. ``dag1\|dag2``) and ``~`` matches everything. Regular expressions are not supported by these parameters; regex-capable endpoints expose a separate parameter.

	The version of the OpenAPI document: 2
	Generated by OpenAPI Generator (https://openapi-generator.tech)

	Do not edit the class manually.
	""" # noqa: E501


	import warnings
	from pydantic import validate_call, Field, StrictFloat, StrictStr, StrictInt
	from typing import Any, Dict, List, Optional, Tuple, Union
	from typing_extensions import Annotated

	from pydantic import StrictInt, StrictStr, field_validator
	from typing import Optional
	from airflow_client.client.models.dag_source_response import DAGSourceResponse

	from airflow_client.client.api_client import ApiClient, RequestSerialized
	from airflow_client.client.api_response import ApiResponse
	from airflow_client.client.rest import RESTResponseType


	class DagSourceApi:
	"""NOTE: This class is auto generated by OpenAPI Generator
	Ref: https://openapi-generator.tech

	Do not edit the class manually.
	"""

	def __init__(self, api_client=None) -> None:
	if api_client is None:
	api_client = ApiClient.get_default()
	self.api_client = api_client


	@validate_call
	def get_dag_source(
	self,
	dag_id: StrictStr,
	version_number: Optional[StrictInt] = None,
	accept: Optional[StrictStr] = None,
	_request_timeout: Union[
	None,
	Annotated[StrictFloat, Field(gt=0)],
	Tuple[
	Annotated[StrictFloat, Field(gt=0)],
	Annotated[StrictFloat, Field(gt=0)]
	]
	] = None,
	_request_auth: Optional[Dict[StrictStr, Any]] = None,
	_content_type: Optional[StrictStr] = None,
	_headers: Optional[Dict[StrictStr, Any]] = None,
	_host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
	) -> DAGSourceResponse:
	"""Get Dag Source

	Get source code using file token.

	:param dag_id: (required)
	:type dag_id: str
	:param version_number:
	:type version_number: int
	:param accept:
	:type accept: str
	: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.
	:type _request_timeout: int, tuple(int, int), optional
	: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
	:param _content_type: force content-type for the request.
	:type _content_type: str, Optional
	:param _headers: set to override the headers for a single
	request; this effectively ignores the headers
	in the spec for a single request.
	:type _headers: dict, optional
	:param _host_index: set to override the host_index for a single
	request; this effectively ignores the host_index
	in the spec for a single request.
	:type _host_index: int, optional
	:return: Returns the result object.
	""" # noqa: E501

	_param = self._get_dag_source_serialize(
	dag_id=dag_id,
	version_number=version_number,
	accept=accept,
	_request_auth=_request_auth,
	_content_type=_content_type,
	_headers=_headers,
	_host_index=_host_index
	)

	_response_types_map: Dict[str, Optional[str]] = {
	'200': "DAGSourceResponse",
	'400': "HTTPExceptionResponse",
	'401': "HTTPExceptionResponse",
	'403': "HTTPExceptionResponse",
	'404': "HTTPExceptionResponse",
	'406': "HTTPExceptionResponse",
	'422': "HTTPValidationError",
	}
	response_data = self.api_client.call_api(
	*_param,
	_request_timeout=_request_timeout
	)
	response_data.read()
	return self.api_client.response_deserialize(
	response_data=response_data,
	response_types_map=_response_types_map,
	).data


	@validate_call
	def get_dag_source_with_http_info(
	self,
	dag_id: StrictStr,
	version_number: Optional[StrictInt] = None,
	accept: Optional[StrictStr] = None,
	_request_timeout: Union[
	None,
	Annotated[StrictFloat, Field(gt=0)],
	Tuple[
	Annotated[StrictFloat, Field(gt=0)],
	Annotated[StrictFloat, Field(gt=0)]
	]
	] = None,
	_request_auth: Optional[Dict[StrictStr, Any]] = None,
	_content_type: Optional[StrictStr] = None,
	_headers: Optional[Dict[StrictStr, Any]] = None,
	_host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
	) -> ApiResponse[DAGSourceResponse]:
	"""Get Dag Source

	Get source code using file token.

	:param dag_id: (required)
	:type dag_id: str
	:param version_number:
	:type version_number: int
	:param accept:
	:type accept: str
	: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.
	:type _request_timeout: int, tuple(int, int), optional
	: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
	:param _content_type: force content-type for the request.
	:type _content_type: str, Optional
	:param _headers: set to override the headers for a single
	request; this effectively ignores the headers
	in the spec for a single request.
	:type _headers: dict, optional
	:param _host_index: set to override the host_index for a single
	request; this effectively ignores the host_index
	in the spec for a single request.
	:type _host_index: int, optional
	:return: Returns the result object.
	""" # noqa: E501

	_param = self._get_dag_source_serialize(
	dag_id=dag_id,
	version_number=version_number,
	accept=accept,
	_request_auth=_request_auth,
	_content_type=_content_type,
	_headers=_headers,
	_host_index=_host_index
	)

	_response_types_map: Dict[str, Optional[str]] = {
	'200': "DAGSourceResponse",
	'400': "HTTPExceptionResponse",
	'401': "HTTPExceptionResponse",
	'403': "HTTPExceptionResponse",
	'404': "HTTPExceptionResponse",
	'406': "HTTPExceptionResponse",
	'422': "HTTPValidationError",
	}
	response_data = self.api_client.call_api(
	*_param,
	_request_timeout=_request_timeout
	)
	response_data.read()
	return self.api_client.response_deserialize(
	response_data=response_data,
	response_types_map=_response_types_map,
	)


	@validate_call
	def get_dag_source_without_preload_content(
	self,
	dag_id: StrictStr,
	version_number: Optional[StrictInt] = None,
	accept: Optional[StrictStr] = None,
	_request_timeout: Union[
	None,
	Annotated[StrictFloat, Field(gt=0)],
	Tuple[
	Annotated[StrictFloat, Field(gt=0)],
	Annotated[StrictFloat, Field(gt=0)]
	]
	] = None,
	_request_auth: Optional[Dict[StrictStr, Any]] = None,
	_content_type: Optional[StrictStr] = None,
	_headers: Optional[Dict[StrictStr, Any]] = None,
	_host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
	) -> RESTResponseType:
	"""Get Dag Source

	Get source code using file token.

	:param dag_id: (required)
	:type dag_id: str
	:param version_number:
	:type version_number: int
	:param accept:
	:type accept: str
	: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.
	:type _request_timeout: int, tuple(int, int), optional
	: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
	:param _content_type: force content-type for the request.
	:type _content_type: str, Optional
	:param _headers: set to override the headers for a single
	request; this effectively ignores the headers
	in the spec for a single request.
	:type _headers: dict, optional
	:param _host_index: set to override the host_index for a single
	request; this effectively ignores the host_index
	in the spec for a single request.
	:type _host_index: int, optional
	:return: Returns the result object.
	""" # noqa: E501

	_param = self._get_dag_source_serialize(
	dag_id=dag_id,
	version_number=version_number,
	accept=accept,
	_request_auth=_request_auth,
	_content_type=_content_type,
	_headers=_headers,
	_host_index=_host_index
	)

	_response_types_map: Dict[str, Optional[str]] = {
	'200': "DAGSourceResponse",
	'400': "HTTPExceptionResponse",
	'401': "HTTPExceptionResponse",
	'403': "HTTPExceptionResponse",
	'404': "HTTPExceptionResponse",
	'406': "HTTPExceptionResponse",
	'422': "HTTPValidationError",
	}
	response_data = self.api_client.call_api(
	*_param,
	_request_timeout=_request_timeout
	)
	return response_data.response


	def _get_dag_source_serialize(
	self,
	dag_id,
	version_number,
	accept,
	_request_auth,
	_content_type,
	_headers,
	_host_index,
	) -> RequestSerialized:

	_host = None

	_collection_formats: Dict[str, str] = {
	}

	_path_params: Dict[str, str] = {}
	_query_params: List[Tuple[str, str]] = []
	_header_params: Dict[str, Optional[str]] = _headers or {}
	_form_params: List[Tuple[str, str]] = []
	_files: Dict[
	str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
	] = {}
	_body_params: Optional[bytes] = None

	# process the path parameters
	if dag_id is not None:
	_path_params['dag_id'] = dag_id
	# process the query parameters
	if version_number is not None:

	_query_params.append(('version_number', version_number))

	# process the header parameters
	if accept is not None:
	_header_params['accept'] = accept
	# process the form parameters
	# process the body parameter


	# set the HTTP header `Accept`
	if 'Accept' not in _header_params:
	_header_params['Accept'] = self.api_client.select_header_accept(
	[
	'application/json',
	'text/plain'
	]
	)


	# authentication setting
	_auth_settings: List[str] = [
	'OAuth2PasswordBearer',
	'HTTPBearer'
	]

	return self.api_client.param_serialize(
	method='GET',
	resource_path='/api/v2/dagSources/{dag_id}',
	path_params=_path_params,
	query_params=_query_params,
	header_params=_header_params,
	body=_body_params,
	post_params=_form_params,
	files=_files,
	auth_settings=_auth_settings,
	collection_formats=_collection_formats,
	_host=_host,
	_request_auth=_request_auth
	)