durabletask

--- durabletask-1.4.0/durabletask/worker.py (excerpt) ---
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.

import asyncio
import inspect
import json
import logging
import os
import random
import time
from concurrent.futures import ThreadPoolExecutor
from dataclasses import dataclass, field
from datetime import datetime, timedelta, timezone
from threading import Event, Thread
from types import GeneratorType
from enum import Enum
from typing import Any, Generator, Optional, Sequence, Tuple, TypeVar, Union, overload
import uuid
from packaging.version import InvalidVersion, parse

import grpc
from google.protobuf import empty_pb2

from durabletask.entities.entity_operation_failed_exception import EntityOperationFailedException
from durabletask.internal import helpers
from durabletask.internal.entity_state_shim import StateShim
from durabletask.internal.helpers import new_timestamp
from durabletask.entities import DurableEntity, EntityLock, EntityInstanceId, EntityContext
from durabletask.internal.json_encode_output_exception import JsonEncodeOutputException
from durabletask.internal.orchestration_entity_context import OrchestrationEntityContext
from durabletask.internal.proto_task_hub_sidecar_service_stub import ProtoTaskHubSidecarServiceStub
import durabletask.internal.helpers as ph
import durabletask.internal.exceptions as pe
import durabletask.internal.orchestrator_service_pb2 as pb
import durabletask.internal.orchestrator_service_pb2_grpc as stubs
import durabletask.internal.shared as shared
import durableta

--- durabletask-1.4.0/durabletask/extensions/azure_blob_payloads/__init__.py (excerpt) ---
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.

"""Azure Blob Storage payload externalization for Durable Task.

This optional extension package provides a :class:`BlobPayloadStore`
that stores large orchestration / activity payloads in Azure Blob
Storage, keeping gRPC message sizes within safe limits.

Install the required dependency with::

    pip install durabletask[azure-blob-payloads]

Usage::

    from durabletask.extensions.azure_blob_payloads import (
        BlobPayloadStore,
        BlobPayloadStoreOptions,
    )

    store = BlobPayloadStore(BlobPayloadStoreOptions(
        connection_string="DefaultEndpointsProtocol=https;...",
    ))
    worker = TaskHubGrpcWorker(payload_store=store)
"""

try:
    from azure.storage.blob import BlobServiceClient  # noqa: F401
except ImportError as exc:
    raise ImportError(
        "The 'azure-storage-blob' package is required for blob payload "
        "support. Install it with:  pip install durabletask[azure-blob-payloads]"
    ) from exc

from durabletask.extensions.azure_blob_payloads.blob_payload_store import BlobPayloadStore
from durabletask.extensions.azure_blob_payloads.options import BlobPayloadStoreOptions

__all__ = ["BlobPayloadStore", "BlobPayloadStoreOptions"]


--- durabletask-1.4.0/durabletask/extensions/azure_blob_payloads/blob_payload_store.py (excerpt) ---
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.

"""Azure Blob Storage implementation of :class:`PayloadStore`."""

from __future__ import annotations

import gzip
import logging
import uuid
from typing import Optional

from azure.core.exceptions import ResourceExistsError
from azure.storage.blob import BlobServiceClient
from azure.storage.blob.aio import BlobServiceClient as AsyncBlobServiceClient

from durabletask.extensions.azure_blob_payloads.options import BlobPayloadStoreOptions
from durabletask.payload.store import PayloadStore

logger = logging.getLogger("durabletask-blobpayloads")

# Token format matching the .NET SDK: blob:v1:<container>:<blobName>
_TOKEN_PREFIX = "blob:v1:"


class BlobPayloadStore(PayloadStore):
    """Stores and retrieves large payloads in Azure Blob Storage.

    This implementation is compatible with the .NET SDK's
    ``AzureBlobPayloadsSideCarInterceptor`` – both SDKs use the same
    token format (``blob:v1:<container>:<blobName>``) and the same
    storage layout, allowing cross-language interoperability.

    Example::

        store = BlobPayloadStore(BlobPayloadStoreOptions(
            connection_string="...",
        ))

    Args:
        options: A :class:`BlobPayloadStoreOptions` with all settings.
    """

    def __init__(self, options: BlobPayloadStoreOptions):
        if not options.connection_string and not options.account_url:
            raise ValueError(
                "Either 'connection_string' or