apache-airflow-client 3.2.0__tar.gz → 3.2.1__tar.gz

{apache_airflow_client-3.2.0 → apache_airflow_client-3.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: apache-airflow-client
-Version: 3.2.0
+Version: 3.2.1
 Summary: Apache Airflow API (Stable)
 Project-URL: Bug Tracker, https://github.com/apache/airflow/issues
 Project-URL: Changelog, https://github.com/apache/airflow-client-python/blob/main/CHANGELOG.md

{apache_airflow_client-3.2.0 → apache_airflow_client-3.2.1}/airflow_client/client/__init__.py

@@ -14,7 +14,7 @@
 """  # noqa: E501
 
 
-__version__ = "3.2.0"
+__version__ = "3.2.1"
 
 # Define package exports
 __all__ = [
@@ -157,6 +157,7 @@ __all__ = [
     "JobResponse",
     "LastAssetEventResponse",
     "LocationInner",
+    "MaterializeAssetBody",
     "PatchTaskInstanceBody",
     "PluginCollectionResponse",
     "PluginImportErrorCollectionResponse",
@@ -352,6 +353,7 @@ from airflow_client.client.models.job_collection_response import JobCollectionRe
 from airflow_client.client.models.job_response import JobResponse as JobResponse
 from airflow_client.client.models.last_asset_event_response import LastAssetEventResponse as LastAssetEventResponse
 from airflow_client.client.models.location_inner import LocationInner as LocationInner
+from airflow_client.client.models.materialize_asset_body import MaterializeAssetBody as MaterializeAssetBody
 from airflow_client.client.models.patch_task_instance_body import PatchTaskInstanceBody as PatchTaskInstanceBody
 from airflow_client.client.models.plugin_collection_response import PluginCollectionResponse as PluginCollectionResponse
 from airflow_client.client.models.plugin_import_error_collection_response import PluginImportErrorCollectionResponse as PluginImportErrorCollectionResponse

{apache_airflow_client-3.2.0 → apache_airflow_client-3.2.1}/airflow_client/client/api/asset_api.py

@@ -26,6 +26,7 @@ from airflow_client.client.models.asset_event_response import AssetEventResponse
 from airflow_client.client.models.asset_response import AssetResponse
 from airflow_client.client.models.create_asset_events_body import CreateAssetEventsBody
 from airflow_client.client.models.dag_run_response import DAGRunResponse
+from airflow_client.client.models.materialize_asset_body import MaterializeAssetBody
 from airflow_client.client.models.queued_event_collection_response import QueuedEventCollectionResponse
 from airflow_client.client.models.queued_event_response import QueuedEventResponse
 
@@ -3884,6 +3885,7 @@ class AssetApi:
     def materialize_asset(
         self,
         asset_id: StrictInt,
+        materialize_asset_body: Optional[MaterializeAssetBody] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -3903,6 +3905,8 @@ class AssetApi:
 
         :param asset_id: (required)
         :type asset_id: int
+        :param materialize_asset_body:
+        :type materialize_asset_body: MaterializeAssetBody
         :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
@@ -3927,6 +3931,7 @@ class AssetApi:
 
         _param = self._materialize_asset_serialize(
             asset_id=asset_id,
+            materialize_asset_body=materialize_asset_body,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -3957,6 +3962,7 @@ class AssetApi:
     def materialize_asset_with_http_info(
         self,
         asset_id: StrictInt,
+        materialize_asset_body: Optional[MaterializeAssetBody] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -3976,6 +3982,8 @@ class AssetApi:
 
         :param asset_id: (required)
         :type asset_id: int
+        :param materialize_asset_body:
+        :type materialize_asset_body: MaterializeAssetBody
         :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
@@ -4000,6 +4008,7 @@ class AssetApi:
 
         _param = self._materialize_asset_serialize(
             asset_id=asset_id,
+            materialize_asset_body=materialize_asset_body,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -4030,6 +4039,7 @@ class AssetApi:
     def materialize_asset_without_preload_content(
         self,
         asset_id: StrictInt,
+        materialize_asset_body: Optional[MaterializeAssetBody] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -4049,6 +4059,8 @@ class AssetApi:
 
         :param asset_id: (required)
         :type asset_id: int
+        :param materialize_asset_body:
+        :type materialize_asset_body: MaterializeAssetBody
         :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
@@ -4073,6 +4085,7 @@ class AssetApi:
 
         _param = self._materialize_asset_serialize(
             asset_id=asset_id,
+            materialize_asset_body=materialize_asset_body,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -4098,6 +4111,7 @@ class AssetApi:
     def _materialize_asset_serialize(
         self,
         asset_id,
+        materialize_asset_body,
         _request_auth,
         _content_type,
         _headers,
@@ -4125,6 +4139,8 @@ class AssetApi:
         # process the header parameters
         # process the form parameters
         # process the body parameter
+        if materialize_asset_body is not None:
+            _body_params = materialize_asset_body
 
 
         # set the HTTP header `Accept`
@@ -4135,6 +4151,19 @@ class AssetApi:
                 ]
             )
 
+        # set the HTTP header `Content-Type`
+        if _content_type:
+            _header_params['Content-Type'] = _content_type
+        else:
+            _default_content_type = (
+                self.api_client.select_header_content_type(
+                    [
+                        'application/json'
+                    ]
+                )
+            )
+            if _default_content_type is not None:
+                _header_params['Content-Type'] = _default_content_type
 
         # authentication setting
         _auth_settings: List[str] = [

{apache_airflow_client-3.2.0 → apache_airflow_client-3.2.1}/airflow_client/client/api_client.py

@@ -91,7 +91,7 @@ class ApiClient:
             self.default_headers[header_name] = header_value
         self.cookie = cookie
         # Set default User-Agent.
-        self.user_agent = 'OpenAPI-Generator/3.2.0/python'
+        self.user_agent = 'OpenAPI-Generator/3.2.1/python'
         self.client_side_validation = configuration.client_side_validation
 
     def __enter__(self):

{apache_airflow_client-3.2.0 → apache_airflow_client-3.2.1}/airflow_client/client/configuration.py

@@ -541,7 +541,7 @@ class Configuration:
                "OS: {env}\n"\
                "Python Version: {pyversion}\n"\
                "Version of the API: 2\n"\
-               "SDK Package Version: 3.2.0".\
+               "SDK Package Version: 3.2.1".\
                format(env=sys.platform, pyversion=sys.version)
 
     def get_host_settings(self) -> List[HostSetting]:

{apache_airflow_client-3.2.0 → apache_airflow_client-3.2.1}/airflow_client/client/models/__init__.py

@@ -117,6 +117,7 @@ from airflow_client.client.models.job_collection_response import JobCollectionRe
 from airflow_client.client.models.job_response import JobResponse
 from airflow_client.client.models.last_asset_event_response import LastAssetEventResponse
 from airflow_client.client.models.location_inner import LocationInner
+from airflow_client.client.models.materialize_asset_body import MaterializeAssetBody
 from airflow_client.client.models.patch_task_instance_body import PatchTaskInstanceBody
 from airflow_client.client.models.plugin_collection_response import PluginCollectionResponse
 from airflow_client.client.models.plugin_import_error_collection_response import PluginImportErrorCollectionResponse

apache_airflow_client-3.2.1/airflow_client/client/models/materialize_asset_body.py

@@ -0,0 +1,116 @@
+# coding: utf-8
+
+"""
+    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.
+
+    The version of the OpenAPI document: 2
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from datetime import datetime
+from pydantic import BaseModel, ConfigDict, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from typing import Optional, Set
+from typing_extensions import Self
+from pydantic_core import to_jsonable_python
+
+class MaterializeAssetBody(BaseModel):
+    """
+    Materialize asset request.
+    """ # noqa: E501
+    conf: Optional[Dict[str, Any]] = None
+    dag_run_id: Optional[StrictStr] = None
+    data_interval_end: Optional[datetime] = None
+    data_interval_start: Optional[datetime] = None
+    logical_date: Optional[datetime] = None
+    note: Optional[StrictStr] = None
+    partition_key: Optional[StrictStr] = None
+    run_after: Optional[datetime] = None
+    additional_properties: Dict[str, Any] = {}
+    __properties: ClassVar[List[str]] = ["conf", "dag_run_id", "data_interval_end", "data_interval_start", "logical_date", "note", "partition_key", "run_after"]
+
+    model_config = ConfigDict(
+        validate_by_name=True,
+        validate_by_alias=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        return json.dumps(to_jsonable_python(self.to_dict()))
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of MaterializeAssetBody from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        * Fields in `self.additional_properties` are added to the output dict.
+        """
+        excluded_fields: Set[str] = set([
+            "additional_properties",
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # puts key-value pairs in additional_properties in the top level
+        if self.additional_properties is not None:
+            for _key, _value in self.additional_properties.items():
+                _dict[_key] = _value
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of MaterializeAssetBody from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "conf": obj.get("conf"),
+            "dag_run_id": obj.get("dag_run_id"),
+            "data_interval_end": obj.get("data_interval_end"),
+            "data_interval_start": obj.get("data_interval_start"),
+            "logical_date": obj.get("logical_date"),
+            "note": obj.get("note"),
+            "partition_key": obj.get("partition_key"),
+            "run_after": obj.get("run_after")
+        })
+        # store additional fields in additional_properties
+        for _key in obj.keys():
+            if _key not in cls.__properties:
+                _obj.additional_properties[_key] = obj.get(_key)
+
+        return _obj
+
+

{apache_airflow_client-3.2.0 → apache_airflow_client-3.2.1}/docs/AssetApi.md

@@ -1087,7 +1087,7 @@ Name | Type | Description  | Notes
 [[Back to top]](#) [[Back to API list]](../README.md#documentation-for-api-endpoints) [[Back to Model list]](../README.md#documentation-for-models) [[Back to README]](../README.md)
 
 # **materialize_asset**
-> DAGRunResponse materialize_asset(asset_id)
+> DAGRunResponse materialize_asset(asset_id, materialize_asset_body=materialize_asset_body)
 
 Materialize Asset
 
@@ -1101,6 +1101,7 @@ Materialize an asset by triggering a DAG run that produces it.
 ```python
 import airflow_client.client
 from airflow_client.client.models.dag_run_response import DAGRunResponse
+from airflow_client.client.models.materialize_asset_body import MaterializeAssetBody
 from airflow_client.client.rest import ApiException
 from pprint import pprint
 
@@ -1127,10 +1128,11 @@ with airflow_client.client.ApiClient(configuration) as api_client:
     # Create an instance of the API class
     api_instance = airflow_client.client.AssetApi(api_client)
     asset_id = 56 # int | 
+    materialize_asset_body = airflow_client.client.MaterializeAssetBody() # MaterializeAssetBody |  (optional)
 
     try:
         # Materialize Asset
-        api_response = api_instance.materialize_asset(asset_id)
+        api_response = api_instance.materialize_asset(asset_id, materialize_asset_body=materialize_asset_body)
         print("The response of AssetApi->materialize_asset:\n")
         pprint(api_response)
     except Exception as e:
@@ -1145,6 +1147,7 @@ with airflow_client.client.ApiClient(configuration) as api_client:
 Name | Type | Description  | Notes
 ------------- | ------------- | ------------- | -------------
  **asset_id** | **int**|  | 
+ **materialize_asset_body** | [**MaterializeAssetBody**](MaterializeAssetBody.md)|  | [optional] 
 
 ### Return type
 
@@ -1156,7 +1159,7 @@ Name | Type | Description  | Notes
 
 ### HTTP request headers
 
- - **Content-Type**: Not defined
+ - **Content-Type**: application/json
  - **Accept**: application/json
 
 ### HTTP response details

apache_airflow_client-3.2.1/docs/MaterializeAssetBody.md

@@ -0,0 +1,37 @@
+# MaterializeAssetBody
+
+Materialize asset request.
+
+## Properties
+
+Name | Type | Description | Notes
+------------ | ------------- | ------------- | -------------
+**conf** | **Dict[str, object]** |  | [optional] 
+**dag_run_id** | **str** |  | [optional] 
+**data_interval_end** | **datetime** |  | [optional] 
+**data_interval_start** | **datetime** |  | [optional] 
+**logical_date** | **datetime** |  | [optional] 
+**note** | **str** |  | [optional] 
+**partition_key** | **str** |  | [optional] 
+**run_after** | **datetime** |  | [optional] 
+
+## Example
+
+```python
+from airflow_client.client.models.materialize_asset_body import MaterializeAssetBody
+
+# TODO update the JSON string below
+json = "{}"
+# create an instance of MaterializeAssetBody from a JSON string
+materialize_asset_body_instance = MaterializeAssetBody.from_json(json)
+# print the JSON string representation of the object
+print(MaterializeAssetBody.to_json())
+
+# convert the object into a dict
+materialize_asset_body_dict = materialize_asset_body_instance.to_dict()
+# create an instance of MaterializeAssetBody from a dict
+materialize_asset_body_from_dict = MaterializeAssetBody.from_dict(materialize_asset_body_dict)
+```
+[[Back to Model list]](../README.md#documentation-for-models) [[Back to API list]](../README.md#documentation-for-api-endpoints) [[Back to README]](../README.md)
+
+

{apache_airflow_client-3.2.0 → apache_airflow_client-3.2.1}/pyproject.toml

@@ -18,7 +18,7 @@
 [build-system]
 requires = [
     "hatchling==1.29.0",
-    "packaging==26.0",
+    "packaging==26.1",
     "pathspec==1.0.4",
     "pluggy==1.6.0",
     "tomli==2.4.1; python_version < '3.11'",

apache_airflow_client-3.2.1/test/test_materialize_asset_body.py

@@ -0,0 +1,58 @@
+# coding: utf-8
+
+"""
+    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.
+
+    The version of the OpenAPI document: 2
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+import unittest
+
+from airflow_client.client.models.materialize_asset_body import MaterializeAssetBody
+
+class TestMaterializeAssetBody(unittest.TestCase):
+    """MaterializeAssetBody unit test stubs"""
+
+    def setUp(self):
+        pass
+
+    def tearDown(self):
+        pass
+
+    def make_instance(self, include_optional) -> MaterializeAssetBody:
+        """Test MaterializeAssetBody
+            include_optional is a boolean, when False only required
+            params are included, when True both required and
+            optional params are included """
+        # uncomment below to create an instance of `MaterializeAssetBody`
+        """
+        model = MaterializeAssetBody()
+        if include_optional:
+            return MaterializeAssetBody(
+                conf = { },
+                dag_run_id = '',
+                data_interval_end = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'),
+                data_interval_start = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'),
+                logical_date = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'),
+                note = '',
+                partition_key = '',
+                run_after = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f')
+            )
+        else:
+            return MaterializeAssetBody(
+        )
+        """
+
+    def testMaterializeAssetBody(self):
+        """Test MaterializeAssetBody"""
+        # inst_req_only = self.make_instance(include_optional=False)
+        # inst_req_and_optional = self.make_instance(include_optional=True)
+
+if __name__ == '__main__':
+    unittest.main()

{apache_airflow_client-3.2.0 → apache_airflow_client-3.2.1}/v2.yaml

@@ -2540,6 +2540,41 @@ components:
           type: string
       title: LastAssetEventResponse
       type: object
+    MaterializeAssetBody:
+      additionalProperties: false
+      description: Materialize asset request.
+      properties:
+        conf:
+          additionalProperties: true
+          nullable: true
+          type: object
+        dag_run_id:
+          nullable: true
+          type: string
+        data_interval_end:
+          format: date-time
+          nullable: true
+          type: string
+        data_interval_start:
+          format: date-time
+          nullable: true
+          type: string
+        logical_date:
+          format: date-time
+          nullable: true
+          type: string
+        note:
+          nullable: true
+          type: string
+        partition_key:
+          nullable: true
+          type: string
+        run_after:
+          format: date-time
+          nullable: true
+          type: string
+      title: MaterializeAssetBody
+      type: object
     PatchTaskInstanceBody:
       additionalProperties: false
       description: Request body for Clear Task Instances endpoint.
@@ -4525,6 +4560,12 @@ paths:
         schema:
           title: Asset Id
           type: integer
+      requestBody:
+        content:
+          application/json:
+            schema:
+              $ref: '#/components/schemas/MaterializeAssetBody'
+              nullable: true
       responses:
         '200':
           content:

apache_airflow_client-3.2.1/version.txt

@@ -0,0 +1 @@
+3.2.1

apache_airflow_client-3.2.0/version.txt

@@ -1 +0,0 @@
-3.2.0