feat: support env

Author: NguyenHoangSon96

influxdb_client_3/__init__.py

@@ -56,6 +56,7 @@ def file_parser_options(**kwargs):
 INFLUX_DATABASE = "INFLUX_DATABASE"
 INFLUX_ORG = "INFLUX_ORG"
 INFLUX_PRECISION = "INFLUX_PRECISION"
+INFLUX_AUTH_SCHEME = "INFLUX_AUTH_SCHEME"
 
 
 def from_env(**kwargs: Any) -> 'InfluxDBClient3':
@@ -93,6 +94,9 @@ def from_env(**kwargs: Any) -> 'InfluxDBClient3':
 
     org = os.getenv(INFLUX_ORG, "default")
 
+    if os.getenv(INFLUX_AUTH_SCHEME) is not None:
+        kwargs['auth_scheme'] = os.getenv(INFLUX_AUTH_SCHEME)
+    
     write_client_option = None
     if os.getenv(INFLUX_PRECISION) is not None:
         write_client_option = default_client_options(

tests/test_influxdb_client_3_integration.py

@@ -4,6 +4,7 @@
 import string
 import time
 import unittest
+from unittest.mock import patch
 
 import pyarrow
 import pytest
@@ -276,15 +277,21 @@ async def test_verify_query_async(self):
             assert lp_to_py_object(item) in result_list, f"original lp data \"{item}\" should be in result list"
 
     def test_from_env(self):
-        c = from_env()
-        with c:
+        with from_env() as client:
             id_test = time.time_ns()
-            c.write(f"integration_test_python,type=used value=123.0,id_test={id_test}i")
+            client.write(f"integration_test_python,type=used value=123.0,id_test={id_test}i")
 
             sql = 'SELECT * FROM integration_test_python where type=$type and id_test=$id_test'
-            data = c.query(sql, mode="pandas", query_parameters={'type': 'used', 'id_test': id_test})
+            data = client.query(sql, mode="pandas", query_parameters={'type': 'used', 'id_test': id_test})
 
             self.assertIsNotNone(data)
             self.assertEqual(1, len(data))
             self.assertEqual(id_test, data['id_test'][0])
             self.assertEqual(123.0, data['value'][0])
+
+    @patch.dict('os.environ', {'INFLUX_AUTH_SCHEME': 'invalid_schema'})
+    def test_from_env_invalid_auth_schema(self):
+        with from_env() as client:
+            with self.assertRaises(InfluxDBError) as err:
+                client.write("integration_test_python,type=used value=123.0")
+            self.assertEqual('unauthorized access', err.exception.message)