|
| 1 | +""" |
| 2 | + Airflow API (Stable) |
| 3 | +
|
| 4 | + Apache Airflow management API. # noqa: E501 |
| 5 | +
|
| 6 | + The version of the OpenAPI document: 1.0.0 |
| 7 | + |
| 8 | + Generated by: https://openapi-generator.tech |
| 9 | +""" |
| 10 | + |
| 11 | + |
| 12 | +import re # noqa: F401 |
| 13 | +import sys # noqa: F401 |
| 14 | + |
| 15 | +import nulltype # noqa: F401 |
| 16 | + |
| 17 | +from airflow_python_sdk.model_utils import ( # noqa: F401 |
| 18 | + ApiTypeError, |
| 19 | + ModelComposed, |
| 20 | + ModelNormal, |
| 21 | + ModelSimple, |
| 22 | + cached_property, |
| 23 | + change_keys_js_to_python, |
| 24 | + convert_js_args_to_python_args, |
| 25 | + date, |
| 26 | + datetime, |
| 27 | + file_type, |
| 28 | + none_type, |
| 29 | + validate_get_composed_info, |
| 30 | +) |
| 31 | + |
| 32 | + |
| 33 | +class UpdateTaskInstancesState(ModelNormal): |
| 34 | + """NOTE: This class is auto generated by OpenAPI Generator. |
| 35 | + Ref: https://openapi-generator.tech |
| 36 | +
|
| 37 | + Do not edit the class manually. |
| 38 | +
|
| 39 | + Attributes: |
| 40 | + allowed_values (dict): The key is the tuple path to the attribute |
| 41 | + and the for var_name this is (var_name,). The value is a dict |
| 42 | + with a capitalized key describing the allowed value and an allowed |
| 43 | + value. These dicts store the allowed enum values. |
| 44 | + attribute_map (dict): The key is attribute name |
| 45 | + and the value is json key in definition. |
| 46 | + discriminator_value_class_map (dict): A dict to go from the discriminator |
| 47 | + variable value to the discriminator class name. |
| 48 | + validations (dict): The key is the tuple path to the attribute |
| 49 | + and the for var_name this is (var_name,). The value is a dict |
| 50 | + that stores validations for max_length, min_length, max_items, |
| 51 | + min_items, exclusive_maximum, inclusive_maximum, exclusive_minimum, |
| 52 | + inclusive_minimum, and regex. |
| 53 | + additional_properties_type (tuple): A tuple of classes accepted |
| 54 | + as additional properties values. |
| 55 | + """ |
| 56 | + |
| 57 | + allowed_values = { |
| 58 | + ('new_state',): { |
| 59 | + 'SUCCESS': "success", |
| 60 | + 'FAILED': "failed", |
| 61 | + }, |
| 62 | + } |
| 63 | + |
| 64 | + validations = { |
| 65 | + } |
| 66 | + |
| 67 | + additional_properties_type = None |
| 68 | + |
| 69 | + _nullable = False |
| 70 | + |
| 71 | + @cached_property |
| 72 | + def openapi_types(): |
| 73 | + """ |
| 74 | + This must be a method because a model may have properties that are |
| 75 | + of type self, this must run after the class is loaded |
| 76 | +
|
| 77 | + Returns |
| 78 | + openapi_types (dict): The key is attribute name |
| 79 | + and the value is attribute type. |
| 80 | + """ |
| 81 | + return { |
| 82 | + 'dry_run': (bool,), # noqa: E501 |
| 83 | + 'execution_date': (str,), # noqa: E501 |
| 84 | + 'include_downstream': (bool,), # noqa: E501 |
| 85 | + 'include_future': (bool,), # noqa: E501 |
| 86 | + 'include_past': (bool,), # noqa: E501 |
| 87 | + 'include_upstream': (bool,), # noqa: E501 |
| 88 | + 'new_state': (str,), # noqa: E501 |
| 89 | + 'task_id': (str,), # noqa: E501 |
| 90 | + } |
| 91 | + |
| 92 | + @cached_property |
| 93 | + def discriminator(): |
| 94 | + return None |
| 95 | + |
| 96 | + |
| 97 | + attribute_map = { |
| 98 | + 'dry_run': 'dry_run', # noqa: E501 |
| 99 | + 'execution_date': 'execution_date', # noqa: E501 |
| 100 | + 'include_downstream': 'include_downstream', # noqa: E501 |
| 101 | + 'include_future': 'include_future', # noqa: E501 |
| 102 | + 'include_past': 'include_past', # noqa: E501 |
| 103 | + 'include_upstream': 'include_upstream', # noqa: E501 |
| 104 | + 'new_state': 'new_state', # noqa: E501 |
| 105 | + 'task_id': 'task_id', # noqa: E501 |
| 106 | + } |
| 107 | + |
| 108 | + _composed_schemas = {} |
| 109 | + |
| 110 | + required_properties = set([ |
| 111 | + '_data_store', |
| 112 | + '_check_type', |
| 113 | + '_spec_property_naming', |
| 114 | + '_path_to_item', |
| 115 | + '_configuration', |
| 116 | + '_visited_composed_classes', |
| 117 | + ]) |
| 118 | + |
| 119 | + @convert_js_args_to_python_args |
| 120 | + def __init__(self, *args, **kwargs): # noqa: E501 |
| 121 | + """UpdateTaskInstancesState - a model defined in OpenAPI |
| 122 | +
|
| 123 | + Keyword Args: |
| 124 | + _check_type (bool): if True, values for parameters in openapi_types |
| 125 | + will be type checked and a TypeError will be |
| 126 | + raised if the wrong type is input. |
| 127 | + Defaults to True |
| 128 | + _path_to_item (tuple/list): This is a list of keys or values to |
| 129 | + drill down to the model in received_data |
| 130 | + when deserializing a response |
| 131 | + _spec_property_naming (bool): True if the variable names in the input data |
| 132 | + are serialized names, as specified in the OpenAPI document. |
| 133 | + False if the variable names in the input data |
| 134 | + are pythonic names, e.g. snake case (default) |
| 135 | + _configuration (Configuration): the instance to use when |
| 136 | + deserializing a file_type parameter. |
| 137 | + If passed, type conversion is attempted |
| 138 | + If omitted no type conversion is done. |
| 139 | + _visited_composed_classes (tuple): This stores a tuple of |
| 140 | + classes that we have traveled through so that |
| 141 | + if we see that class again we will not use its |
| 142 | + discriminator again. |
| 143 | + When traveling through a discriminator, the |
| 144 | + composed schema that is |
| 145 | + is traveled through is added to this set. |
| 146 | + For example if Animal has a discriminator |
| 147 | + petType and we pass in "Dog", and the class Dog |
| 148 | + allOf includes Animal, we move through Animal |
| 149 | + once using the discriminator, and pick Dog. |
| 150 | + Then in Dog, we will make an instance of the |
| 151 | + Animal class but this time we won't travel |
| 152 | + through its discriminator because we passed in |
| 153 | + _visited_composed_classes = (Animal,) |
| 154 | + dry_run (bool): If set, don't actually run this operation. The response will contain a list of task instances planned to be affected, but won't be modified in any way. . [optional] if omitted the server will use the default value of True # noqa: E501 |
| 155 | + execution_date (str): The execution date.. [optional] # noqa: E501 |
| 156 | + include_downstream (bool): If set to true, downstream tasks are also affected.. [optional] # noqa: E501 |
| 157 | + include_future (bool): If set to True, also tasks from future DAG Runs are affected.. [optional] # noqa: E501 |
| 158 | + include_past (bool): If set to True, also tasks from past DAG Runs are affected.. [optional] # noqa: E501 |
| 159 | + include_upstream (bool): If set to true, upstream tasks are also affected.. [optional] # noqa: E501 |
| 160 | + new_state (str): Expected new state.. [optional] # noqa: E501 |
| 161 | + task_id (str): The task ID.. [optional] # noqa: E501 |
| 162 | + """ |
| 163 | + |
| 164 | + _check_type = kwargs.pop('_check_type', True) |
| 165 | + _spec_property_naming = kwargs.pop('_spec_property_naming', False) |
| 166 | + _path_to_item = kwargs.pop('_path_to_item', ()) |
| 167 | + _configuration = kwargs.pop('_configuration', None) |
| 168 | + _visited_composed_classes = kwargs.pop('_visited_composed_classes', ()) |
| 169 | + |
| 170 | + if args: |
| 171 | + raise ApiTypeError( |
| 172 | + "Invalid positional arguments=%s passed to %s. Remove those invalid positional arguments." % ( |
| 173 | + args, |
| 174 | + self.__class__.__name__, |
| 175 | + ), |
| 176 | + path_to_item=_path_to_item, |
| 177 | + valid_classes=(self.__class__,), |
| 178 | + ) |
| 179 | + |
| 180 | + self._data_store = {} |
| 181 | + self._check_type = _check_type |
| 182 | + self._spec_property_naming = _spec_property_naming |
| 183 | + self._path_to_item = _path_to_item |
| 184 | + self._configuration = _configuration |
| 185 | + self._visited_composed_classes = _visited_composed_classes + (self.__class__,) |
| 186 | + |
| 187 | + for var_name, var_value in kwargs.items(): |
| 188 | + if var_name not in self.attribute_map and \ |
| 189 | + self._configuration is not None and \ |
| 190 | + self._configuration.discard_unknown_keys and \ |
| 191 | + self.additional_properties_type is None: |
| 192 | + # discard variable. |
| 193 | + continue |
| 194 | + setattr(self, var_name, var_value) |
0 commit comments