feat(Connections): import and re-export connection classes in openhex…

…a.sdk (#93)
BLSQ · Dec 4, 2023 · 2e3db63 · 2e3db63
1 parent 549f73d
commit 2e3db63
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 3 deletions.
diff --git a/openhexa/sdk/__init__.py b/openhexa/sdk/__init__.py
@@ -1,4 +1,15 @@
 from .pipelines import current_run, parameter, pipeline
 from .workspaces import workspace
+from .workspaces.connection import DHIS2Connection, IASOConnection, PostgreSQLConnection, GCSConnection, S3Connection
 
-__all__ = ["workspace", "pipeline", "parameter", "current_run"]
+__all__ = [
+    "workspace",
+    "pipeline",
+    "parameter",
+    "current_run",
+    "DHIS2Connection",
+    "IASOConnection",
+    "PostgreSQLConnection",
+    "GCSConnection",
+    "S3Connection",
+]
diff --git a/tests/test_parameter.py b/tests/test_parameter.py
@@ -2,7 +2,7 @@
 import os
 import stringcase
 
-from openhexa.sdk.workspaces.connection import (
+from openhexa.sdk import (
     DHIS2Connection,
     IASOConnection,
     PostgreSQLConnection,

diff --git a/tests/test_pipeline.py b/tests/test_pipeline.py
@@ -4,7 +4,7 @@
 import stringcase
 import os
 
-from openhexa.sdk.workspaces.connection import (
+from openhexa.sdk import (
     DHIS2Connection,
     IASOConnection,
     PostgreSQLConnection,