port-ocean 0.14.0__py3-none-any.whl → 0.14.1__py3-none-any.whl

port_ocean/tests/utils/test_cache.py

@@ -0,0 +1,189 @@
+from typing import Any
+import asyncio
+from port_ocean.utils import cache  # Import the module where 'event' is used
+import pytest
+from dataclasses import dataclass, field
+from typing import AsyncGenerator, AsyncIterator, List, TypeVar
+
+
+@dataclass
+class EventContext:
+    attributes: dict[str, Any] = field(default_factory=dict)
+
+
+@pytest.fixture
+def event() -> EventContext:
+    return EventContext()
+
+
+T = TypeVar("T")
+
+
+async def collect_iterator_results(iterator: AsyncIterator[List[T]]) -> List[T]:
+    results = []
+    async for item in iterator:
+        results.extend(item)
+    return results
+
+
+@pytest.mark.asyncio
+async def test_cache_iterator_result(event: EventContext, monkeypatch: Any) -> None:
+    monkeypatch.setattr(cache, "event", event)
+
+    call_count = 0
+
+    @cache.cache_iterator_result()
+    async def sample_iterator(x: int) -> AsyncGenerator[List[int], None]:
+        nonlocal call_count
+        call_count += 1
+        for i in range(x):
+            await asyncio.sleep(0.1)
+            yield [i]
+
+    result1 = await collect_iterator_results(sample_iterator(3))
+    assert result1 == [0, 1, 2]
+    assert call_count == 1
+
+    result2 = await collect_iterator_results(sample_iterator(3))
+    assert result2 == [0, 1, 2]
+    assert call_count == 1
+
+    result3 = await collect_iterator_results(sample_iterator(4))
+    assert result3 == [0, 1, 2, 3]
+    assert call_count == 2
+
+
+@pytest.mark.asyncio
+async def test_cache_iterator_result_with_kwargs(
+    event: EventContext, monkeypatch: Any
+) -> None:
+    monkeypatch.setattr(cache, "event", event)
+
+    call_count = 0
+
+    @cache.cache_iterator_result()
+    async def sample_iterator(x: int, y: int = 1) -> AsyncGenerator[List[int], None]:
+        nonlocal call_count
+        call_count += 1
+        for i in range(x * y):
+            await asyncio.sleep(0.1)
+            yield [i]
+
+    result1 = await collect_iterator_results(sample_iterator(2, y=2))
+    assert result1 == [0, 1, 2, 3]
+    assert call_count == 1
+
+    result2 = await collect_iterator_results(sample_iterator(2, y=2))
+    assert result2 == [0, 1, 2, 3]
+    assert call_count == 1
+
+    result3 = await collect_iterator_results(sample_iterator(2, y=3))
+    assert result3 == [0, 1, 2, 3, 4, 5]
+    assert call_count == 2
+
+
+@pytest.mark.asyncio
+async def test_cache_iterator_result_no_cache(
+    event: EventContext, monkeypatch: Any
+) -> None:
+    monkeypatch.setattr(cache, "event", event)
+
+    call_count = 0
+
+    @cache.cache_iterator_result()
+    async def sample_iterator(x: int) -> AsyncGenerator[List[int], None]:
+        nonlocal call_count
+        call_count += 1
+        for i in range(x):
+            await asyncio.sleep(0.1)
+            yield [i]
+
+    result1 = await collect_iterator_results(sample_iterator(3))
+    assert result1 == [0, 1, 2]
+    assert call_count == 1
+
+    event.attributes.clear()
+
+    result2 = await collect_iterator_results(sample_iterator(3))
+    assert result2 == [0, 1, 2]
+    assert call_count == 2
+
+
+@pytest.mark.asyncio
+async def test_cache_coroutine_result(event: EventContext, monkeypatch: Any) -> None:
+    monkeypatch.setattr(cache, "event", event)
+
+    call_count = 0
+
+    @cache.cache_coroutine_result()
+    async def sample_coroutine(x: int) -> int:
+        nonlocal call_count
+        call_count += 1
+        await asyncio.sleep(0.1)
+        return x * 2
+
+    result1 = await sample_coroutine(2)
+    assert result1 == 4
+    assert call_count == 1
+
+    result2 = await sample_coroutine(2)
+    assert result2 == 4
+    assert call_count == 1
+
+    result3 = await sample_coroutine(3)
+    assert result3 == 6
+    assert call_count == 2
+
+
+@pytest.mark.asyncio
+async def test_cache_coroutine_result_with_kwargs(
+    event: EventContext, monkeypatch: Any
+) -> None:
+    monkeypatch.setattr(cache, "event", event)
+
+    call_count = 0
+
+    @cache.cache_coroutine_result()
+    async def sample_coroutine(x: int, y: int = 1) -> int:
+        nonlocal call_count
+        call_count += 1
+        await asyncio.sleep(0.1)
+        return x * y
+
+    result1 = await sample_coroutine(2, y=3)
+    assert result1 == 6
+    assert call_count == 1
+
+    result2 = await sample_coroutine(2, y=3)
+    assert result2 == 6
+    assert call_count == 1
+
+    result3 = await sample_coroutine(2, y=4)
+    assert result3 == 8
+    assert call_count == 2
+
+
+@pytest.mark.asyncio
+async def test_cache_coroutine_result_no_cache(
+    event: EventContext, monkeypatch: Any
+) -> None:
+    monkeypatch.setattr(cache, "event", event)
+
+    call_count = 0
+
+    @cache.cache_coroutine_result()
+    async def sample_coroutine(x: int) -> int:
+        nonlocal call_count
+        call_count += 1
+        await asyncio.sleep(0.1)
+        return x * 2
+
+    result1 = await sample_coroutine(2)
+    assert result1 == 4
+    assert call_count == 1
+
+    event.attributes.clear()
+
+    result2 = await sample_coroutine(2)
+    assert result2 == 4
+    assert call_count == 2

port_ocean/utils/cache.py

@@ -1,9 +1,10 @@
 import functools
 import hashlib
-from typing import Callable, AsyncIterator, Any
+from typing import Callable, AsyncIterator, Awaitable, Any
 from port_ocean.context.event import event
 
 AsyncIteratorCallable = Callable[..., AsyncIterator[list[Any]]]
+AsyncCallable = Callable[..., Awaitable[Any]]
 
 
 def hash_func(function_name: str, *args: Any, **kwargs: Any) -> str:
@@ -59,3 +60,38 @@ def cache_iterator_result() -> Callable[[AsyncIteratorCallable], AsyncIteratorCa
         return wrapper
 
     return decorator
+
+
+def cache_coroutine_result() -> Callable[[AsyncCallable], AsyncCallable]:
+    """Coroutine version of `cache_iterator_result` from port_ocean.utils.cache
+
+    Decorator that caches the result of a coroutine function.
+    It checks if the result is already in the cache, and if not,
+    fetches the result, caches it, and returns the cached value.
+
+    The cache is stored in the scope of the running event and is
+    removed when the event is finished.
+
+    Usage:
+    ```python
+    @cache_coroutine_result()
+    async def my_coroutine_function():
+        # Your code here
+    ```
+    """
+
+    def decorator(func: AsyncCallable) -> AsyncCallable:
+        @functools.wraps(func)
+        async def wrapper(*args: Any, **kwargs: Any) -> Any:
+            cache_key = hash_func(func.__name__, *args, **kwargs)
+
+            if cache := event.attributes.get(cache_key):
+                return cache
+
+            result = await func(*args, **kwargs)
+            event.attributes[cache_key] = result
+            return result
+
+        return wrapper
+
+    return decorator

{port_ocean-0.14.0.dist-info → port_ocean-0.14.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: port-ocean
-Version: 0.14.0
+Version: 0.14.1
 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.14.0.dist-info → port_ocean-0.14.1.dist-info}/RECORD

@@ -133,18 +133,19 @@ port_ocean/tests/helpers/smoke_test.py,sha256=_9aJJFRfuGJEg2D2YQJVJRmpreS6gEPHHQ
 port_ocean/tests/log/test_handlers.py,sha256=bTOGnuj8fMIEXepwYblRvcg0FKqApCdyCBtAQZ2BlXM,2115
 port_ocean/tests/test_smoke.py,sha256=uix2uIg_yOm8BHDgHw2hTFPy1fiIyxBGW3ENU_KoFlo,2557
 port_ocean/tests/utils/test_async_iterators.py,sha256=3PLk1emEXekb8LcC5GgVh3OicaX15i5WyaJT_eFnu_4,1336
+port_ocean/tests/utils/test_cache.py,sha256=GzoS8xGCBDbBcPwSDbdimsMMkRvJATrBC7UmFhdW3fw,4906
 port_ocean/utils/__init__.py,sha256=KMGnCPXZJbNwtgxtyMycapkDz8tpSyw23MSYT3iVeHs,91
 port_ocean/utils/async_http.py,sha256=arnH458TExn2Dju_Sy6pHas_vF5RMWnOp-jBz5WAAcE,1226
 port_ocean/utils/async_iterators.py,sha256=CPXskYWkhkZtAG-ducEwM8537t3z5usPEqXR9vcivzw,3715
-port_ocean/utils/cache.py,sha256=3KItZDE2yVrbVDr-hoM8lNna8s2dlpxhP4ICdLjH4LQ,2231
+port_ocean/utils/cache.py,sha256=RgfN4SjjHrEkbqUChyboeD1mrXomolUUjsJtvbkmr3U,3353
 port_ocean/utils/misc.py,sha256=0q2cJ5psqxn_5u_56pT7vOVQ3shDM02iC1lzyWQ_zl0,2098
 port_ocean/utils/queue_utils.py,sha256=KWWl8YVnG-glcfIHhM6nefY-2sou_C6DVP1VynQwzB4,2762
 port_ocean/utils/repeat.py,sha256=0EFWM9d8lLXAhZmAyczY20LAnijw6UbIECf5lpGbOas,3231
 port_ocean/utils/signal.py,sha256=K-6kKFQTltcmKDhtyZAcn0IMa3sUpOHGOAUdWKgx0_E,1369
 port_ocean/utils/time.py,sha256=pufAOH5ZQI7gXvOvJoQXZXZJV-Dqktoj9Qp9eiRwmJ4,1939
 port_ocean/version.py,sha256=UsuJdvdQlazzKGD3Hd5-U7N69STh8Dq9ggJzQFnu9fU,177
-port_ocean-0.14.0.dist-info/LICENSE.md,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-port_ocean-0.14.0.dist-info/METADATA,sha256=A_jE7vo5o-cE6dE8fLzRZGFh7wOJFNquJN5LyZ_ygCA,6673
-port_ocean-0.14.0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-port_ocean-0.14.0.dist-info/entry_points.txt,sha256=F_DNUmGZU2Kme-8NsWM5LLE8piGMafYZygRYhOVtcjA,54
-port_ocean-0.14.0.dist-info/RECORD,,
+port_ocean-0.14.1.dist-info/LICENSE.md,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+port_ocean-0.14.1.dist-info/METADATA,sha256=wOX1EZAWACc4Y91LXEhR497CEE8Vtk8UPsHZDStMaME,6673
+port_ocean-0.14.1.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+port_ocean-0.14.1.dist-info/entry_points.txt,sha256=F_DNUmGZU2Kme-8NsWM5LLE8piGMafYZygRYhOVtcjA,54
+port_ocean-0.14.1.dist-info/RECORD,,