linting

Author: nazarfil

openhexa/graphql/__init__.py

@@ -1,5 +1,4 @@
 """GraphQL package."""
 
-
 from .base_openhexa_client import BaseOpenHexaClient  # noqa: F401 -> Expose base client class
 from .graphql_client import *  # noqa: F403 -> Expose autogenerated types

openhexa/sdk/client/__init__.py

@@ -1,4 +1,5 @@
 """OpenHexa Client."""
+
 from ..utils import OpenHexaClient
 
 openhexa = OpenHexaClient()

openhexa/sdk/pipelines/run.py

@@ -3,15 +3,13 @@
 import datetime
 import errno
 import os
-import uuid
 from typing import Any
 
+from openhexa.graphql import CurrentPipelineRunCurrentPipelineRunPipeline
+from openhexa.sdk.client import OpenHexaClient
 from openhexa.sdk.pipelines.log_level import LogLevel
 from openhexa.sdk.utils import Environment, get_environment, graphql
 from openhexa.sdk.workspaces import workspace
-from openhexa.sdk.client import OpenHexaClient
-
-from openhexa.graphql import CurrentPipelineRunCurrentPipelineRunPipeline
 
 
 class CurrentRun:
@@ -28,11 +26,24 @@ def _connected(self):
     @property
     def pipeline_run_id(self) -> Any:
         """Get the current pipeline run ID."""
-        return OpenHexaClient().current_pipeline_run()["id"]
+        if self._connected:
+            # Use the OpenHexaClient to get the current pipeline run ID
+            return OpenHexaClient().current_pipeline_run()["id"]
+        else:
+            raise RuntimeError(
+                "CurrentRun is not connected to the OpenHEXA server. "
+                "Make sure to set the HEXA_SERVER_URL environment variable."
+            )
 
     def get_pipline(self) -> CurrentPipelineRunCurrentPipelineRunPipeline:
         """Get the current pipeline."""
-        return OpenHexaClient().current_pipeline_run()["pipeline"]
+        if self._connected:
+            return OpenHexaClient().current_pipeline_run()["pipeline"]
+        else:
+            raise RuntimeError(
+                "CurrentRun is not connected to the OpenHEXA server. "
+                "Make sure to set the HEXA_SERVER_URL environment variable."
+            )
 
     def add_file_output(self, path: str):
         """Record a run output for a file creation operation.

tests/test_current_run.py

@@ -28,4 +28,3 @@ def test_filtering_log_messages_based_on_settings(mock_graphql, settings):
     assert mock_graphql.call_count == 2
     mock_graphql.assert_any_call(ANY, {"input": {"priority": "ERROR", "message": "This is an error message"}})
     mock_graphql.assert_any_call(ANY, {"input": {"priority": "CRITICAL", "message": "This is a critical message"}})
-