port-ocean 0.5.13__tar.gz → 0.5.14__tar.gz

{port_ocean-0.5.13 → port_ocean-0.5.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: port-ocean
-Version: 0.5.13
+Version: 0.5.14
 Summary: Port Ocean is a CLI tool for managing your Port projects.
 Home-page: https://app.getport.io
 Keywords: ocean,port-ocean,port
@@ -25,7 +25,7 @@ Requires-Dist: aiostream (>=0.5.2,<0.6.0)
 Requires-Dist: click (>=8.1.3,<9.0.0) ; extra == "cli"
 Requires-Dist: confluent-kafka (>=2.1.1,<3.0.0)
 Requires-Dist: cookiecutter (>=2.1.1,<3.0.0) ; extra == "cli"
-Requires-Dist: fastapi (>=0.100,<0.110)
+Requires-Dist: fastapi (>=0.100,<0.111)
 Requires-Dist: httpx (>=0.24.1,<0.28.0)
 Requires-Dist: jinja2-time (>=0.2.0,<0.3.0) ; extra == "cli"
 Requires-Dist: loguru (>=0.7.0,<0.8.0)
@@ -37,7 +37,7 @@ Requires-Dist: pyyaml (>=6.0,<7.0)
 Requires-Dist: rich (>=13.4.1,<14.0.0) ; extra == "cli"
 Requires-Dist: six (>=1.16.0,<2.0.0)
 Requires-Dist: tomli (>=2.0.1,<3.0.0)
-Requires-Dist: urllib3 (>=1.26.16,<2.0.0)
+Requires-Dist: urllib3 (>=1.26.16,<3.0.0)
 Requires-Dist: uvicorn (>=0.22,<0.30)
 Requires-Dist: werkzeug (>=2.3.4,<4.0.0)
 Project-URL: Repository, https://github.com/port-labs/Port-Ocean

{port_ocean-0.5.13 → port_ocean-0.5.14}/port_ocean/clients/port/mixins/entities.py

@@ -173,27 +173,33 @@ class EntityClientMixin:
         handle_status_code(response)
         return [Entity.parse_obj(result) for result in response.json()["entities"]]
 
-    async def does_integration_has_ownership_over_entity(
-        self, entity: Entity, user_agent_type: UserAgentType
-    ) -> bool:
-        logger.info(f"Validating ownership on entity {entity.identifier}")
-        found_entities: list[Entity] = await self.search_entities(
+    async def search_batch_entities(
+        self, user_agent_type: UserAgentType, entities_to_search: list[Entity]
+    ) -> list[Entity]:
+        search_rules = []
+        for entity in entities_to_search:
+            search_rules.append(
+                {
+                    "combinator": "and",
+                    "rules": [
+                        {
+                            "property": "$identifier",
+                            "operator": "=",
+                            "value": entity.identifier,
+                        },
+                        {
+                            "property": "$blueprint",
+                            "operator": "=",
+                            "value": entity.blueprint,
+                        },
+                    ],
+                }
+            )
+
+        return await self.search_entities(
             user_agent_type,
             {
                 "combinator": "and",
-                "rules": [
-                    {
-                        "property": "$identifier",
-                        "operator": "contains",
-                        "value": entity.identifier,
-                    },
-                    {
-                        "property": "$blueprint",
-                        "operator": "contains",
-                        "value": entity.blueprint,
-                    },
-                ],
+                "rules": [{"combinator": "or", "rules": search_rules}],
             },
         )
-
-        return len(found_entities) > 0

{port_ocean-0.5.13 → port_ocean-0.5.14}/port_ocean/core/handlers/entity_processor/base.py

@@ -7,7 +7,7 @@ from port_ocean.core.handlers.base import BaseHandler
 from port_ocean.core.handlers.port_app_config.models import ResourceConfig
 from port_ocean.core.models import Entity
 from port_ocean.core.ocean_types import (
-    RawEntity,
+    RAW_ITEM,
     EntitySelectorDiff,
 )
 
@@ -38,7 +38,7 @@ class BaseEntityProcessor(BaseHandler):
     async def _parse_items(
         self,
         mapping: ResourceConfig,
-        raw_data: list[RawEntity],
+        raw_data: list[RAW_ITEM],
         parse_all: bool = False,
     ) -> EntitySelectorDiff:
         pass
@@ -46,7 +46,7 @@ class BaseEntityProcessor(BaseHandler):
     async def parse_items(
         self,
         mapping: ResourceConfig,
-        raw_data: list[RawEntity],
+        raw_data: list[RAW_ITEM],
         parse_all: bool = False,
     ) -> EntitySelectorDiff:
         """Public method to parse raw entity data and map it to an EntityDiff.

{port_ocean-0.5.13 → port_ocean-0.5.14}/port_ocean/core/handlers/entity_processor/jq_entity_processor.py

@@ -10,7 +10,7 @@ from port_ocean.core.handlers.entity_processor.base import BaseEntityProcessor
 from port_ocean.core.handlers.port_app_config.models import ResourceConfig
 from port_ocean.core.models import Entity
 from port_ocean.core.ocean_types import (
-    RawEntity,
+    RAW_ITEM,
     EntitySelectorDiff,
 )
 from port_ocean.exceptions.core import EntityProcessorException
@@ -122,7 +122,7 @@ class JQEntityProcessor(BaseEntityProcessor):
     async def _parse_items(
         self,
         mapping: ResourceConfig,
-        raw_results: list[RawEntity],
+        raw_results: list[RAW_ITEM],
         parse_all: bool = False,
     ) -> EntitySelectorDiff:
         raw_entity_mappings: dict[str, Any] = mapping.port.entity.mappings.dict(
@@ -153,4 +153,4 @@ class JQEntityProcessor(BaseEntityProcessor):
                     else:
                         failed_entities.append(parsed_entity)
 
-        return {"passed": passed_entities, "failed": failed_entities}
+        return EntitySelectorDiff(passed=passed_entities, failed=failed_entities)

{port_ocean-0.5.13 → port_ocean-0.5.14}/port_ocean/core/integrations/mixins/sync_raw.py

@@ -22,9 +22,8 @@ from port_ocean.core.ocean_types import (
     RAW_RESULT,
     RESYNC_RESULT,
     RawEntityDiff,
-    EntityDiff,
     ASYNC_GENERATOR_RESYNC_TYPE,
-    RawEntity,
+    RAW_ITEM,
     EntitySelectorDiff,
 )
 from port_ocean.core.utils import zip_and_sum
@@ -123,7 +122,7 @@ class SyncRawMixin(HandlerMixin, EventsMixin):
 
     async def _calculate_raw(
         self,
-        raw_diff: list[tuple[ResourceConfig, list[RawEntity]]],
+        raw_diff: list[tuple[ResourceConfig, list[RAW_ITEM]]],
         parse_all: bool = False,
     ) -> list[EntitySelectorDiff]:
         return await asyncio.gather(
@@ -139,40 +138,23 @@ class SyncRawMixin(HandlerMixin, EventsMixin):
         results: list[dict[Any, Any]],
         user_agent_type: UserAgentType,
         parse_all: bool = False,
-    ) -> list[Entity]:
+    ) -> EntitySelectorDiff:
         objects_diff = await self._calculate_raw([(resource, results)], parse_all)
+        await self.entities_state_applier.upsert(
+            objects_diff[0].passed, user_agent_type
+        )
 
-        entities_after: list[Entity] = objects_diff[0]["passed"]
-        await self.entities_state_applier.upsert(entities_after, user_agent_type)
-
-        # If an entity didn't pass the JQ selector, we want to delete it if it exists in Port
-        for entity_to_delete in objects_diff[0]["failed"]:
-            is_owner = (
-                await ocean.port_client.does_integration_has_ownership_over_entity(
-                    entity_to_delete, user_agent_type
-                )
-            )
-            if not is_owner:
-                logger.info(
-                    f"Skipping deletion of entity {entity_to_delete.identifier}, "
-                    f"Couldn't find an entity that's related to the current integration."
-                )
-                continue
-            await self.entities_state_applier.delete(
-                objects_diff[0]["failed"], user_agent_type
-            )
-
-        return entities_after
+        return objects_diff[0]
 
     async def _unregister_resource_raw(
         self,
         resource: ResourceConfig,
-        results: list[RawEntity],
+        results: list[RAW_ITEM],
         user_agent_type: UserAgentType,
     ) -> list[Entity]:
         objects_diff = await self._calculate_raw([(resource, results)])
 
-        entities_after: list[Entity] = objects_diff[0]["passed"]
+        entities_after: list[Entity] = objects_diff[0].passed
         await self.entities_state_applier.delete(entities_after, user_agent_type)
         logger.info("Finished unregistering change")
         return entities_after
@@ -189,17 +171,21 @@ class SyncRawMixin(HandlerMixin, EventsMixin):
             else:
                 async_generators.append(result)
 
-        entities = await self._register_resource_raw(
-            resource_config, raw_results, user_agent_type
-        )
+        entities = (
+            await self._register_resource_raw(
+                resource_config, raw_results, user_agent_type
+            )
+        ).passed
 
         for generator in async_generators:
             try:
                 async for items in generator:
                     entities.extend(
-                        await self._register_resource_raw(
-                            resource_config, items, user_agent_type
-                        )
+                        (
+                            await self._register_resource_raw(
+                                resource_config, items, user_agent_type
+                            )
+                        ).passed
                     )
             except* OceanAbortException as error:
                 errors.append(error)
@@ -233,13 +219,44 @@ class SyncRawMixin(HandlerMixin, EventsMixin):
             resource for resource in config.resources if resource.kind == kind
         ]
 
-        return await asyncio.gather(
+        diffs: list[EntitySelectorDiff] = await asyncio.gather(
             *(
                 self._register_resource_raw(resource, results, user_agent_type, True)
                 for resource in resource_mappings
             )
         )
 
+        registered_entities, entities_to_delete = zip_and_sum(
+            (entities_diff.passed, entities_diff.failed) for entities_diff in diffs
+        )
+
+        registered_entities_attributes = {
+            (entity.identifier, entity.blueprint) for entity in registered_entities
+        }
+
+        filtered_entities_to_delete: list[Entity] = (
+            await ocean.port_client.search_batch_entities(
+                user_agent_type,
+                [
+                    entity
+                    for entity in entities_to_delete
+                    if (entity.identifier, entity.blueprint)
+                    not in registered_entities_attributes
+                ],
+            )
+        )
+
+        if filtered_entities_to_delete:
+            logger.info(
+                f"Deleting {len(filtered_entities_to_delete)} entities that didn't pass any of the selectors"
+            )
+
+            await self.entities_state_applier.delete(
+                filtered_entities_to_delete, user_agent_type
+            )
+
+        return registered_entities
+
     async def unregister_raw(
         self,
         kind: str,
@@ -306,16 +323,13 @@ class SyncRawMixin(HandlerMixin, EventsMixin):
                 [(mapping, raw_desired_state["after"]) for mapping in resource_mappings]
             )
 
-            entities_before_flatten = [
-                item
-                for sublist in [d["passed"] for d in entities_before]
-                for item in sublist
-            ]
-            entities_after_flatten = [
-                item
-                for sublist in [d["passed"] for d in entities_after]
-                for item in sublist
-            ]
+            entities_before_flatten: list[Entity] = sum(
+                (entities_diff.passed for entities_diff in entities_before), []
+            )
+
+            entities_after_flatten: list[Entity] = sum(
+                (entities_diff.passed for entities_diff in entities_after), []
+            )
 
             await self.entities_state_applier.apply_diff(
                 {"before": entities_before_flatten, "after": entities_after_flatten},

{port_ocean-0.5.13 → port_ocean-0.5.14}/port_ocean/core/ocean_types.py

@@ -1,14 +1,21 @@
-from typing import TypedDict, Any, AsyncIterator, Callable, Awaitable
+from typing import TypedDict, Any, AsyncIterator, Callable, Awaitable, NamedTuple
 
 from port_ocean.core.models import Entity
 
 
-RawEntity = dict[Any, Any]
+RAW_ITEM = dict[Any, Any]
+RAW_RESULT = list[RAW_ITEM]
+ASYNC_GENERATOR_RESYNC_TYPE = AsyncIterator[RAW_RESULT]
+RESYNC_RESULT = list[RAW_ITEM | ASYNC_GENERATOR_RESYNC_TYPE]
+
+LISTENER_RESULT = Awaitable[RAW_RESULT] | ASYNC_GENERATOR_RESYNC_TYPE
+RESYNC_EVENT_LISTENER = Callable[[str], LISTENER_RESULT]
+START_EVENT_LISTENER = Callable[[], Awaitable[None]]
 
 
 class RawEntityDiff(TypedDict):
-    before: list[RawEntity]
-    after: list[RawEntity]
+    before: list[RAW_ITEM]
+    after: list[RAW_ITEM]
 
 
 class EntityDiff(TypedDict):
@@ -16,21 +23,11 @@ class EntityDiff(TypedDict):
     after: list[Entity]
 
 
-class EntitySelectorDiff(TypedDict):
+class EntitySelectorDiff(NamedTuple):
     passed: list[Entity]
     failed: list[Entity]
 
 
-RAW_ITEM = dict[Any, Any]
-RAW_RESULT = list[RAW_ITEM]
-ASYNC_GENERATOR_RESYNC_TYPE = AsyncIterator[RAW_RESULT]
-RESYNC_RESULT = list[RAW_ITEM | ASYNC_GENERATOR_RESYNC_TYPE]
-
-LISTENER_RESULT = Awaitable[RAW_RESULT] | ASYNC_GENERATOR_RESYNC_TYPE
-RESYNC_EVENT_LISTENER = Callable[[str], LISTENER_RESULT]
-START_EVENT_LISTENER = Callable[[], Awaitable[None]]
-
-
 class IntegrationEventsCallbacks(TypedDict):
     start: list[START_EVENT_LISTENER]
     resync: dict[str | None, list[RESYNC_EVENT_LISTENER]]

{port_ocean-0.5.13 → port_ocean-0.5.14}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "port-ocean"
-version = "0.5.13"
+version = "0.5.14"
 description = "Port Ocean is a CLI tool for managing your Port projects."
 readme = "README.md"
 homepage = "https://app.getport.io"
@@ -40,12 +40,12 @@ pydantic = "^1.10.8"
 loguru = "^0.7.0"
 pyyaml = "^6.0"
 werkzeug = ">=2.3.4,<4.0.0"
-fastapi = ">=0.100,<0.110"
+fastapi = ">=0.100,<0.111"
 uvicorn = ">=0.22,<0.30"
 confluent-kafka = "^2.1.1"
 httpx = ">=0.24.1,<0.28.0"
 pyjq = "^2.6.0"
-urllib3 = "^1.26.16"
+urllib3 = ">=1.26.16,<3.0.0"
 six = "^1.16.0"
 pyhumps = "^3.8.0"
 pydispatcher = "^2.0.7"
@@ -66,7 +66,7 @@ cli = ["click", "rich", "cookiecutter", "jinja2-time"]
 pytest = ">=7.2,<9.0"
 black = ">=23.3,<25.0"
 mypy = "^1.3.0"
-pylint = "^2.17.4"
+pylint = ">=2.17.4,<4.0.0"
 types-pyyaml = "^6.0.12.10"
 ruff = "^0.0.278"
 types-toml = "^0.10.8.6"