feat: improve error handling

Author: NguyenHoangSon96

Examples/handle_query_error.py

@@ -3,7 +3,7 @@
 """
 import logging
 from config import Config
-from influxdb_client_3.exceptions.exceptions import InfluxDB3ClientQueryError
+from influxdb_client_3.exceptions import InfluxDB3ClientQueryError
 
 import influxdb_client_3 as InfluxDBClient3
 

influxdb_client_3/__init__.py

@@ -6,11 +6,11 @@
 import pyarrow as pa
 from pyarrow import ArrowException
 
-from influxdb_client_3.exceptions.exceptions import InfluxDB3ClientQueryError
+from influxdb_client_3.exceptions import InfluxDB3ClientQueryError
 from influxdb_client_3.query.query_api import QueryApi as _QueryApi, QueryApiOptionsBuilder
 from influxdb_client_3.read_file import UploadFile
 from influxdb_client_3.write_client import InfluxDBClient as _InfluxDBClient, WriteOptions, Point
-from influxdb_client_3.exceptions.exceptions import InfluxDBError
+from influxdb_client_3.exceptions import InfluxDBError
 from influxdb_client_3.write_client.client.write_api import WriteApi as _WriteApi, SYNCHRONOUS, ASYNCHRONOUS, \
     PointSettings, DefaultWriteOptions, WriteType
 from influxdb_client_3.write_client.domain.write_precision import WritePrecision

influxdb_client_3/exceptions/__init__.py

@@ -0,0 +1,3 @@
+# flake8: noqa
+
+from .exceptions import InfluxDB3ClientQueryError, InfluxDBError, InfluxDB3ClientError

influxdb_client_3/exceptions/exceptions.py

@@ -4,7 +4,7 @@
 
 from urllib3 import HTTPResponse
 
-logger = logging.getLogger('influxdb_client_3.exceptions.exceptions')
+logger = logging.getLogger('influxdb_client_3.exceptions')
 
 
 class InfluxDB3ClientError(Exception):

influxdb_client_3/write_client/client/util/multiprocessing_helper.py

@@ -8,7 +8,7 @@
 import multiprocessing
 
 from influxdb_client_3.write_client import InfluxDBClient, WriteOptions
-from influxdb_client_3.exceptions.exceptions import InfluxDBError
+from influxdb_client_3.exceptions import InfluxDBError
 
 logger = logging.getLogger('influxdb_client.client.util.multiprocessing_helper')
 

influxdb_client_3/write_client/client/write/retry.py

@@ -9,7 +9,7 @@
 from urllib3 import Retry
 from urllib3.exceptions import MaxRetryError, ResponseError
 
-from influxdb_client_3.exceptions.exceptions import InfluxDBError
+from influxdb_client_3.exceptions import InfluxDBError
 
 logger = logging.getLogger('influxdb_client.client.write.retry')
 

influxdb_client_3/write_client/rest.py

@@ -5,7 +5,7 @@
 import logging
 from typing import Dict
 
-from influxdb_client_3.exceptions.exceptions import InfluxDBError
+from influxdb_client_3.exceptions import InfluxDBError
 from influxdb_client_3.write_client.configuration import Configuration
 
 _UTF_8_encoding = 'utf-8'

tests/test_api_client.py

@@ -6,7 +6,7 @@
 
 from influxdb_client_3.write_client._sync.api_client import ApiClient
 from influxdb_client_3.write_client.configuration import Configuration
-from influxdb_client_3.exceptions.exceptions import InfluxDBError
+from influxdb_client_3.exceptions import InfluxDBError
 from influxdb_client_3.write_client.service import WriteService
 from influxdb_client_3.version import VERSION
 

tests/test_influxdb_client_3.py

@@ -2,7 +2,7 @@
 from unittest.mock import patch
 
 from influxdb_client_3 import InfluxDBClient3, WritePrecision, DefaultWriteOptions, Point, WriteOptions, WriteType
-from influxdb_client_3.exceptions.exceptions import InfluxDB3ClientQueryError
+from influxdb_client_3.exceptions import InfluxDB3ClientQueryError
 from tests.util import asyncio_run
 from tests.util.mocks import ConstantFlightServer, ConstantData, ErrorFlightServer
 

tests/test_influxdb_client_3_integration.py

@@ -7,7 +7,7 @@
 
 import pyarrow
 import pytest
-from influxdb_client_3.exceptions.exceptions import InfluxDB3ClientQueryError, InfluxDBError
+from influxdb_client_3.exceptions import InfluxDB3ClientQueryError, InfluxDBError
 
 from influxdb_client_3 import InfluxDBClient3, write_client_options, WriteOptions
 from tests.util import asyncio_run, lp_to_py_object