port-ocean 0.8.0__tar.gz → 0.9.0__tar.gz

{port_ocean-0.8.0 → port_ocean-0.9.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: port-ocean
-Version: 0.8.0
+Version: 0.9.0
 Summary: Port Ocean is a CLI tool for managing your Port projects.
 Home-page: https://app.getport.io
 Keywords: ocean,port-ocean,port

{port_ocean-0.8.0 → port_ocean-0.9.0}/port_ocean/cli/commands/sail.py

@@ -71,4 +71,10 @@ def sail(
         console.print("Setting event listener to Once")
         override["event_listener"] = {"type": "ONCE"}
 
-    run(path, log_level, port, initialize_port_resources, override)
+    run(
+        path,
+        log_level,
+        port,
+        initialize_port_resources=initialize_port_resources,
+        config_override=override,
+    )

{port_ocean-0.8.0 → port_ocean-0.9.0}/port_ocean/config/settings.py

@@ -2,9 +2,10 @@ from typing import Any, Literal
 
 from port_ocean.config.base import BaseOceanSettings, BaseOceanModel
 from port_ocean.core.event_listener import EventListenerSettingsType
-from port_ocean.utils.misc import get_integration_name
+from port_ocean.core.models import Runtime
+from port_ocean.utils.misc import get_integration_name, get_spec_file
 from pydantic import Extra, AnyHttpUrl, parse_obj_as
-from pydantic.class_validators import root_validator
+from pydantic.class_validators import root_validator, validator
 from pydantic.env_settings import InitSettingsSource, EnvSettingsSource, BaseSettings
 from pydantic.fields import Field
 from pydantic.main import BaseModel
@@ -69,3 +70,20 @@ class IntegrationConfiguration(BaseOceanSettings, extra=Extra.allow):
     event_listener: EventListenerSettingsType
     # If an identifier or type is not provided, it will be generated based on the integration name
     integration: IntegrationSettings = IntegrationSettings(type="", identifier="")
+    runtime: Runtime = "OnPrem"
+
+    @validator("runtime")
+    def validate_runtime(cls, runtime: Literal["OnPrem", "Saas"]) -> Runtime:
+        if runtime == "Saas":
+            spec = get_spec_file()
+            if spec is None:
+                raise ValueError(
+                    "Could not determine whether it's safe to run "
+                    "the integration due to not found spec.yaml."
+                )
+
+            saas_config = spec.get("saas")
+            if saas_config and not saas_config["enabled"]:
+                raise ValueError("This integration can't be ran as Saas")
+
+        return runtime

{port_ocean-0.8.0 → port_ocean-0.9.0}/port_ocean/core/handlers/entity_processor/jq_entity_processor.py

@@ -71,11 +71,15 @@ class JQEntityProcessor(BaseEntityProcessor):
     async def _search_as_object(
         self, data: dict[str, Any], obj: dict[str, Any]
     ) -> dict[str, Any | None]:
-        search_tasks: dict[str,Task[dict[str,Any | None]] | list[Task[dict[str,Any | None]]]] = {}
+        search_tasks: dict[
+            str, Task[dict[str, Any | None]] | list[Task[dict[str, Any | None]]]
+        ] = {}
         for key, value in obj.items():
-
-            if isinstance(value,list):
-                search_tasks[key] = [asyncio.create_task(self._search_as_object(data,obj)) for obj in value]
+            if isinstance(value, list):
+                search_tasks[key] = [
+                    asyncio.create_task(self._search_as_object(data, obj))
+                    for obj in value
+                ]
 
             elif isinstance(value, dict):
                 search_tasks[key] = asyncio.create_task(
@@ -87,7 +91,7 @@ class JQEntityProcessor(BaseEntityProcessor):
         result: dict[str, Any | None] = {}
         for key, task in search_tasks.items():
             try:
-                if isinstance(task,list):
+                if isinstance(task, list):
                     result[key] = [await task for task in task]
                 else:
                     result[key] = await task

{port_ocean-0.8.0 → port_ocean-0.9.0}/port_ocean/core/models.py

@@ -1,10 +1,13 @@
 from dataclasses import dataclass, field
-from typing import Any
+from typing import Any, Literal
 
 from pydantic import BaseModel
 from pydantic.fields import Field
 
 
+Runtime = Literal["OnPrem", "Saas"]
+
+
 class Entity(BaseModel):
     identifier: Any
     blueprint: Any

{port_ocean-0.8.0 → port_ocean-0.9.0}/port_ocean/core/utils.py

@@ -1,12 +1,17 @@
 import asyncio
 from typing import Iterable, Any, TypeVar, Callable, Awaitable
 
+from loguru import logger
 from pydantic import parse_obj_as, ValidationError
 
-from port_ocean.core.models import Entity
+from port_ocean.clients.port.client import PortClient
+from port_ocean.core.models import Entity, Runtime
 from port_ocean.core.models import EntityPortDiff
 from port_ocean.core.ocean_types import RAW_RESULT
-from port_ocean.exceptions.core import RawObjectValidationException
+from port_ocean.exceptions.core import (
+    RawObjectValidationException,
+    IntegrationRuntimeException,
+)
 
 T = TypeVar("T", bound=tuple[list[Any], ...])
 
@@ -29,6 +34,18 @@ def is_same_entity(first_entity: Entity, second_entity: Entity) -> bool:
     )
 
 
+async def validate_integration_runtime(
+    port_client: PortClient, requested_runtime: Runtime
+) -> None:
+    logger.debug("Validating integration runtime")
+    current_integration = await port_client.get_current_integration(should_raise=False)
+    current_runtime = current_integration.get("installationType", "OnPrem")
+    if current_integration and current_runtime != requested_runtime:
+        raise IntegrationRuntimeException(
+            f"Invalid Runtime! Requested to run existing {current_runtime} integration in {requested_runtime} runtime."
+        )
+
+
 Q = TypeVar("Q")
 
 

{port_ocean-0.8.0 → port_ocean-0.9.0}/port_ocean/exceptions/core.py

@@ -33,3 +33,7 @@ class IntegrationAlreadyStartedException(BaseOceanException):
 
 class IntegrationNotStartedException(BaseOceanException):
     pass
+
+
+class IntegrationRuntimeException(BaseOceanException):
+    pass

{port_ocean-0.8.0 → port_ocean-0.9.0}/port_ocean/run.py

@@ -1,3 +1,4 @@
+import asyncio
 from inspect import getmembers
 from typing import Dict, Any, Type
 
@@ -8,6 +9,7 @@ from port_ocean.bootstrap import create_default_app
 from port_ocean.config.dynamic import default_config_factory
 from port_ocean.config.settings import ApplicationSettings, LogLevelType
 from port_ocean.core.defaults.initialize import initialize_defaults
+from port_ocean.core.utils import validate_integration_runtime
 from port_ocean.log.logger_setup import setup_logger
 from port_ocean.ocean import Ocean
 from port_ocean.utils.misc import get_spec_file, load_module
@@ -45,6 +47,11 @@ def run(
         "app", default_app
     )
 
+    # Validate that the current integration's runtime matches the execution parameters
+    asyncio.get_event_loop().run_until_complete(
+        validate_integration_runtime(app.port_client, app.config.runtime)
+    )
+
     # Override config with arguments
     if initialize_port_resources is not None:
         app.config.initialize_port_resources = initialize_port_resources

{port_ocean-0.8.0 → port_ocean-0.9.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "port-ocean"
-version = "0.8.0"
+version = "0.9.0"
 description = "Port Ocean is a CLI tool for managing your Port projects."
 readme = "README.md"
 homepage = "https://app.getport.io"