EngineTribe/models.py at main · KeiDongo/EngineTribe · 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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
from pydantic import BaseModel as PydanticModel
from typing import Optional, Union


class ErrorMessage(PydanticModel):
    error_type: str
    message: str


class APIKeyErrorMessage(ErrorMessage):
    error_type: str = "004"
    message: str = "Invalid API key."
    api_key: str


class UserErrorMessage(ErrorMessage):
    error_type: Optional[str] = "006"
    message: Optional[str] = "User not found."
    im_id: Optional[str | int]
    username: Optional[str]


class SuccessMessage(PydanticModel):
    success: str
    type: str


class StageSuccessMessage(SuccessMessage):
    success: str
    type: Optional[str] = "stage"
    id: Optional[str]


class UserSuccessMessage(SuccessMessage):
    success: str
    type: Optional[str] = "user"
    username: Optional[str]
    im_id: Optional[str]


class UserPermissionSuccessMessage(UserSuccessMessage):
    permission: str
    value: bool


class ClientSuccessMessage(SuccessMessage):
    success: Optional[str | None]
    type: Optional[str] = "client"
    token: str
    client_type: Optional[str]
    locale: Optional[str]
    mobile: Optional[bool]
    proxied: Optional[bool]


class ClientListMessage(PydanticModel):
    type: Optional[str] = "client_list"
    result: list[ClientSuccessMessage]


class UserLoginProfile(PydanticModel):
    username: str
    admin: bool
    mod: bool
    booster: bool
    goomba: Optional[bool] = True
    alias: str
    id: str
    uploads: int
    mobile: bool
    auth_code: str


class LegacyUserLoginProfile(PydanticModel):
    alias: str
    id: str
    goomba: Optional[bool] = True
    auth_code: str
    ip: Optional[str] = "127.0.0.1"


class UserInfo(PydanticModel):
    username: str
    im_id: int
    uploads: int
    is_admin: bool
    is_mod: bool
    is_booster: bool
    is_valid: bool
    is_banned: bool


class UserInfoMessage(PydanticModel):
    type: Optional[str] = "user"
    result: UserInfo


class LevelDetailsUserData(PydanticModel):
    completed: str
    liked: str


class LevelDetails(PydanticModel):
    name: str
    likes: int
    dislikes: int
    comments: Optional[int] = 0
    intentos: int
    muertes: int
    victorias: int
    apariencia: int
    entorno: int
    etiquetas: str
    featured: int
    user_data: LevelDetailsUserData
    record: dict[str, Union[str, int]]
    date: str
    author: str
    descripcion: str
    archivo: str
    id: str


class DetailedSearchResults(PydanticModel):
    type: Optional[str] = "detailed_search"
    num_rows: int
    rows_perpage: int
    pages: int
    result: list[LevelDetails]


class SingleLevelDetails(PydanticModel):
    type: Optional[str] = "id"
    result: LevelDetails


class ErrorMessageException(Exception):
    def __init__(
            self,
            error_type: str,
            message: str
    ) -> None:
        self.error_type = error_type
        self.message = message


class UpdatePermissionRequestBody(PydanticModel):
    api_key: str
    username: Optional[str]
    im_id: Optional[str | int]
    permission: str
    value: bool


class UpdatePasswordRequestBody(PydanticModel):
    api_key: str
    username: str
    password_hash: str
    im_id: str | int


class UserInfoRequestBody(PydanticModel):
    username: Optional[str]
    im_id: Optional[str | int]