adding clearTaskInstances endpoint

Author: zachliu

README.md

@@ -4,7 +4,7 @@ Apache Airflow management API.
 This Python package is automatically generated by the [OpenAPI Generator](https://openapi-generator.tech) project based on [the specs](https://github.com/zachliu/airflow-openapi-specs):
 
 - API version: 1.0.0
-- Package version: 0.0.5
+- Package version: 0.1.5
 - Build package: org.openapitools.codegen.languages.PythonClientCodegen
 For more information, please visit [https://github.com/zachliu](https://github.com/zachliu)
 
@@ -98,6 +98,7 @@ Class | Method | HTTP request | Description
 *ConnectionApi* | [**get_connection**](docs/ConnectionApi.md#get_connection) | **GET** /connections/{connection_id} | Get a connection entry
 *ConnectionApi* | [**get_connections**](docs/ConnectionApi.md#get_connections) | **GET** /connections | Get all connection entries
 *ConnectionApi* | [**patch_connection**](docs/ConnectionApi.md#patch_connection) | **PATCH** /connections/{connection_id} | Update a connection entry
+*DAGApi* | [**clear_task_instances**](docs/DAGApi.md#clear_task_instances) | **POST** /dags/{dag_id}/clearTaskInstances | Clear a set of task instances
 *DAGApi* | [**get_dag**](docs/DAGApi.md#get_dag) | **GET** /dags/{dag_id} | Get basic information about a DAG
 *DAGApi* | [**get_dag_source**](docs/DAGApi.md#get_dag_source) | **GET** /dagSources/{file_token} | Get source code using file token
 *DAGApi* | [**get_dag_structure**](docs/DAGApi.md#get_dag_structure) | **GET** /dags/{dag_id}/structure | Get simplified representation of DAG.
@@ -144,6 +145,7 @@ Class | Method | HTTP request | Description
 ## Documentation For Models
 
  - [ClassReference](docs/ClassReference.md)
+ - [ClearTaskInstance](docs/ClearTaskInstance.md)
  - [CollectionInfo](docs/CollectionInfo.md)
  - [Color](docs/Color.md)
  - [Config](docs/Config.md)
@@ -183,6 +185,8 @@ Class | Method | HTTP request | Description
  - [TaskFail](docs/TaskFail.md)
  - [TaskInstance](docs/TaskInstance.md)
  - [TaskInstanceCollection](docs/TaskInstanceCollection.md)
+ - [TaskInstanceReference](docs/TaskInstanceReference.md)
+ - [TaskInstanceReferenceCollection](docs/TaskInstanceReferenceCollection.md)
  - [TaskState](docs/TaskState.md)
  - [TimeDelta](docs/TimeDelta.md)
  - [TriggerRule](docs/TriggerRule.md)

airflow_python_sdk/__init__.py

@@ -11,7 +11,7 @@
 """
 
 
-__version__ = "0.0.5"
+__version__ = "0.1.5"
 
 # import ApiClient
 from airflow_python_sdk.api_client import ApiClient

airflow_python_sdk/api/dag_api.py

@@ -22,13 +22,15 @@
     none_type,
     validate_and_convert_types
 )
+from airflow_python_sdk.model.clear_task_instance import ClearTaskInstance
 from airflow_python_sdk.model.dag import DAG
 from airflow_python_sdk.model.dag_collection import DAGCollection
 from airflow_python_sdk.model.dag_structure import DagStructure
 from airflow_python_sdk.model.error import Error
 from airflow_python_sdk.model.inline_response2001 import InlineResponse2001
 from airflow_python_sdk.model.task import Task
 from airflow_python_sdk.model.task_collection import TaskCollection
+from airflow_python_sdk.model.task_instance_reference_collection import TaskInstanceReferenceCollection
 
 
 class DAGApi(object):
@@ -43,6 +45,136 @@ def __init__(self, api_client=None):
             api_client = ApiClient()
         self.api_client = api_client
 
+        def __clear_task_instances(
+            self,
+            dag_id,
+            clear_task_instance,
+            **kwargs
+        ):
+            """Clear a set of task instances  # noqa: E501
+
+            Clears a set of task instances associated with the DAG for a specified date range.   # noqa: E501
+            This method makes a synchronous HTTP request by default. To make an
+            asynchronous HTTP request, please pass async_req=True
+
+            >>> thread = api.clear_task_instances(dag_id, clear_task_instance, async_req=True)
+            >>> result = thread.get()
+
+            Args:
+                dag_id (int): The DAG ID.
+                clear_task_instance (ClearTaskInstance): Parameters of action
+
+            Keyword Args:
+                _return_http_data_only (bool): response data without head status
+                    code and headers. Default is True.
+                _preload_content (bool): if False, the urllib3.HTTPResponse object
+                    will be returned without reading/decoding response data.
+                    Default is True.
+                _request_timeout (float/tuple): 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.
+                    Default is None.
+                _check_input_type (bool): specifies if type checking
+                    should be done one the data sent to the server.
+                    Default is True.
+                _check_return_type (bool): specifies if type checking
+                    should be done one the data received from the server.
+                    Default is True.
+                _host_index (int/None): specifies the index of the server
+                    that we want to use.
+                    Default is read from the configuration.
+                async_req (bool): execute request asynchronously
+
+            Returns:
+                TaskInstanceReferenceCollection
+                    If the method is called asynchronously, returns the request
+                    thread.
+            """
+            kwargs['async_req'] = kwargs.get(
+                'async_req', False
+            )
+            kwargs['_return_http_data_only'] = kwargs.get(
+                '_return_http_data_only', True
+            )
+            kwargs['_preload_content'] = kwargs.get(
+                '_preload_content', True
+            )
+            kwargs['_request_timeout'] = kwargs.get(
+                '_request_timeout', None
+            )
+            kwargs['_check_input_type'] = kwargs.get(
+                '_check_input_type', True
+            )
+            kwargs['_check_return_type'] = kwargs.get(
+                '_check_return_type', True
+            )
+            kwargs['_host_index'] = kwargs.get('_host_index')
+            kwargs['dag_id'] = \
+                dag_id
+            kwargs['clear_task_instance'] = \
+                clear_task_instance
+            return self.call_with_http_info(**kwargs)
+
+        self.clear_task_instances = Endpoint(
+            settings={
+                'response_type': (TaskInstanceReferenceCollection,),
+                'auth': [
+                    'basicAuth'
+                ],
+                'endpoint_path': '/dags/{dag_id}/clearTaskInstances',
+                'operation_id': 'clear_task_instances',
+                'http_method': 'POST',
+                'servers': None,
+            },
+            params_map={
+                'all': [
+                    'dag_id',
+                    'clear_task_instance',
+                ],
+                'required': [
+                    'dag_id',
+                    'clear_task_instance',
+                ],
+                'nullable': [
+                ],
+                'enum': [
+                ],
+                'validation': [
+                ]
+            },
+            root_map={
+                'validations': {
+                },
+                'allowed_values': {
+                },
+                'openapi_types': {
+                    'dag_id':
+                        (int,),
+                    'clear_task_instance':
+                        (ClearTaskInstance,),
+                },
+                'attribute_map': {
+                    'dag_id': 'dag_id',
+                },
+                'location_map': {
+                    'dag_id': 'path',
+                    'clear_task_instance': 'body',
+                },
+                'collection_format_map': {
+                }
+            },
+            headers_map={
+                'accept': [
+                    'application/json'
+                ],
+                'content_type': [
+                    'application/json'
+                ]
+            },
+            api_client=api_client,
+            callable=__clear_task_instances
+        )
+
         def __get_dag(
             self,
             dag_id,

airflow_python_sdk/api_client.py

@@ -77,7 +77,7 @@ def __init__(self, configuration=None, header_name=None, header_value=None,
             self.default_headers[header_name] = header_value
         self.cookie = cookie
         # Set default User-Agent.
-        self.user_agent = 'OpenAPI-Generator/0.0.5/python'
+        self.user_agent = 'OpenAPI-Generator/0.1.5/python'
 
     def __enter__(self):
         return self

airflow_python_sdk/configuration.py

@@ -400,7 +400,7 @@ def to_debug_report(self):
                "OS: {env}\n"\
                "Python Version: {pyversion}\n"\
                "Version of the API: 1.0.0\n"\
-               "SDK Package Version: 0.0.5".\
+               "SDK Package Version: 0.1.5".\
                format(env=sys.platform, pyversion=sys.version)
 
     def get_host_settings(self):