tasklease 0.1.0__py3-none-any.whl

tasklease/generated/protocol.py

@@ -0,0 +1,478 @@
+# Generated by scripts/codegen.mjs. Do not edit.
+
+from __future__ import annotations
+
+from typing import Literal, NotRequired, TypedDict
+
+
+type DomainName = str
+
+
+type TopicName1 = str
+
+
+type TopicName = TopicName1
+
+
+type TopicGlob = str
+
+
+type TaskId = str
+
+
+type Timestamp = int
+
+
+type WorkerLabel = str
+
+
+type IdentityName = str
+
+
+type ErrorCode = Literal[
+    "invalid",
+    "unauthenticated",
+    "forbidden",
+    "not_found",
+    "empty",
+    "exists",
+    "too_large",
+]
+
+
+class Error(TypedDict):
+    code: ErrorCode
+    message: str
+
+
+class ErrorResponse(TypedDict):
+    ok: Literal[False]
+    error: Error
+
+
+class Owner(TypedDict):
+    identity: IdentityName
+    worker: WorkerLabel
+
+
+class Scopes(TypedDict):
+    create: list[TopicGlob]
+    own: list[TopicGlob]
+    read: list[TopicGlob]
+
+
+class Identity(TypedDict):
+    name: IdentityName
+    domains: list[DomainName]
+    scopes: Scopes
+    operator: bool
+
+
+class IdentitySummary(TypedDict):
+    name: IdentityName
+    domains: list[DomainName]
+    scopes: Scopes
+    operator: bool
+    created_at: Timestamp
+    last_seen_at: Timestamp | None
+
+
+class Poller(TypedDict):
+    identity: IdentityName
+    worker: WorkerLabel
+    last_polled_at: Timestamp
+    last_owned_at: Timestamp | None
+
+
+class TopicSummary(TypedDict):
+    topic: TopicName
+    available: int
+    held: int
+    oldest_available_at: Timestamp | None
+    declared: bool
+
+
+class TaskSummary(TypedDict):
+    id: TaskId
+    created_at: Timestamp
+    lease_until: Timestamp
+    owned_count: int
+    creator: IdentityName
+    owner: Owner | None
+    owned_at: Timestamp | None
+    state: Literal["available", "held"]
+
+
+class SnapshotTopic(TypedDict):
+    topic: TopicName
+    available: int
+    held: int
+    oldest_available_at: Timestamp | None
+    declared: bool
+    pollers: list[Poller]
+    tasks: list[TaskSummary]
+    truncated: NotRequired[Literal[True]]
+
+
+class HourlyStats(TypedDict):
+    topic: TopicName
+    hour: Timestamp
+    created: int
+    owned: int
+    reowned: int
+    consumed: int
+    deleted: int
+    wait_ms_sum: int
+    wait_ms_max: int
+    wait_count: int
+    run_ms_sum: int
+    run_ms_max: int
+    run_count: int
+
+
+class BlobGetUrlRequest(TypedDict):
+    domain: DomainName
+    key: str
+
+
+class BlobGetUrlResponse(TypedDict):
+    ok: Literal[True]
+    url: str
+    expires_at: Timestamp
+
+
+class BlobPutUrlRequest(TypedDict):
+    domain: DomainName
+    name: str
+    size: int
+    sha256: str
+
+
+class BlobPutUrlResponse(TypedDict):
+    ok: Literal[True]
+    key: str
+    url: str
+    expires_at: Timestamp
+
+
+class ConsumeResponse(TypedDict):
+    ok: Literal[True]
+    ids: list[TaskId]
+
+
+class CreateResponse(TypedDict):
+    ok: Literal[True]
+    id: TaskId
+
+
+class DeleteRequest(TypedDict):
+    domain: DomainName
+    topic: TopicName
+    id: TaskId
+    lease_before: NotRequired[Timestamp]
+
+
+class DeleteResponse(TypedDict):
+    ok: Literal[True]
+
+
+class IdentityAddRequest(TypedDict):
+    name: IdentityName
+    domains: list[DomainName]
+    scopes: Scopes
+    operator: bool
+
+
+class IdentityAddResponse(TypedDict):
+    ok: Literal[True]
+    name: IdentityName
+    token: str
+
+
+class IdentityListRequest(TypedDict):
+    pass
+
+
+class IdentityListResponse(TypedDict):
+    ok: Literal[True]
+    identities: list[IdentitySummary]
+
+
+class IdentityRevokeRequest(TypedDict):
+    name: IdentityName
+
+
+class IdentityRevokeResponse(TypedDict):
+    ok: Literal[True]
+
+
+class IdentityRotateRequest(TypedDict):
+    name: IdentityName
+
+
+class IdentityRotateResponse(TypedDict):
+    ok: Literal[True]
+    token: str
+
+
+class IdentityWhoamiRequest(TypedDict):
+    pass
+
+
+class IdentityWhoamiResponse(TypedDict):
+    ok: Literal[True]
+    name: IdentityName
+    domains: list[DomainName]
+    scopes: Scopes
+    operator: bool
+
+
+class MonitorSnapshotRequest(TypedDict):
+    domain: DomainName
+    hours: NotRequired[int]
+
+
+class MonitorSnapshotResponse(TypedDict):
+    ok: Literal[True]
+    domain: DomainName
+    generated_at: Timestamp
+    topics: list[SnapshotTopic]
+    stats: list[HourlyStats]
+
+
+class OwnRequest(TypedDict):
+    domain: DomainName
+    topic: TopicName
+    hold_seconds: int
+
+
+class ReadRequest(TypedDict):
+    domain: DomainName
+    topic: TopicName
+    id: TaskId
+
+
+class RenewRequest(TypedDict):
+    domain: DomainName
+    topic: TopicName
+    id: TaskId
+    hold_seconds: int
+
+
+class TasksListRequest(TypedDict):
+    domain: DomainName
+    topic: TopicName
+    cursor: NotRequired[str]
+
+
+class TasksListResponse(TypedDict):
+    ok: Literal[True]
+    tasks: list[TaskSummary]
+    cursor: NotRequired[str]
+
+
+class TopicDeclareRequest(TypedDict):
+    domain: DomainName
+    topic: TopicName
+    declared: bool
+
+
+class TopicDeclareResponse(TypedDict):
+    ok: Literal[True]
+
+
+class TopicPollersRequest(TypedDict):
+    domain: DomainName
+    topic: TopicName
+
+
+class TopicPollersResponse(TypedDict):
+    ok: Literal[True]
+    pollers: list[Poller]
+
+
+class TopicsListRequest(TypedDict):
+    domain: DomainName
+    glob: NotRequired[TopicGlob]
+
+
+class TopicsListResponse(TypedDict):
+    ok: Literal[True]
+    topics: list[TopicSummary]
+
+
+class VersionRequest(TypedDict):
+    pass
+
+
+class VersionResponse(TypedDict):
+    service_version: str
+    protocol: Literal["v1"]
+
+
+class ProtocolTypes(TypedDict):
+    DomainName: NotRequired[DomainName]
+    TopicName: NotRequired[TopicName]
+    TopicGlob: NotRequired[TopicGlob]
+    TaskId: NotRequired[TaskId]
+    Timestamp: NotRequired[Timestamp]
+    WorkerLabel: NotRequired[WorkerLabel]
+    IdentityName: NotRequired[IdentityName]
+    Data: NotRequired[Data]
+    ErrorCode: NotRequired[ErrorCode]
+    Error: NotRequired[Error]
+    ErrorResponse: NotRequired[ErrorResponse]
+    Owner: NotRequired[Owner]
+    Scopes: NotRequired[Scopes]
+    Identity: NotRequired[Identity]
+    IdentitySummary: NotRequired[IdentitySummary]
+    NewTask: NotRequired[NewTask]
+    Poller: NotRequired[Poller]
+    TopicSummary: NotRequired[TopicSummary]
+    Task: NotRequired[Task]
+    TaskSummary: NotRequired[TaskSummary]
+    SnapshotTopic: NotRequired[SnapshotTopic]
+    HourlyStats: NotRequired[HourlyStats]
+    JsonValue: NotRequired[JsonValue]
+    BlobGetUrlRequest: NotRequired[BlobGetUrlRequest]
+    BlobGetUrlResponse: NotRequired[BlobGetUrlResponse]
+    BlobPutUrlRequest: NotRequired[BlobPutUrlRequest]
+    BlobPutUrlResponse: NotRequired[BlobPutUrlResponse]
+    ConsumeRequest: NotRequired[ConsumeRequest]
+    ConsumeResponse: NotRequired[ConsumeResponse]
+    CreateRequest: NotRequired[CreateRequest]
+    CreateResponse: NotRequired[CreateResponse]
+    DeleteRequest: NotRequired[DeleteRequest]
+    DeleteResponse: NotRequired[DeleteResponse]
+    IdentityAddRequest: NotRequired[IdentityAddRequest]
+    IdentityAddResponse: NotRequired[IdentityAddResponse]
+    IdentityListRequest: NotRequired[IdentityListRequest]
+    IdentityListResponse: NotRequired[IdentityListResponse]
+    IdentityRevokeRequest: NotRequired[IdentityRevokeRequest]
+    IdentityRevokeResponse: NotRequired[IdentityRevokeResponse]
+    IdentityRotateRequest: NotRequired[IdentityRotateRequest]
+    IdentityRotateResponse: NotRequired[IdentityRotateResponse]
+    IdentityWhoamiRequest: NotRequired[IdentityWhoamiRequest]
+    IdentityWhoamiResponse: NotRequired[IdentityWhoamiResponse]
+    MonitorSnapshotRequest: NotRequired[MonitorSnapshotRequest]
+    MonitorSnapshotResponse: NotRequired[MonitorSnapshotResponse]
+    OwnRequest: NotRequired[OwnRequest]
+    OwnResponse: NotRequired[OwnResponse]
+    ReadRequest: NotRequired[ReadRequest]
+    ReadResponse: NotRequired[ReadResponse]
+    RenewRequest: NotRequired[RenewRequest]
+    RenewResponse: NotRequired[RenewResponse]
+    TasksListRequest: NotRequired[TasksListRequest]
+    TasksListResponse: NotRequired[TasksListResponse]
+    TopicDeclareRequest: NotRequired[TopicDeclareRequest]
+    TopicDeclareResponse: NotRequired[TopicDeclareResponse]
+    TopicPollersRequest: NotRequired[TopicPollersRequest]
+    TopicPollersResponse: NotRequired[TopicPollersResponse]
+    TopicsListRequest: NotRequired[TopicsListRequest]
+    TopicsListResponse: NotRequired[TopicsListResponse]
+    VersionRequest: NotRequired[VersionRequest]
+    VersionResponse: NotRequired[VersionResponse]
+    OperationRequests: NotRequired[OperationRequests]
+    OperationResponses: NotRequired[OperationResponses]
+
+
+type Data = dict[str, JsonValue]
+
+
+class NewTask(TypedDict):
+    topic: TopicName
+    data: Data
+
+
+class Task(TypedDict):
+    id: TaskId
+    domain: DomainName
+    topic: TopicName
+    created_at: Timestamp
+    lease_until: Timestamp
+    owned_count: int
+    creator: IdentityName
+    owner: Owner | None
+    owned_at: Timestamp | None
+    data: Data
+
+
+type JsonValue = bool | float | str | list[JsonValue] | Data | None
+
+
+class ConsumeRequest(TypedDict):
+    domain: DomainName
+    topic: TopicName
+    id: TaskId
+    tasks: list[NewTask]
+
+
+class CreateRequest(TypedDict):
+    domain: DomainName
+    topic: TopicName
+    data: Data
+
+
+class OwnResponse(TypedDict):
+    ok: Literal[True]
+    task: Task
+
+
+class ReadResponse(TypedDict):
+    ok: Literal[True]
+    task: Task
+
+
+class RenewResponse(TypedDict):
+    ok: Literal[True]
+    task: Task
+
+
+OperationRequests = TypedDict(
+    "OperationRequests",
+    {
+        "blob.get_url": BlobGetUrlRequest,
+        "blob.put_url": BlobPutUrlRequest,
+        "consume": ConsumeRequest,
+        "create": CreateRequest,
+        "delete": DeleteRequest,
+        "identity.add": IdentityAddRequest,
+        "identity.list": IdentityListRequest,
+        "identity.revoke": IdentityRevokeRequest,
+        "identity.rotate": IdentityRotateRequest,
+        "identity.whoami": IdentityWhoamiRequest,
+        "monitor.snapshot": MonitorSnapshotRequest,
+        "own": OwnRequest,
+        "read": ReadRequest,
+        "renew": RenewRequest,
+        "tasks.list": TasksListRequest,
+        "topic.declare": TopicDeclareRequest,
+        "topic.pollers": TopicPollersRequest,
+        "topics.list": TopicsListRequest,
+    },
+)
+
+
+OperationResponses = TypedDict(
+    "OperationResponses",
+    {
+        "blob.get_url": BlobGetUrlResponse,
+        "blob.put_url": BlobPutUrlResponse,
+        "consume": ConsumeResponse,
+        "create": CreateResponse,
+        "delete": DeleteResponse,
+        "identity.add": IdentityAddResponse,
+        "identity.list": IdentityListResponse,
+        "identity.revoke": IdentityRevokeResponse,
+        "identity.rotate": IdentityRotateResponse,
+        "identity.whoami": IdentityWhoamiResponse,
+        "monitor.snapshot": MonitorSnapshotResponse,
+        "own": OwnResponse,
+        "read": ReadResponse,
+        "renew": RenewResponse,
+        "tasks.list": TasksListResponse,
+        "topic.declare": TopicDeclareResponse,
+        "topic.pollers": TopicPollersResponse,
+        "topics.list": TopicsListResponse,
+    },
+)

tasklease/monitor/__init__.py

@@ -0,0 +1 @@
+"""Read-only local access to remote snapshots and durable requester state."""