mp3printer/_types.py at master · firetech/mp3printer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
# pyright: strict

from typing import Literal, TypeAlias

import pydantic as pyd

#############
# Responses #
#############


class AddressResponse(pyd.BaseModel):
    type: Literal["address"] = "address"
    address: str | None


class ListContentResponseEntry(pyd.BaseModel):
    id: str
    title: str
    duration: float | None
    nick: str | None
    address: str | None
    prio: int

    model_config = {"extra": "ignore"}


class ListContentResponse(pyd.BaseModel):
    type: Literal["list"] = "list"
    position: float
    list: list[ListContentResponseEntry]


class ListFallbackResponse(pyd.BaseModel):
    type: Literal["fallback"] = "fallback"
    description: str


ListResponse: TypeAlias = ListContentResponse | ListFallbackResponse


class ProgressResponse(pyd.BaseModel):
    type: Literal["progress"] = "progress"
    position: float


Response: TypeAlias = AddressResponse | ListResponse | ProgressResponse

############
# Requests #
############


class CancelRequest(pyd.BaseModel):
    type: Literal["cancel"] = "cancel"
    id: str


Request: TypeAlias = CancelRequest