prefect-client 3.7.4.dev4__tar.gz → 3.7.5.dev1__tar.gz

{prefect_client-3.7.4.dev4 → prefect_client-3.7.5.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prefect-client
-Version: 3.7.4.dev4
+Version: 3.7.5.dev1
 Summary: Workflow orchestration and management.
 Project-URL: Changelog, https://github.com/PrefectHQ/prefect/releases
 Project-URL: Documentation, https://docs.prefect.io

prefect_client-3.7.5.dev1/src/prefect/_build_info.py

@@ -0,0 +1,5 @@
+# Generated by versioningit
+__version__ = "3.7.5.dev1"
+__build_date__ = "2026-06-06 08:58:56.783483+00:00"
+__git_commit__ = "db66b14dbaea18e726fc4ea0100fd194383c6c59"
+__dirty__ = False

{prefect_client-3.7.4.dev4 → prefect_client-3.7.5.dev1}/src/prefect/client/schemas/schedules.py

@@ -26,6 +26,26 @@ MAX_ITERATIONS = 1000
 MAX_RRULE_LENGTH = 6500
 
 
+def _iana_timezone_name(tzinfo: datetime.tzinfo) -> str:
+    """Return a valid IANA timezone name for *tzinfo*, or `"UTC"`."""
+    for attr in ("key", "name"):
+        value: Optional[str] = getattr(tzinfo, attr, None)
+        if value and is_valid_timezone(value):
+            return value
+
+    filename: Optional[str] = getattr(tzinfo, "_filename", None)
+    if filename:
+        normalized = filename.replace("\\", "/")
+        marker = "/zoneinfo/"
+        idx = normalized.find(marker)
+        if idx != -1:
+            candidate = normalized[idx + len(marker) :]
+            if candidate and is_valid_timezone(candidate):
+                return candidate
+
+    return "UTC"
+
+
 def is_valid_timezone(v: str) -> bool:
     """
     Validate that the provided timezone is a valid IANA timezone.
@@ -204,26 +224,27 @@ class RRuleSchedule(PrefectBaseModel):
         if isinstance(rrule, dateutil.rrule.rrule):
             dtstart = _rrule_dt(rrule)
             if dtstart and dtstart.tzinfo is not None:
-                timezone = dtstart.tzinfo.tzname(dtstart)
+                timezone = _iana_timezone_name(dtstart.tzinfo)
             else:
                 timezone = "UTC"
             return RRuleSchedule(rrule=str(rrule), timezone=timezone)
         rrules = _rrule(rrule)
         dtstarts = [dts for rr in rrules if (dts := _rrule_dt(rr)) is not None]
         unique_dstarts = set(d.astimezone(ZoneInfo("UTC")) for d in dtstarts)
-        unique_timezones = set(d.tzinfo for d in dtstarts if d.tzinfo is not None)
+        unique_tz_names = set(
+            _iana_timezone_name(d.tzinfo) for d in dtstarts if d.tzinfo is not None
+        )
 
-        if len(unique_timezones) > 1:
+        if len(unique_tz_names) > 1:
             raise ValueError(
-                f"rruleset has too many dtstart timezones: {unique_timezones}"
+                f"rruleset has too many dtstart timezones: {unique_tz_names}"
             )
 
         if len(unique_dstarts) > 1:
             raise ValueError(f"rruleset has too many dtstarts: {unique_dstarts}")
 
-        if unique_dstarts and unique_timezones:
-            [unique_tz] = unique_timezones
-            timezone = unique_tz.tzname(dtstarts[0])
+        if unique_dstarts and unique_tz_names:
+            [timezone] = unique_tz_names
         else:
             timezone = "UTC"
 

{prefect_client-3.7.4.dev4 → prefect_client-3.7.5.dev1}/src/prefect/deployments/base.py

@@ -278,6 +278,33 @@ def _interval_schedule_to_dict(schedule: IntervalSchedule) -> dict[str, Any]:
     return schedule_config
 
 
+def _deployment_already_saved_to_prefect_file(
+    deployment: dict[str, Any],
+    prefect_file: Path = Path("prefect.yaml"),
+) -> bool:
+    """
+    Return True if `prefect.yaml` already contains a deployment entry matching the
+    given deployment's name and entrypoint.
+
+    Used to decide whether the interactive `prefect deploy` flow should offer to
+    persist a newly-created deployment: a deployment that is already declared in
+    the file does not need to be saved again, but a brand-new one (even when the
+    file otherwise exists) should still be offered up for saving.
+    """
+    if not prefect_file.exists():
+        return False
+
+    with prefect_file.open(mode="r") as f:
+        contents = yaml.safe_load(f) or {}
+
+    for existing in contents.get("deployments") or []:
+        if existing.get("name") == deployment.get("name") and existing.get(
+            "entrypoint"
+        ) == deployment.get("entrypoint"):
+            return True
+    return False
+
+
 def _save_deployment_to_prefect_file(
     deployment: dict[str, Any],
     build_steps: list[dict[str, Any]] | None = None,

{prefect_client-3.7.4.dev4 → prefect_client-3.7.5.dev1}/src/prefect/server/api/workers.py

@@ -25,8 +25,10 @@ import prefect.server.models as models
 import prefect.server.schemas as schemas
 from prefect._internal.uuid7 import uuid7
 from prefect.client.schemas.worker_channel import (
+    CLEANUP_DELIVERY_CAPABILITY,
     WORK_POOL_SNAPSHOT_CAPABILITY,
     WORKER_HEARTBEAT_CAPABILITY,
+    WorkerChannelCapability,
     WorkerChannelCloseReason,
     WorkerChannelProtocolError,
     WorkerHelloFrame,
@@ -48,6 +50,10 @@ from prefect.server.schemas.statuses import WorkQueueStatus
 from prefect.server.utilities import subscriptions
 from prefect.server.utilities import worker_channel as worker_channel_utils
 from prefect.server.utilities.server import PrefectRouter
+from prefect.server.worker_communication.cleanup_queue import (
+    WorkerCleanupQueue,
+    get_worker_cleanup_queue,
+)
 from prefect.types import DateTime
 from prefect.types._datetime import now
 
@@ -61,7 +67,7 @@ router: PrefectRouter = PrefectRouter(
 )
 logger: Logger = get_logger("prefect.server.api.workers")
 
-_OSS_WORKER_CHANNEL_ACCEPTED_CAPABILITIES = [
+_OSS_WORKER_CHANNEL_REQUIRED_CAPABILITIES: list[WorkerChannelCapability] = [
     WORKER_HEARTBEAT_CAPABILITY,
     WORK_POOL_SNAPSHOT_CAPABILITY,
 ]
@@ -183,6 +189,26 @@ class WorkerChannelWorkPoolUpdateEvent:
     changed_fields: dict[str, dict[str, Any]]
 
 
+def _worker_requested_cleanup_delivery(hello: WorkerHelloFrame) -> bool:
+    return (
+        CLEANUP_DELIVERY_CAPABILITY in hello.payload.requested_capabilities
+        and bool(hello.payload.handled_cleanup_kinds)
+        and hello.payload.max_cleanup_concurrency > 0
+    )
+
+
+def _accepted_worker_channel_capabilities(
+    hello: WorkerHelloFrame,
+    *,
+    cleanup_queue_available: bool,
+) -> list[WorkerChannelCapability]:
+    accepted = list(_OSS_WORKER_CHANNEL_REQUIRED_CAPABILITIES)
+    if cleanup_queue_available and _worker_requested_cleanup_delivery(hello):
+        accepted.append(CLEANUP_DELIVERY_CAPABILITY)
+
+    return accepted
+
+
 async def _receive_worker_hello(websocket: WebSocket) -> WorkerHelloFrame:
     try:
         message = await websocket.receive_json()
@@ -290,6 +316,7 @@ async def _build_worker_ready_frame(
     session: AsyncSession,
     work_pool_name: str,
     hello: WorkerHelloFrame,
+    cleanup_queue_available: bool,
 ) -> tuple[WorkerReadyFrame, WorkerChannelWorkPoolUpdateEvent | None]:
     try:
         selected_channel_version = select_worker_channel_version(
@@ -371,7 +398,10 @@ async def _build_worker_ready_frame(
     )
 
     requested_capabilities = list(dict.fromkeys(hello.payload.requested_capabilities))
-    accepted = _OSS_WORKER_CHANNEL_ACCEPTED_CAPABILITIES
+    accepted = _accepted_worker_channel_capabilities(
+        hello,
+        cleanup_queue_available=cleanup_queue_available,
+    )
     accepted_set = set(accepted)
     rejected = [
         capability
@@ -393,7 +423,11 @@ async def _build_worker_ready_frame(
                 ),
                 "accepted_capabilities": accepted,
                 "rejected_capabilities": rejected,
-                "effective_max_cleanup_concurrency": 0,
+                "effective_max_cleanup_concurrency": (
+                    hello.payload.max_cleanup_concurrency
+                    if CLEANUP_DELIVERY_CAPABILITY in accepted_set
+                    else 0
+                ),
                 "resolved_work_queues": [
                     {"id": work_queue.id, "name": work_queue.name}
                     for work_queue in work_queues
@@ -1058,6 +1092,16 @@ async def worker_channel_connect(
 
     try:
         hello = await _receive_worker_hello(websocket)
+        cleanup_queue: WorkerCleanupQueue | None = None
+        if _worker_requested_cleanup_delivery(hello):
+            try:
+                cleanup_queue = get_worker_cleanup_queue()
+            except Exception:
+                logger.exception(
+                    "Worker cleanup delivery queue initialization failed; "
+                    "rejecting cleanup delivery capability"
+                )
+
         async with worker_channel_utils.messaging.ephemeral_subscription(
             worker_channel_utils.WORKER_CHANNEL_SNAPSHOT_TOPIC,
         ) as consumer_kwargs:
@@ -1066,6 +1110,7 @@ async def worker_channel_connect(
                     session=session,
                     work_pool_name=work_pool_name,
                     hello=hello,
+                    cleanup_queue_available=cleanup_queue is not None,
                 )
 
             if work_pool_update_event is not None:
@@ -1094,6 +1139,17 @@ async def worker_channel_connect(
                 work_pool_id=ready.payload.initial_snapshot.work_pool.id,
                 consumer_id=hello.payload.consumer_id,
                 worker_name=hello.payload.worker_name,
+                cleanup_queue=(
+                    cleanup_queue
+                    if CLEANUP_DELIVERY_CAPABILITY
+                    in ready.payload.accepted_capabilities
+                    else None
+                ),
+                cleanup_kinds=tuple(hello.payload.handled_cleanup_kinds),
+                cleanup_work_queue_ids=tuple(
+                    work_queue.id for work_queue in ready.payload.resolved_work_queues
+                ),
+                max_cleanup_concurrency=ready.payload.effective_max_cleanup_concurrency,
             )
             await connection.run(ready, consumer_kwargs)
 

prefect_client-3.7.4.dev4/src/prefect/_build_info.py

@@ -1,5 +0,0 @@
-# Generated by versioningit
-__version__ = "3.7.4.dev4"
-__build_date__ = "2026-06-05 09:16:11.920641+00:00"
-__git_commit__ = "0bad21f4ce99db95716f64cc2eab836eb274b52c"
-__dirty__ = False