prefect-client 3.2.0__tar.gz → 3.2.2__tar.gz

{prefect-client-3.2.0/src/prefect_client.egg-info → prefect-client-3.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prefect-client
-Version: 3.2.0
+Version: 3.2.2
 Summary: Workflow orchestration and management.
 Home-page: https://www.prefect.io
 Author: Prefect Technologies, Inc.

{prefect-client-3.2.0 → prefect-client-3.2.2}/requirements-dev.txt

@@ -28,7 +28,7 @@ respx
 # Linters and dev tools that are also in .pre-commit-config.yaml, which
 # should usually be updated together.
 codespell==2.4.1
-ruff==0.9.4
+ruff==0.9.5
 mypy==1.14.1
 vermin==1.6.0
 

{prefect-client-3.2.0 → prefect-client-3.2.2}/src/prefect/_internal/schemas/bases.py

@@ -3,15 +3,14 @@ Utilities for creating and working with Prefect REST API schemas.
 """
 
 import datetime
-from typing import Any, ClassVar, Optional, TypeVar, cast
+from typing import Any, ClassVar, Optional, TypeVar
 from uuid import UUID, uuid4
 
-import pendulum
 from pydantic import BaseModel, ConfigDict, Field
 from rich.repr import RichReprResult
 from typing_extensions import Self
 
-from prefect.types import DateTime
+from prefect.types._datetime import DateTime, create_datetime_instance
 from prefect.utilities.generics import validate_list
 
 T = TypeVar("T")
@@ -73,11 +72,9 @@ class PrefectBaseModel(BaseModel):
                 and name == "timestamp"
                 and value
             ):
-                value = cast(pendulum.DateTime, pendulum.instance(value)).isoformat()
+                value = create_datetime_instance(value).isoformat()
             elif isinstance(field.annotation, datetime.datetime) and value:
-                value = cast(
-                    pendulum.DateTime, pendulum.instance(value)
-                ).diff_for_humans()
+                value = create_datetime_instance(value).diff_for_humans()
 
             yield name, value, field.get_default()
 

{prefect-client-3.2.0 → prefect-client-3.2.2}/src/prefect/_internal/schemas/validators.py

@@ -19,9 +19,8 @@ from typing import TYPE_CHECKING, Any, Optional, TypeVar, Union, overload
 from uuid import UUID
 
 import jsonschema
-import pendulum
-import pendulum.tz
 
+from prefect.types._datetime import DateTime, create_datetime_instance, get_timezones
 from prefect.utilities.collections import isiterable
 from prefect.utilities.filesystem import relative_path_to_current_platform
 from prefect.utilities.importtools import from_qualified_name
@@ -253,8 +252,8 @@ def reconcile_paused_deployment(values: MM) -> MM:
     return values
 
 
-def default_anchor_date(v: pendulum.DateTime) -> pendulum.DateTime:
-    return pendulum.instance(v)
+def default_anchor_date(v: DateTime) -> DateTime:
+    return create_datetime_instance(v)
 
 
 @overload
@@ -271,7 +270,7 @@ def default_timezone(
     v: Optional[str], values: Optional[Mapping[str, Any]] = None
 ) -> Optional[str]:
     values = values or {}
-    timezones = pendulum.tz.timezones()
+    timezones = get_timezones()
 
     if v is not None:
         if v and v not in timezones:
@@ -283,7 +282,7 @@ def default_timezone(
 
     # anchor schedules
     elif "anchor_date" in values:
-        anchor_date: pendulum.DateTime = values["anchor_date"]
+        anchor_date: DateTime = values["anchor_date"]
         tz = "UTC" if anchor_date.tz is None else anchor_date.tz.name
         # sometimes anchor dates have "timezones" that are UTC offsets
         # like "-04:00". This happens when parsing ISO8601 strings.

{prefect-client-3.2.0 → prefect-client-3.2.2}/src/prefect/_result_records.py

@@ -9,6 +9,7 @@ from typing import (
     Optional,
     TypeVar,
     Union,
+    cast,
 )
 from uuid import UUID
 
@@ -146,6 +147,8 @@ class ResultRecord(BaseModel, Generic[R]):
     @classmethod
     def coerce_old_format(cls, value: dict[str, Any] | Any) -> dict[str, Any]:
         if isinstance(value, dict):
+            if TYPE_CHECKING:  # TODO: # isintance doesn't accept generic parameters
+                value = cast(dict[str, Any], value)
             if "data" in value:
                 value["result"] = value.pop("data")
             if "metadata" not in value:
@@ -232,4 +235,6 @@ class ResultRecord(BaseModel, Generic[R]):
     def __eq__(self, other: Any | "ResultRecord[Any]") -> bool:
         if not isinstance(other, ResultRecord):
             return False
-        return self.metadata == other.metadata and self.result == other.result
+        return self.model_dump(include={"metadata", "result"}) == other.model_dump(
+            include={"metadata", "result"}
+        )

{prefect-client-3.2.0 → prefect-client-3.2.2}/src/prefect/_version.py

@@ -8,11 +8,11 @@ import json
 
 version_json = '''
 {
- "date": "2025-02-07T18:02:21-0800",
+ "date": "2025-02-13T10:53:49-0500",
  "dirty": true,
  "error": null,
- "full-revisionid": "c8986edebb2dde3e2a931adbe24d2eaefcb799cb",
- "version": "3.2.0"
+ "full-revisionid": "d982c69a8bd4fb92cb250bc91dea25d361601260",
+ "version": "3.2.2"
 }
 '''  # END VERSION_JSON
 

{prefect-client-3.2.0 → prefect-client-3.2.2}/src/prefect/client/orchestration/__init__.py

@@ -11,7 +11,7 @@ from uuid import UUID
 import certifi
 import httpcore
 import httpx
-import pendulum
+
 import pydantic
 from asgi_lifespan import LifespanManager
 from packaging import version
@@ -130,6 +130,7 @@ from prefect.settings import (
     PREFECT_SERVER_ALLOW_EPHEMERAL_MODE,
     PREFECT_TESTING_UNIT_TEST_MODE,
 )
+from prefect.types._datetime import now
 
 if TYPE_CHECKING:
     from prefect.tasks import Task as TaskObject
@@ -148,7 +149,7 @@ T = TypeVar("T")
 
 
 @overload
-def get_client(
+def get_client(  # type: ignore # TODO
     *,
     httpx_settings: Optional[dict[str, Any]] = ...,
     sync_client: Literal[False] = False,
@@ -607,7 +608,7 @@ class PrefectClient(
             List[FlowRun]: a list of FlowRun objects read from the queue
         """
         if scheduled_before is None:
-            scheduled_before = pendulum.now("UTC")
+            scheduled_before = now("UTC")
 
         try:
             response = await self._client.post(

{prefect-client-3.2.0 → prefect-client-3.2.2}/src/prefect/client/schemas/schedules.py

@@ -30,7 +30,7 @@ def is_valid_timezone(v: str) -> bool:
     Validate that the provided timezone is a valid IANA timezone.
 
     Unfortunately this list is slightly different from the list of valid
-    timezones in pendulum that we use for cron and interval timezone validation.
+    timezones we use for cron and interval timezone validation.
     """
     from prefect._internal.pytz import HAS_PYTZ
 
@@ -309,7 +309,7 @@ class RRuleSchedule(PrefectBaseModel):
         Validate that the provided timezone is a valid IANA timezone.
 
         Unfortunately this list is slightly different from the list of valid
-        timezones in pendulum that we use for cron and interval timezone validation.
+        timezones we use for cron and interval timezone validation.
         """
         if v is None:
             return "UTC"

{prefect-client-3.2.0 → prefect-client-3.2.2}/src/prefect/concurrency/asyncio.py

@@ -3,7 +3,8 @@ from contextlib import asynccontextmanager
 from typing import Optional, Union
 
 import anyio
-import pendulum
+
+from prefect.types._datetime import now
 
 from ._asyncio import (
     AcquireConcurrencySlotTimeoutError as AcquireConcurrencySlotTimeoutError,
@@ -69,13 +70,13 @@ async def concurrency(
         max_retries=max_retries,
         strict=strict,
     )
-    acquisition_time = pendulum.now("UTC")
+    acquisition_time = now("UTC")
     emitted_events = emit_concurrency_acquisition_events(limits, occupy)
 
     try:
         yield
     finally:
-        occupancy_period = pendulum.now("UTC") - acquisition_time
+        occupancy_period = now("UTC") - acquisition_time
         try:
             await arelease_concurrency_slots(
                 names, occupy, occupancy_period.total_seconds()

{prefect-client-3.2.0 → prefect-client-3.2.2}/src/prefect/concurrency/sync.py

@@ -2,10 +2,10 @@ from collections.abc import Generator
 from contextlib import contextmanager
 from typing import Optional, TypeVar, Union
 
-import pendulum
 from typing_extensions import Literal
 
 from prefect.client.schemas.responses import MinimalConcurrencyLimitResponse
+from prefect.types._datetime import now
 from prefect.utilities.asyncutils import run_coro_as_sync
 
 from ._asyncio import (
@@ -98,13 +98,13 @@ def concurrency(
         strict=strict,
         max_retries=max_retries,
     )
-    acquisition_time = pendulum.now("UTC")
+    acquisition_time = now("UTC")
     emitted_events = emit_concurrency_acquisition_events(limits, occupy)
 
     try:
         yield
     finally:
-        occupancy_period = pendulum.now("UTC") - acquisition_time
+        occupancy_period = now("UTC") - acquisition_time
         _release_concurrency_slots(names, occupy, occupancy_period.total_seconds())
         emit_concurrency_release_events(limits, occupy, emitted_events)
 

{prefect-client-3.2.0 → prefect-client-3.2.2}/src/prefect/concurrency/v1/asyncio.py

@@ -4,7 +4,6 @@ from typing import TYPE_CHECKING, Optional, Union
 from uuid import UUID
 
 import anyio
-import pendulum
 
 from prefect.concurrency.v1._asyncio import (
     acquire_concurrency_slots,
@@ -15,6 +14,7 @@ from prefect.concurrency.v1._events import (
     emit_concurrency_release_events,
 )
 from prefect.concurrency.v1.context import ConcurrencyContext
+from prefect.types._datetime import now
 
 from ._asyncio import (
     AcquireConcurrencySlotTimeoutError as AcquireConcurrencySlotTimeoutError,
@@ -67,13 +67,13 @@ async def concurrency(
     if TYPE_CHECKING:
         assert not isinstance(acquire_slots, list)
     limits = await acquire_slots
-    acquisition_time = pendulum.now("UTC")
+    acquisition_time = now("UTC")
     emitted_events = emit_concurrency_acquisition_events(limits, task_run_id)
 
     try:
         yield
     finally:
-        occupancy_period = pendulum.now("UTC") - acquisition_time
+        occupancy_period = now("UTC") - acquisition_time
         try:
             release_slots = release_concurrency_slots(
                 names_normalized, task_run_id, occupancy_period.total_seconds()

{prefect-client-3.2.0 → prefect-client-3.2.2}/src/prefect/concurrency/v1/sync.py

@@ -4,7 +4,7 @@ from contextlib import contextmanager
 from typing import Optional, TypeVar, Union
 from uuid import UUID
 
-import pendulum
+from prefect.types._datetime import now
 
 from ._asyncio import acquire_concurrency_slots, release_concurrency_slots
 from ._events import (
@@ -59,13 +59,13 @@ def concurrency(
     )
     assert not asyncio.iscoroutine(result)
     limits = result
-    acquisition_time = pendulum.now("UTC")
+    acquisition_time = now("UTC")
     emitted_events = emit_concurrency_acquisition_events(limits, task_run_id)
 
     try:
         yield
     finally:
-        occupancy_period = pendulum.now("UTC") - acquisition_time
+        occupancy_period = now("UTC") - acquisition_time
         release_concurrency_slots(
             names, task_run_id, occupancy_period.total_seconds(), **force
         )

{prefect-client-3.2.0 → prefect-client-3.2.2}/src/prefect/deployments/flow_runs.py

@@ -3,7 +3,6 @@ from typing import TYPE_CHECKING, Any, Iterable, Optional, Union
 from uuid import UUID
 
 import anyio
-import pendulum
 
 import prefect
 from prefect._result_records import ResultRecordMetadata
@@ -16,6 +15,7 @@ from prefect.tasks import Task
 from prefect.telemetry.run_telemetry import (
     LABELS_TRACEPARENT_KEY,
 )
+from prefect.types._datetime import now
 from prefect.utilities.asyncutils import sync_compatible
 from prefect.utilities.slugify import slugify
 
@@ -96,7 +96,7 @@ async def run_deployment(
         raise ValueError("`timeout` cannot be negative")
 
     if scheduled_time is None:
-        scheduled_time = pendulum.now("UTC")
+        scheduled_time = now("UTC")
 
     parameters = parameters or {}
 

{prefect-client-3.2.0 → prefect-client-3.2.2}/src/prefect/docker/docker_image.py

@@ -1,11 +1,10 @@
 from pathlib import Path
 from typing import Any, Optional
 
-from pendulum import now as pendulum_now
-
 from prefect.settings import (
     PREFECT_DEFAULT_DOCKER_BUILD_NAMESPACE,
 )
+from prefect.types._datetime import now
 from prefect.utilities.dockerutils import (
     PushError,
     build_image,
@@ -54,7 +53,7 @@ class DockerImage:
         # join the namespace and repository to create the full image name
         # ignore namespace if it is None
         self.name: str = "/".join(filter(None, [namespace, repository]))
-        self.tag: str = tag or image_tag or slugify(pendulum_now("utc").isoformat())
+        self.tag: str = tag or image_tag or slugify(now("UTC").isoformat())
         self.dockerfile: str = dockerfile
         self.build_kwargs: dict[str, Any] = build_kwargs
 

{prefect-client-3.2.0 → prefect-client-3.2.2}/src/prefect/engine.py

@@ -64,7 +64,7 @@ if __name__ == "__main__":
 
     except Abort:
         engine_logger.info(
-            "Engine execution of flow run '{flow_run_id}' aborted by orchestrator."
+            f"Engine execution of flow run '{flow_run_id}' aborted by orchestrator."
         )
         exit(0)
     except Pause:

{prefect-client-3.2.0 → prefect-client-3.2.2}/src/prefect/events/clients.py

@@ -2,6 +2,7 @@ import abc
 import asyncio
 import os
 import ssl
+from datetime import timedelta
 from types import TracebackType
 from typing import (
     TYPE_CHECKING,
@@ -21,7 +22,6 @@ from urllib.request import proxy_bypass
 from uuid import UUID
 
 import orjson
-import pendulum
 from cachetools import TTLCache
 from prometheus_client import Counter
 from python_socks.async_.asyncio import Proxy
@@ -44,6 +44,7 @@ from prefect.settings import (
     PREFECT_DEBUG_MODE,
     PREFECT_SERVER_ALLOW_EPHEMERAL_MODE,
 )
+from prefect.types._datetime import add_years, now
 
 if TYPE_CHECKING:
     from prefect.events.filters import EventFilter
@@ -653,8 +654,8 @@ class PrefectEventSubscriber:
         from prefect.events.filters import EventOccurredFilter
 
         self._filter.occurred = EventOccurredFilter(
-            since=pendulum.now("UTC").subtract(minutes=1),
-            until=pendulum.now("UTC").add(years=1),
+            since=now("UTC") - timedelta(minutes=1),
+            until=add_years(now("UTC"), 1),
         )
 
         logger.debug("  filtering events since %s...", self._filter.occurred.since)

{prefect-client-3.2.0 → prefect-client-3.2.2}/src/prefect/events/related.py

@@ -14,9 +14,7 @@ from typing import (
 )
 from uuid import UUID
 
-import pendulum
-
-from prefect.types import DateTime
+from prefect.types._datetime import DateTime, now
 
 from .schemas.events import RelatedResource
 
@@ -79,7 +77,7 @@ async def related_resources_from_run_context(
 
     related_objects: List[ResourceCacheEntry] = []
 
-    async def dummy_read():
+    async def dummy_read() -> ResourceCacheEntry:
         return {}
 
     if flow_run_context:
@@ -207,7 +205,7 @@ async def _get_and_cache_related_object(
             "object": obj_,
         }
 
-    cache[cache_key] = (entry, pendulum.now("UTC"))
+    cache[cache_key] = (entry, now("UTC"))
 
     # In the case of a worker or agent this cache could be long-lived. To keep
     # from running out of memory only keep `MAX_CACHE_SIZE` entries in the

{prefect-client-3.2.0 → prefect-client-3.2.2}/src/prefect/flow_engine.py

@@ -49,12 +49,18 @@ from prefect.context import (
 )
 from prefect.exceptions import (
     Abort,
+    MissingFlowError,
     Pause,
     PrefectException,
     TerminationSignal,
     UpstreamTaskError,
 )
-from prefect.flows import Flow, load_flow_from_entrypoint, load_flow_from_flow_run
+from prefect.flows import (
+    Flow,
+    load_flow_from_entrypoint,
+    load_flow_from_flow_run,
+    load_function_and_convert_to_flow,
+)
 from prefect.futures import PrefectFuture, resolve_futures_to_states
 from prefect.logging.loggers import (
     flow_run_logger,
@@ -125,7 +131,10 @@ def load_flow(flow_run: FlowRun) -> Flow[..., Any]:
 
     if entrypoint:
         # we should not accept a placeholder flow at runtime
-        flow = load_flow_from_entrypoint(entrypoint, use_placeholder_flow=False)
+        try:
+            flow = load_flow_from_entrypoint(entrypoint, use_placeholder_flow=False)
+        except MissingFlowError:
+            flow = load_function_and_convert_to_flow(entrypoint)
     else:
         flow = run_coro_as_sync(
             load_flow_from_flow_run(flow_run, use_placeholder_flow=False)

{prefect-client-3.2.0 → prefect-client-3.2.2}/src/prefect/flows.py

@@ -2080,6 +2080,29 @@ def load_flow_from_entrypoint(
     return flow
 
 
+def load_function_and_convert_to_flow(entrypoint: str) -> Flow[P, Any]:
+    """
+    Loads a function from an entrypoint and converts it to a flow if it is not already a flow.
+    """
+
+    if ":" in entrypoint:
+        # split by the last colon once to handle Windows paths with drive letters i.e C:\path\to\file.py:do_stuff
+        path, func_name = entrypoint.rsplit(":", maxsplit=1)
+    else:
+        path, func_name = entrypoint.rsplit(".", maxsplit=1)
+    try:
+        func = import_object(entrypoint)  # pyright: ignore[reportRedeclaration]
+    except AttributeError as exc:
+        raise RuntimeError(
+            f"Function with name {func_name!r} not found in {path!r}."
+        ) from exc
+
+    if isinstance(func, Flow):
+        return func
+    else:
+        return Flow(func, log_prints=True)
+
+
 def serve(
     *args: "RunnerDeployment",
     pause_on_shutdown: bool = True,
@@ -2337,11 +2360,17 @@ async def load_flow_from_flow_run(
     import_path = relative_path_to_current_platform(deployment.entrypoint)
     run_logger.debug(f"Importing flow code from '{import_path}'")
 
-    flow = await run_sync_in_worker_thread(
-        load_flow_from_entrypoint,
-        str(import_path),
-        use_placeholder_flow=use_placeholder_flow,
-    )
+    try:
+        flow = await run_sync_in_worker_thread(
+            load_flow_from_entrypoint,
+            str(import_path),
+            use_placeholder_flow=use_placeholder_flow,
+        )
+    except MissingFlowError:
+        flow = await run_sync_in_worker_thread(
+            load_function_and_convert_to_flow,
+            str(import_path),
+        )
 
     return flow
 

{prefect-client-3.2.0 → prefect-client-3.2.2}/src/prefect/locking/filesystem.py

@@ -1,15 +1,15 @@
 import time
+from datetime import timedelta
 from logging import Logger
 from pathlib import Path
 from typing import Optional
 
 import anyio
-import pendulum
 import pydantic_core
 from typing_extensions import TypedDict
 
 from prefect.logging.loggers import get_logger
-from prefect.types._datetime import DateTime, PendulumDuration
+from prefect.types._datetime import DateTime, now, parse_datetime
 
 from .protocol import LockManager
 
@@ -27,7 +27,7 @@ class _LockInfo(TypedDict):
     """
 
     holder: str
-    expiration: Optional[pendulum.DateTime]
+    expiration: Optional[DateTime]
     path: Path
 
 
@@ -64,7 +64,7 @@ class FileSystemLockManager(LockManager):
                 lock_info["path"] = lock_path
                 expiration = lock_info.get("expiration")
                 lock_info["expiration"] = (
-                    pendulum.parse(expiration) if expiration is not None else None
+                    parse_datetime(expiration) if expiration is not None else None
                 )
             self._locks[key] = lock_info
             return lock_info
@@ -86,7 +86,7 @@ class FileSystemLockManager(LockManager):
             lock_info["path"] = lock_path
             expiration = lock_info.get("expiration")
             lock_info["expiration"] = (
-                pendulum.parse(expiration) if expiration is not None else None
+                parse_datetime(expiration) if expiration is not None else None
             )
             self._locks[key] = lock_info
             return lock_info
@@ -117,7 +117,7 @@ class FileSystemLockManager(LockManager):
                 )
                 return self.acquire_lock(key, holder, acquire_timeout, hold_timeout)
         expiration = (
-            DateTime.now("utc") + PendulumDuration(seconds=hold_timeout)
+            now("UTC") + timedelta(seconds=hold_timeout)
             if hold_timeout is not None
             else None
         )
@@ -166,7 +166,7 @@ class FileSystemLockManager(LockManager):
                 )
                 return self.acquire_lock(key, holder, acquire_timeout, hold_timeout)
         expiration = (
-            DateTime.now("utc") + PendulumDuration(seconds=hold_timeout)
+            now("UTC") + timedelta(seconds=hold_timeout)
             if hold_timeout is not None
             else None
         )
@@ -208,7 +208,7 @@ class FileSystemLockManager(LockManager):
         if (expiration := lock_info.get("expiration")) is None:
             return True
 
-        expired = expiration < DateTime.now("utc")
+        expired = expiration < now("UTC")
         if expired:
             Path(lock_info["path"]).unlink()
             self._locks.pop(key, None)

{prefect-client-3.2.0 → prefect-client-3.2.2}/src/prefect/logging/handlers.py

@@ -8,9 +8,8 @@ import traceback
 import uuid
 import warnings
 from contextlib import asynccontextmanager
-from typing import TYPE_CHECKING, Any, Dict, List, TextIO, Type
+from typing import TYPE_CHECKING, Any, Dict, TextIO, Type
 
-import pendulum
 from rich.console import Console
 from rich.highlighter import Highlighter, NullHighlighter
 from rich.theme import Theme
@@ -35,6 +34,7 @@ from prefect.settings import (
     PREFECT_LOGGING_TO_API_MAX_LOG_SIZE,
     PREFECT_LOGGING_TO_API_WHEN_MISSING_FLOW,
 )
+from prefect.types._datetime import from_timestamp
 
 if sys.version_info >= (3, 12):
     StreamHandler = logging.StreamHandler[TextIO]
@@ -47,7 +47,7 @@ else:
 
 class APILogWorker(BatchedQueueService[Dict[str, Any]]):
     @property
-    def _max_batch_size(self):
+    def _max_batch_size(self) -> int:
         return max(
             PREFECT_LOGGING_TO_API_BATCH_SIZE.value()
             - PREFECT_LOGGING_TO_API_MAX_LOG_SIZE.value(),
@@ -55,10 +55,10 @@ class APILogWorker(BatchedQueueService[Dict[str, Any]]):
         )
 
     @property
-    def _min_interval(self):
+    def _min_interval(self) -> float | None:
         return PREFECT_LOGGING_TO_API_BATCH_INTERVAL.value()
 
-    async def _handle_batch(self, items: List):
+    async def _handle_batch(self, items: list[dict[str, Any]]):
         try:
             await self._client.create_logs(items)
         except Exception as e:
@@ -229,9 +229,7 @@ class APILogHandler(logging.Handler):
             worker_id=worker_id,
             name=record.name,
             level=record.levelno,
-            timestamp=pendulum.from_timestamp(
-                getattr(record, "created", None) or time.time()
-            ),
+            timestamp=from_timestamp(getattr(record, "created", None) or time.time()),
             message=self.format(record),
         ).model_dump(mode="json")
 
@@ -272,9 +270,7 @@ class WorkerAPILogHandler(APILogHandler):
             worker_id=worker_id,
             name=record.name,
             level=record.levelno,
-            timestamp=pendulum.from_timestamp(
-                getattr(record, "created", None) or time.time()
-            ),
+            timestamp=from_timestamp(getattr(record, "created", None) or time.time()),
             message=self.format(record),
         ).model_dump(mode="json")