fix: write_api TypeError in kwargs

PR #158 introduced a bug writing dataframes.
Filtering now the kwargs before forwarding them to lower levels

Author: Adrian Immer

.vscode/settings.json

@@ -0,0 +1,7 @@
+{
+    "python.testing.pytestEnabled": true,
+    "python.testing.unittestEnabled": false,
+    "python.testing.pytestArgs": [
+        "tests"
+    ]
+}

influxdb_client_3/write_client/client/write_api.py

@@ -27,6 +27,21 @@
 DEFAULT_WRITE_NO_SYNC = False
 DEFAULT_WRITE_TIMEOUT = 10_000
 
+# Kwargs consumed during serialization that should not be passed to _post_write
+SERIALIZER_KWARGS = {
+    # DataFrame-specific kwargs
+    'data_frame_measurement_name',
+    'data_frame_tag_columns',
+    'data_frame_timestamp_column',
+    'data_frame_timestamp_timezone',
+    # Record-specific kwargs (dict, NamedTuple, dataclass)
+    'record_measurement_key',
+    'record_measurement_name',
+    'record_time_key',
+    'record_tag_keys',
+    'record_field_keys',
+}
+
 logger = logging.getLogger('influxdb_client_3.write_client.client.write_api')
 
 if _HAS_DATACLASS:
@@ -397,9 +412,12 @@ def write(self, bucket: str, org: str = None,
 
         _async_req = True if self._write_options.write_type == WriteType.asynchronous else False
 
+        # Filter out serializer-specific kwargs before passing to _post_write
+        http_kwargs = {k: v for k, v in kwargs.items() if k not in SERIALIZER_KWARGS}
+
         def write_payload(payload):
             final_string = b'\n'.join(payload[1])
-            return self._post_write(_async_req, bucket, org, final_string, payload[0], no_sync, **kwargs)
+            return self._post_write(_async_req, bucket, org, final_string, payload[0], no_sync, **http_kwargs)
 
         results = list(map(write_payload, payloads.items()))
         if not _async_req:

tests/test_write_local_server.py

@@ -2,6 +2,7 @@
 import time
 from http import HTTPStatus
 
+import pandas as pd
 import pytest
 from pytest_httpserver import HTTPServer, RequestMatcher
 from urllib3.exceptions import TimeoutError as urllib3_TimeoutError
@@ -183,3 +184,32 @@ def test_write_with_timeout_arg(self, httpserver: HTTPServer):
                 ),
                 enable_gzip=True
             ).write(self.SAMPLE_RECORD, _request_timeout=1)
+
+    def test_write_dataframe_does_not_raise_type_error(self, httpserver: HTTPServer):
+        """
+        Regression test: writing a DataFrame should not raise TypeError.
+
+        Before the fix, serializer kwargs were passed to post_write(), causing a TypeError.
+        """
+        self.set_response_status(httpserver, 200)
+
+        df = pd.DataFrame({
+            'time': pd.to_datetime(['2024-01-01', '2024-01-02']),
+            'city': ['London', 'Paris'],
+            'temperature': [15.0, 18.0]
+        })
+
+        try:
+            InfluxDBClient3(
+                host=(httpserver.url_for("/")), org="ORG", database="DB", token="TOKEN",
+                write_client_options=write_client_options(
+                    write_options=WriteOptions(write_type=WriteType.synchronous)
+                )
+            ).write_dataframe(
+                df,
+                measurement='weather',
+                timestamp_column='time',
+                tags=['city']
+            )
+        except TypeError as e:
+            pytest.fail(f"write_dataframe raised TypeError: {e}")