Added necesary parameters to 'to_iceberg' method to expand functionality

Author: pedrorfdez

awswrangler/athena/_utils.py

@@ -43,6 +43,11 @@
 
 _logger: logging.Logger = logging.getLogger(__name__)
 
+class _MergeClause(TypedDict, total=False):
+    when: Literal["MATCHED", "NOT_MATCHED", "NOT_MATCHED_BY_SOURCE"]
+    condition: str | None
+    action: Literal["UPDATE", "DELETE", "INSERT"]
+    columns: list[str] | None
 
 class _QueryMetadata(NamedTuple):
     execution_id: str
@@ -63,7 +68,6 @@ class _WorkGroupConfig(NamedTuple):
     encryption: str | None
     kms_key: str | None
 
-
 def _get_s3_output(
     s3_output: str | None, wg_config: _WorkGroupConfig, boto3_session: boto3.Session | None = None
 ) -> str:

awswrangler/athena/_write_iceberg.py

@@ -6,7 +6,7 @@
 import re
 import typing
 import uuid
-from typing import Any, Dict, Literal, TypedDict
+from typing import Any, Dict, Literal, TypedDict, Union
 
 import boto3
 import pandas as pd
@@ -18,6 +18,7 @@
     _get_workgroup_config,
     _start_query_execution,
     _WorkGroupConfig,
+    _MergeClause
 )
 from awswrangler.typing import GlueTableSettings
 
@@ -361,7 +362,9 @@ def to_iceberg(  # noqa: PLR0913
     table_location: str | None = None,
     partition_cols: list[str] | None = None,
     merge_cols: list[str] | None = None,
-    merge_condition: Literal["update", "ignore"] = "update",
+    merge_on_condition: str | None = None,
+    merge_condition: Literal["update", "ignore", "conditional_merge"] = "update",
+    merge_conditional_clauses: list[_MergeClause] | None = None,
     merge_match_nulls: bool = False,
     keep_files: bool = True,
     data_source: str | None = None,