InfluxCommunity
diff --git a/‎influxdb_client_3/__init__.py
Lines changed: 28 additions & 7 deletions b/‎influxdb_client_3/__init__.py
Lines changed: 28 additions & 7 deletions
@@ -41,6 +41,23 @@ def file_parser_options(**kwargs):
     return kwargs  
 
 
+def _deep_merge(target, source):
+    if isinstance(target, dict) and isinstance(source, dict):
+        for key, value in source.items():
+            if key in target and isinstance(target[key], (dict, list)) and isinstance(value, (dict, list)):
+                # If both target and source values are dictionaries or lists, merge them recursively
+                target[key] = _deep_merge(target[key], value)
+            else:
+                # Otherwise, replace the target value with the source value
+                target[key] = value
+    elif isinstance(target, list) and isinstance(source, list):
+        # If both target and source are lists, concatenate them
+        target.extend(source)
+    else:
+        # For other types, simply replace the target with the source
+        target = source
+    return target
+
 class InfluxDBClient3:
     def __init__(
             self,
@@ -172,7 +189,8 @@ def _process_dataframe(self, df, measurement_name, tag_columns, timestamp_column
                                   data_frame_measurement_name=measurement_name,
                                   data_frame_tag_columns=tag_columns,
                                   data_frame_timestamp_column=timestamp_column, **kwargs)
-
+  
+    
     def query(self, query, language="sql", mode="all", database=None,**kwargs ):
         """
         Query data from InfluxDB.
@@ -185,7 +203,7 @@ def query(self, query, language="sql", mode="all", database=None,**kwargs ):
         :type mode: str
         :param database: The database to query from. If not provided, uses the database provided during initialization.
         :type database: str
-        :param kwargs: Additional arguments for the query.
+        :param kwargs: FlightClientCallOptions for the query.
         :return: The queried data.
         """
 
@@ -194,10 +212,14 @@ def query(self, query, language="sql", mode="all", database=None,**kwargs ):
             database = self._database
 
         try:
-            headers = [(b"authorization", f"Bearer {self._token}".encode('utf-8'))]
-    
             # Create an authorization header
-            _options = FlightCallOptions(headers=headers, **kwargs)
+            args = {"headers": [(b"authorization", f"Bearer {self._token}".encode('utf-8'))]}
+            custom_args = {key: value for key, value in kwargs.items()}
+            # Merge defaults with user-provided arguments
+            _deep_merge(args, custom_args)
+            print(args["headers"])
+            _options = FlightCallOptions(headers=args["headers"])
+            print(_options)
             ticket_data = {"database": database, "sql_query": query, "query_type": language}
             ticket = Ticket(json.dumps(ticket_data).encode('utf-8'))
             flight_reader = self._flight_client.do_get(ticket, _options)
@@ -225,8 +247,7 @@ def __enter__(self):
 
     def __exit__(self, exc_type, exc_val, exc_tb):
         self.close()
-
-
+        
 __all__ = [
     "InfluxDBClient3",
     "Point",