Merge pull request #9 from hoaihuongbk/allow_http_protocol

Allow to use HTTP protocol from Trino connection
hoaihuongbk · Feb 9, 2025 · 6d6707e · 6d6707e
2 parents a6ce209 + 8079756
commit 6d6707e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "lakeops"
-version = "0.1.5"
+version = "0.1.6"
 description = "Data lake operations toolkit"
 readme = "README.md"
 requires-python = ">=3.10"

diff --git a/src/lakeops/core/engines/trino.py b/src/lakeops/core/engines/trino.py
@@ -14,6 +14,7 @@ class TrinoEngineConfig:
     catalog: str
     schema: str
     password: Optional[str] = None
+    protocol: str = "https"
 
     @property
     def connection(self):
@@ -30,7 +31,7 @@ def connection(self):
             f"trino://{self.user}@{self.host}:{self.port}/{self.catalog}/{self.schema}",
             connect_args={
                 "auth": auth,
-                "http_scheme": "https",
+                "http_scheme": self.protocol,
             },
         )