feat: improve error handling

Author: NguyenHoangSon96

influxdb_client_3/__init__.py

@@ -430,7 +430,7 @@ async def query_async(self, query: str, language: str = "sql", mode: str = "all"
             database = self._database
 
         try:
-             return await self._query_api.query_async(query=query,
+            return await self._query_api.query_async(query=query,
                                                      language=language,
                                                      mode=mode,
                                                      database=database,

influxdb_client_3/query/query_api.py

@@ -168,7 +168,6 @@ def query(self, query: str, language: str, mode: str, database: str, **kwargs):
 
         return self._translate_stream_reader(flight_reader, mode)
 
-
     async def query_async(self, query: str, language: str, mode: str, database: str, **kwargs):
         """Query data from InfluxDB asynchronously.
 
@@ -190,8 +189,8 @@ async def query_async(self, query: str, language: str, mode: str, database: str,
         _flight_reader = await loop.run_in_executor(None,
                                                     self._flight_client.do_get, ticket, options)
         return await loop.run_in_executor(None, self._translate_stream_reader,
-                                              _flight_reader,
-                                              mode)
+                                          _flight_reader,
+                                          mode)
 
     def _translate_stream_reader(self, reader: FlightStreamReader, mode: str):
         from influxdb_client_3 import polars as has_polars

tests/test_influxdb_client_3.py

@@ -158,5 +158,6 @@ async def test_async_query_with_arrow_error(self):
                 await c.query_async("SELECT * FROM my_data")
             self.assertIn("Error while executing query", str(err.exception))
 
+
 if __name__ == '__main__':
     unittest.main()

tests/util/mocks.py

@@ -158,6 +158,7 @@ def number_batches(table):
             buf = struct.pack('<i', idx)
             yield batch, buf
 
+
 class ErrorFlightServer(FlightServerBase):
     def do_get(self, context, ticket):
-        raise ArrowException
+        raise ArrowException