apify 2.2.0b14__py3-none-any.whl → 2.2.1b1__py3-none-any.whl

apify/__init__.py

@@ -2,7 +2,15 @@ from importlib import metadata
 
 from apify_shared.consts import WebhookEventType
 from crawlee import Request
-from crawlee.events._types import Event
+from crawlee.events import (
+    Event,
+    EventAbortingData,
+    EventExitData,
+    EventListener,
+    EventMigratingData,
+    EventPersistStateData,
+    EventSystemInfoData,
+)
 
 from apify._actor import Actor
 from apify._configuration import Configuration
@@ -15,6 +23,12 @@ __all__ = [
     'Actor',
     'Configuration',
     'Event',
+    'EventAbortingData',
+    'EventExitData',
+    'EventListener',
+    'EventMigratingData',
+    'EventPersistStateData',
+    'EventSystemInfoData',
     'ProxyConfiguration',
     'ProxyInfo',
     'Request',

apify/_actor.py

@@ -4,7 +4,7 @@ import asyncio
 import os
 import sys
 from datetime import timedelta
-from typing import TYPE_CHECKING, Any, Callable, TypeVar, cast
+from typing import TYPE_CHECKING, Any, Callable, Literal, TypeVar, cast, overload
 
 from lazy_object_proxy import Proxy
 from more_itertools import flatten
@@ -14,7 +14,15 @@ from apify_client import ApifyClientAsync
 from apify_shared.consts import ActorEnvVars, ActorExitCodes, ApifyEnvVars
 from apify_shared.utils import ignore_docs, maybe_extract_enum_member_value
 from crawlee import service_locator
-from crawlee.events._types import Event, EventMigratingData, EventPersistStateData
+from crawlee.events import (
+    Event,
+    EventAbortingData,
+    EventExitData,
+    EventListener,
+    EventMigratingData,
+    EventPersistStateData,
+    EventSystemInfoData,
+)
 
 from apify._configuration import Configuration
 from apify._consts import EVENT_LISTENERS_TIMEOUT
@@ -498,7 +506,30 @@ class _ActorType:
         key_value_store = await self.open_key_value_store()
         return await key_value_store.set_value(key, value, content_type=content_type)
 
-    def on(self, event_name: Event, listener: Callable) -> Callable:
+    @overload
+    def on(
+        self, event_name: Literal[Event.PERSIST_STATE], listener: EventListener[EventPersistStateData]
+    ) -> EventListener[EventPersistStateData]: ...
+    @overload
+    def on(
+        self, event_name: Literal[Event.SYSTEM_INFO], listener: EventListener[EventSystemInfoData]
+    ) -> EventListener[EventSystemInfoData]: ...
+    @overload
+    def on(
+        self, event_name: Literal[Event.MIGRATING], listener: EventListener[EventMigratingData]
+    ) -> EventListener[EventMigratingData]: ...
+    @overload
+    def on(
+        self, event_name: Literal[Event.ABORTING], listener: EventListener[EventAbortingData]
+    ) -> EventListener[EventAbortingData]: ...
+    @overload
+    def on(
+        self, event_name: Literal[Event.EXIT], listener: EventListener[EventExitData]
+    ) -> EventListener[EventExitData]: ...
+    @overload
+    def on(self, event_name: Event, listener: EventListener[None]) -> EventListener[Any]: ...
+
+    def on(self, event_name: Event, listener: EventListener[Any]) -> EventListener[Any]:
         """Add an event listener to the Actor's event manager.
 
         The following events can be emitted:
@@ -527,6 +558,19 @@ class _ActorType:
         self._event_manager.on(event=event_name, listener=listener)
         return listener
 
+    @overload
+    def off(self, event_name: Literal[Event.PERSIST_STATE], listener: EventListener[EventPersistStateData]) -> None: ...
+    @overload
+    def off(self, event_name: Literal[Event.SYSTEM_INFO], listener: EventListener[EventSystemInfoData]) -> None: ...
+    @overload
+    def off(self, event_name: Literal[Event.MIGRATING], listener: EventListener[EventMigratingData]) -> None: ...
+    @overload
+    def off(self, event_name: Literal[Event.ABORTING], listener: EventListener[EventAbortingData]) -> None: ...
+    @overload
+    def off(self, event_name: Literal[Event.EXIT], listener: EventListener[EventExitData]) -> None: ...
+    @overload
+    def off(self, event_name: Event, listener: EventListener[None]) -> None: ...
+
     def off(self, event_name: Event, listener: Callable | None = None) -> None:
         """Remove a listener, or all listeners, from an Actor event.
 

{apify-2.2.0b14.dist-info → apify-2.2.1b1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: apify
-Version: 2.2.0b14
+Version: 2.2.1b1
 Summary: Apify SDK for Python
 License: Apache-2.0
 Keywords: apify,sdk,automation,chrome,crawlee,crawler,headless,scraper,scraping

{apify-2.2.0b14.dist-info → apify-2.2.1b1.dist-info}/RECORD

@@ -1,5 +1,5 @@
-apify/__init__.py,sha256=99ynaDWBLEcCjdLq7R0Exy_iACsXiXoQ8VUZKmbzTeM,550
-apify/_actor.py,sha256=X8UCTYVJmsq0Nus7aru1ayEgrbo8QE__0jKaN8aWMZ0,44313
+apify/__init__.py,sha256=HpgKg2FZWJuSPfDygzJ62psylhw4NN4tKFnoYUIhcd4,838
+apify/_actor.py,sha256=e9DeE9vqKzPyiyDO-yRufpqvjAn2sV0KldElWmSTy0I,46137
 apify/_configuration.py,sha256=T3Z_o_W98iSyTbrutfb578yW51aexZ_V0FcLwTxFLjI,10878
 apify/_consts.py,sha256=_Xq4hOfOA1iZ3n1P967YWdyncKivpbX6RTlp_qanUoE,330
 apify/_crypto.py,sha256=e0_aM3l9_5Osk-jszYOOjrAKK60OggSHbiw5c30QnsU,5638
@@ -32,7 +32,7 @@ apify/scrapy/utils.py,sha256=758DcHCSAgCTProY0QX74uJ1XrzVsQwvCmFanj2f_3Q,2928
 apify/storages/__init__.py,sha256=FW-z6ubuPnHGM-Wp15T8mR5q6lnpDGrCW-IkgZd5L30,177
 apify/storages/_request_list.py,sha256=-lZJcE5nq69aJhGFJ7Sh2ctqgAWUDyOwYm5_0y1hdAE,5865
 apify/storages/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-apify-2.2.0b14.dist-info/LICENSE,sha256=AsFjHssKjj4LGd2ZCqXn6FBzMqcWdjQre1byPPSypVw,11355
-apify-2.2.0b14.dist-info/METADATA,sha256=vagvl8FfixTKDEiYx3WI7D541NifN21SGjWuFVCuOHE,8714
-apify-2.2.0b14.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-apify-2.2.0b14.dist-info/RECORD,,
+apify-2.2.1b1.dist-info/LICENSE,sha256=AsFjHssKjj4LGd2ZCqXn6FBzMqcWdjQre1byPPSypVw,11355
+apify-2.2.1b1.dist-info/METADATA,sha256=f5wyK4TgVDod-EHYXMy0Dg30LZnnKDDpsI3OXQ2F2Jk,8713
+apify-2.2.1b1.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+apify-2.2.1b1.dist-info/RECORD,,