intellif-aihub 0.1.3__py3-none-any.whl → 0.1.5__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of intellif-aihub might be problematic. Click here for more details.

Files changed (34) hide show
  1. aihub/__init__.py +1 -1
  2. aihub/client.py +95 -91
  3. aihub/exceptions.py +18 -18
  4. aihub/models/artifact.py +137 -137
  5. aihub/models/common.py +13 -13
  6. aihub/models/dataset_management.py +99 -99
  7. aihub/models/document_center.py +28 -28
  8. aihub/models/eval.py +17 -0
  9. aihub/models/labelfree.py +53 -31
  10. aihub/models/model_training_platform.py +234 -230
  11. aihub/models/quota_schedule_management.py +239 -0
  12. aihub/models/tag_resource_management.py +50 -50
  13. aihub/models/task_center.py +117 -117
  14. aihub/models/user_system.py +262 -262
  15. aihub/services/artifact.py +353 -353
  16. aihub/services/dataset_management.py +240 -240
  17. aihub/services/document_center.py +43 -43
  18. aihub/services/eval.py +68 -0
  19. aihub/services/labelfree.py +44 -44
  20. aihub/services/model_training_platform.py +209 -135
  21. aihub/services/quota_schedule_management.py +182 -18
  22. aihub/services/reporter.py +20 -20
  23. aihub/services/tag_resource_management.py +55 -55
  24. aihub/services/task_center.py +190 -190
  25. aihub/services/user_system.py +339 -339
  26. aihub/utils/download.py +69 -69
  27. aihub/utils/http.py +13 -13
  28. aihub/utils/s3.py +77 -77
  29. {intellif_aihub-0.1.3.dist-info → intellif_aihub-0.1.5.dist-info}/METADATA +1 -1
  30. intellif_aihub-0.1.5.dist-info/RECORD +36 -0
  31. {intellif_aihub-0.1.3.dist-info → intellif_aihub-0.1.5.dist-info}/licenses/LICENSE +200 -200
  32. intellif_aihub-0.1.3.dist-info/RECORD +0 -34
  33. {intellif_aihub-0.1.3.dist-info → intellif_aihub-0.1.5.dist-info}/WHEEL +0 -0
  34. {intellif_aihub-0.1.3.dist-info → intellif_aihub-0.1.5.dist-info}/top_level.txt +0 -0
@@ -1,339 +1,339 @@
1
- # !/usr/bin/env python
2
- # -*-coding:utf-8 -*-
3
- from __future__ import annotations
4
-
5
- import httpx
6
-
7
- from ..exceptions import APIError
8
- from ..models.common import APIWrapper
9
- from ..models.user_system import *
10
-
11
-
12
- class UserSystemService:
13
-
14
- def __init__(self, http: httpx.Client):
15
- self._auth = _Auth(http)
16
- self._menu = _Menu(http)
17
- self._role = _Role(http)
18
- self._user = _User(http)
19
-
20
- # ==================================================
21
- # AUTH 一级方法
22
- # ==================================================
23
- def login(self, payload: LoginRequest) -> LoginResponse:
24
- return self._auth.login(payload)
25
-
26
- def signup(self, payload: SignupRequest) -> SignupResponse:
27
- return self._auth.signup(payload)
28
-
29
- # ==================================================
30
- # MENU 一级方法
31
- # ==================================================
32
- def list_menus(self, need_roles: bool = False) -> ListMenusResponse:
33
- return self._menu.list(need_roles)
34
-
35
- def get_menu(self, menu_id: int) -> Menu:
36
- return self._menu.get(menu_id)
37
-
38
- def create_menu(self, payload: CreateMenuRequest) -> int:
39
- return self._menu.create(payload)
40
-
41
- def update_menu(self, menu_id: int, payload: UpdateMenuRequest) -> None:
42
- self._menu.update(menu_id, payload)
43
-
44
- def delete_menu(self, menu_id: int) -> None:
45
- self._menu.delete(menu_id)
46
-
47
- def get_menu_roles(self, menu_id: int) -> List[int]:
48
- return self._menu.get_roles(menu_id)
49
-
50
- def set_menu_roles(self, menu_id: int, role_ids: List[int]) -> None:
51
- self._menu.set_roles(menu_id, role_ids)
52
-
53
- # ==================================================
54
- # ROLE 一级方法
55
- # ==================================================
56
- def list_roles(self, payload: ListRolesRequest) -> ListRolesResponse:
57
- return self._role.list(payload)
58
-
59
- def get_role(self, role_id: int) -> Role:
60
- return self._role.get(role_id)
61
-
62
- def create_role(self, payload: CreateRoleRequest) -> int:
63
- return self._role.create(payload)
64
-
65
- def update_role(self, role_id: int, payload: UpdateRoleRequest) -> None:
66
- self._role.update(role_id, payload)
67
-
68
- def delete_role(self, role_id: int) -> None:
69
- self._role.delete(role_id)
70
-
71
- def get_role_menus(self, role_id: int) -> List[int]:
72
- return self._role.get_menus(role_id)
73
-
74
- def set_role_menus(self, role_id: int, menu_ids: List[int]) -> None:
75
- self._role.set_menus(role_id, menu_ids)
76
-
77
- def search_roles(self, payload: SearchRolesRequest) -> SearchRolesResponse:
78
- return self._role.search(payload)
79
-
80
- # ==================================================
81
- # USER 一级方法
82
- # ==================================================
83
- def list_users(self, payload: ListUsersRequest) -> ListUsersResponse:
84
- return self._user.list(payload)
85
-
86
- def get_user(self, user_id: int) -> User:
87
- return self._user.get(user_id)
88
-
89
- def create_user(self, payload: CreateUserRequest) -> int:
90
- return self._user.create(payload)
91
-
92
- def update_user(self, user_id: int, payload: UpdateUserRequest) -> None:
93
- self._user.update(user_id, payload)
94
-
95
- def delete_user(self, user_id: int) -> None:
96
- self._user.delete(user_id)
97
-
98
- def set_user_roles(self, user_id: int, payload: SetUserRolesRequest) -> None:
99
- self._user.set_roles(user_id, payload)
100
-
101
- def get_user_menus(self, user_id: int, parent_id: int | None = None, auth: str | None = None, ) -> List[TreeMenu]:
102
- return self._user.get_menus(user_id, parent_id=parent_id, auth=auth)
103
-
104
- def search_users(self, payload: SearchUsersRequest) -> SearchUsersResponse:
105
- return self._user.search(payload)
106
-
107
- def search_one(self, payload: SearchUsersRequest) -> int:
108
- return self._user.search_one(payload)
109
-
110
- @property
111
- def auth(self) -> _Auth:
112
- return self._auth
113
-
114
- @property
115
- def menu(self) -> _Menu:
116
- return self._menu
117
-
118
- @property
119
- def role(self) -> _Role:
120
- return self._role
121
-
122
- @property
123
- def user(self) -> _User:
124
- return self._user
125
-
126
-
127
- class _Auth:
128
- _base = "/api/v1/auth"
129
-
130
- def __init__(self, http: httpx.Client):
131
- self._http = http
132
-
133
- def login(self, req: LoginRequest) -> LoginResponse:
134
- resp = self._http.post(
135
- f"{self._base}/login",
136
- json=req.model_dump(by_alias=True, exclude_none=True),
137
- )
138
- wrapper = APIWrapper[LoginResponse].model_validate(resp.json())
139
- if wrapper.code != 0:
140
- raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
141
- return wrapper.data
142
-
143
- def signup(self, req: SignupRequest) -> SignupResponse:
144
- resp = self._http.post(
145
- f"{self._base}/signup",
146
- json=req.model_dump(by_alias=True, exclude_none=True),
147
- )
148
- wrapper = APIWrapper[SignupResponse].model_validate(resp.json())
149
- if wrapper.code != 0:
150
- raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
151
- return wrapper.data
152
-
153
-
154
- class _Menu:
155
- _base = "/api/v1/menus"
156
-
157
- def __init__(self, http: httpx.Client):
158
- self._http = http
159
-
160
- def list(self, need_roles: bool) -> ListMenusResponse:
161
- resp = self._http.get(self._base, params={"need_roles": str(need_roles).lower()})
162
- wrapper = APIWrapper[ListMenusResponse].model_validate(resp.json())
163
- if wrapper.code != 0:
164
- raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
165
- return wrapper.data
166
-
167
- def get(self, menu_id: int) -> Menu:
168
- resp = self._http.get(f"{self._base}/{menu_id}")
169
- wrapper = APIWrapper[Menu].model_validate(resp.json())
170
- if wrapper.code != 0:
171
- raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
172
- return wrapper.data
173
-
174
- def create(self, req: CreateMenuRequest) -> int:
175
- resp = self._http.post(self._base, json=req.model_dump(by_alias=True, exclude_none=True))
176
- wrapper = APIWrapper[CreateMenuResponse].model_validate(resp.json())
177
- if wrapper.code != 0:
178
- raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
179
- return wrapper.data.id
180
-
181
- def update(self, menu_id: int, req: UpdateMenuRequest) -> None:
182
- resp = self._http.put(f"{self._base}/{menu_id}", json=req.model_dump(by_alias=True, exclude_none=True))
183
- wrapper = APIWrapper[dict].model_validate(resp.json())
184
- if wrapper.code != 0:
185
- raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
186
-
187
- def delete(self, menu_id: int) -> None:
188
- resp = self._http.delete(f"{self._base}/{menu_id}")
189
- wrapper = APIWrapper[dict].model_validate(resp.json())
190
- if wrapper.code != 0:
191
- raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
192
-
193
- def get_roles(self, menu_id: int) -> List[int]:
194
- resp = self._http.get(f"{self._base}/{menu_id}/roles")
195
- wrapper = APIWrapper[GetMenuRolesResponse].model_validate(resp.json())
196
- if wrapper.code != 0:
197
- raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
198
- return wrapper.data.role_ids
199
-
200
- def set_roles(self, menu_id: int, role_ids: List[int]) -> None:
201
- resp = self._http.put(f"{self._base}/{menu_id}/roles", json={"role_ids": role_ids})
202
- wrapper = APIWrapper[dict].model_validate(resp.json())
203
- if wrapper.code != 0:
204
- raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
205
-
206
-
207
- class _Role:
208
- _base = "/api/v1/roles"
209
- _search = "/api/v1/search-roles"
210
-
211
- def __init__(self, http: httpx.Client):
212
- self._http = http
213
-
214
- def list(self, req: ListRolesRequest) -> ListRolesResponse:
215
- resp = self._http.get(self._base, params=req.model_dump(by_alias=True, exclude_none=True))
216
- wrapper = APIWrapper[ListRolesResponse].model_validate(resp.json())
217
- if wrapper.code != 0:
218
- raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
219
- return wrapper.data
220
-
221
- def get(self, role_id: int) -> Role:
222
- resp = self._http.get(f"{self._base}/{role_id}")
223
- wrapper = APIWrapper[Role].model_validate(resp.json())
224
- if wrapper.code != 0:
225
- raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
226
- return wrapper.data
227
-
228
- def create(self, req: CreateRoleRequest) -> int:
229
- resp = self._http.post(self._base, json=req.model_dump(by_alias=True, exclude_none=True))
230
- wrapper = APIWrapper[CreateRoleResponse].model_validate(resp.json())
231
- if wrapper.code != 0:
232
- raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
233
- return wrapper.data.id
234
-
235
- def update(self, role_id: int, req: UpdateRoleRequest) -> None:
236
- resp = self._http.put(f"{self._base}/{role_id}", json=req.model_dump(by_alias=True, exclude_none=True))
237
- wrapper = APIWrapper[dict].model_validate(resp.json())
238
- if wrapper.code != 0:
239
- raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
240
-
241
- def delete(self, role_id: int) -> None:
242
- resp = self._http.delete(f"{self._base}/{role_id}")
243
- wrapper = APIWrapper[dict].model_validate(resp.json())
244
- if wrapper.code != 0:
245
- raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
246
-
247
- def get_menus(self, role_id: int) -> List[int]:
248
- resp = self._http.get(f"{self._base}/{role_id}/menus")
249
- wrapper = APIWrapper[GetRoleMenusResponse].model_validate(resp.json())
250
- if wrapper.code != 0:
251
- raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
252
- return wrapper.data.menu_ids
253
-
254
- def set_menus(self, role_id: int, menu_ids: List[int]) -> None:
255
- resp = self._http.put(f"{self._base}/{role_id}/menus", json={"menu_ids": menu_ids})
256
- wrapper = APIWrapper[dict].model_validate(resp.json())
257
- if wrapper.code != 0:
258
- raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
259
-
260
- def search(self, req: SearchRolesRequest) -> SearchRolesResponse:
261
- resp = self._http.post(self._search, json=req.model_dump(by_alias=True, exclude_none=True))
262
- wrapper = APIWrapper[SearchRolesResponse].model_validate(resp.json())
263
- if wrapper.code != 0:
264
- raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
265
- return wrapper.data
266
-
267
-
268
- class _User:
269
- _base = "/api/v1/users"
270
- _search = "/api/v1/search-users"
271
-
272
- def __init__(self, http: httpx.Client):
273
- self._http = http
274
-
275
- def list(self, req: ListUsersRequest) -> ListUsersResponse:
276
- resp = self._http.get(self._base, params=req.model_dump(by_alias=True, exclude_none=True))
277
- wrapper = APIWrapper[ListUsersResponse].model_validate(resp.json())
278
- if wrapper.code != 0:
279
- raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
280
- return wrapper.data
281
-
282
- def get(self, user_id: int) -> User:
283
- resp = self._http.get(f"{self._base}/{user_id}")
284
- wrapper = APIWrapper[User].model_validate(resp.json())
285
- if wrapper.code != 0:
286
- raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
287
- return wrapper.data
288
-
289
- def create(self, req: CreateUserRequest) -> int:
290
- resp = self._http.post(self._base, json=req.model_dump(by_alias=True, exclude_none=True))
291
- wrapper = APIWrapper[CreateUserResponse].model_validate(resp.json())
292
- if wrapper.code != 0:
293
- raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
294
- return wrapper.data.id
295
-
296
- def update(self, user_id: int, req: UpdateUserRequest) -> None:
297
- resp = self._http.put(f"{self._base}/{user_id}", json=req.model_dump(by_alias=True, exclude_none=True))
298
- wrapper = APIWrapper[dict].model_validate(resp.json())
299
- if wrapper.code != 0:
300
- raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
301
-
302
- def delete(self, user_id: int) -> None:
303
- resp = self._http.delete(f"{self._base}/{user_id}")
304
- wrapper = APIWrapper[dict].model_validate(resp.json())
305
- if wrapper.code != 0:
306
- raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
307
-
308
- def set_roles(self, user_id: int, req: SetUserRolesRequest) -> None:
309
- resp = self._http.put(f"{self._base}/{user_id}/roles", json=req.model_dump(by_alias=True, exclude_none=True))
310
- wrapper = APIWrapper[dict].model_validate(resp.json())
311
- if wrapper.code != 0:
312
- raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
313
-
314
- def get_menus(self, user_id: int, parent_id: int | None = None, auth: str | None = None) -> List[TreeMenu]:
315
- params = {}
316
- if parent_id is not None:
317
- params["parent_id"] = parent_id
318
- if auth:
319
- params["auth"] = auth
320
-
321
- resp = self._http.get(f"{self._base}/{user_id}/menus", params=params)
322
- wrapper = APIWrapper[GetUserMenusResponse].model_validate(resp.json())
323
- if wrapper.code != 0:
324
- raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
325
- return wrapper.data.menus
326
-
327
- def search(self, req: SearchUsersRequest) -> SearchUsersResponse:
328
- resp = self._http.post(self._search, json=req.model_dump(by_alias=True, exclude_none=True))
329
- wrapper = APIWrapper[SearchUsersResponse].model_validate(resp.json())
330
- if wrapper.code != 0:
331
- raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
332
- return wrapper.data
333
-
334
- def search_one(self, req: SearchUsersRequest) -> int:
335
- resp = self.search(req)
336
- for user in resp.data:
337
- if user.nickname == req.nickname:
338
- return user.id
339
- raise APIError("no user found")
1
+ # !/usr/bin/env python
2
+ # -*-coding:utf-8 -*-
3
+ from __future__ import annotations
4
+
5
+ import httpx
6
+
7
+ from ..exceptions import APIError
8
+ from ..models.common import APIWrapper
9
+ from ..models.user_system import *
10
+
11
+
12
+ class UserSystemService:
13
+
14
+ def __init__(self, http: httpx.Client):
15
+ self._auth = _Auth(http)
16
+ self._menu = _Menu(http)
17
+ self._role = _Role(http)
18
+ self._user = _User(http)
19
+
20
+ # ==================================================
21
+ # AUTH 一级方法
22
+ # ==================================================
23
+ def login(self, payload: LoginRequest) -> LoginResponse:
24
+ return self._auth.login(payload)
25
+
26
+ def signup(self, payload: SignupRequest) -> SignupResponse:
27
+ return self._auth.signup(payload)
28
+
29
+ # ==================================================
30
+ # MENU 一级方法
31
+ # ==================================================
32
+ def list_menus(self, need_roles: bool = False) -> ListMenusResponse:
33
+ return self._menu.list(need_roles)
34
+
35
+ def get_menu(self, menu_id: int) -> Menu:
36
+ return self._menu.get(menu_id)
37
+
38
+ def create_menu(self, payload: CreateMenuRequest) -> int:
39
+ return self._menu.create(payload)
40
+
41
+ def update_menu(self, menu_id: int, payload: UpdateMenuRequest) -> None:
42
+ self._menu.update(menu_id, payload)
43
+
44
+ def delete_menu(self, menu_id: int) -> None:
45
+ self._menu.delete(menu_id)
46
+
47
+ def get_menu_roles(self, menu_id: int) -> List[int]:
48
+ return self._menu.get_roles(menu_id)
49
+
50
+ def set_menu_roles(self, menu_id: int, role_ids: List[int]) -> None:
51
+ self._menu.set_roles(menu_id, role_ids)
52
+
53
+ # ==================================================
54
+ # ROLE 一级方法
55
+ # ==================================================
56
+ def list_roles(self, payload: ListRolesRequest) -> ListRolesResponse:
57
+ return self._role.list(payload)
58
+
59
+ def get_role(self, role_id: int) -> Role:
60
+ return self._role.get(role_id)
61
+
62
+ def create_role(self, payload: CreateRoleRequest) -> int:
63
+ return self._role.create(payload)
64
+
65
+ def update_role(self, role_id: int, payload: UpdateRoleRequest) -> None:
66
+ self._role.update(role_id, payload)
67
+
68
+ def delete_role(self, role_id: int) -> None:
69
+ self._role.delete(role_id)
70
+
71
+ def get_role_menus(self, role_id: int) -> List[int]:
72
+ return self._role.get_menus(role_id)
73
+
74
+ def set_role_menus(self, role_id: int, menu_ids: List[int]) -> None:
75
+ self._role.set_menus(role_id, menu_ids)
76
+
77
+ def search_roles(self, payload: SearchRolesRequest) -> SearchRolesResponse:
78
+ return self._role.search(payload)
79
+
80
+ # ==================================================
81
+ # USER 一级方法
82
+ # ==================================================
83
+ def list_users(self, payload: ListUsersRequest) -> ListUsersResponse:
84
+ return self._user.list(payload)
85
+
86
+ def get_user(self, user_id: int) -> User:
87
+ return self._user.get(user_id)
88
+
89
+ def create_user(self, payload: CreateUserRequest) -> int:
90
+ return self._user.create(payload)
91
+
92
+ def update_user(self, user_id: int, payload: UpdateUserRequest) -> None:
93
+ self._user.update(user_id, payload)
94
+
95
+ def delete_user(self, user_id: int) -> None:
96
+ self._user.delete(user_id)
97
+
98
+ def set_user_roles(self, user_id: int, payload: SetUserRolesRequest) -> None:
99
+ self._user.set_roles(user_id, payload)
100
+
101
+ def get_user_menus(self, user_id: int, parent_id: int | None = None, auth: str | None = None, ) -> List[TreeMenu]:
102
+ return self._user.get_menus(user_id, parent_id=parent_id, auth=auth)
103
+
104
+ def search_users(self, payload: SearchUsersRequest) -> SearchUsersResponse:
105
+ return self._user.search(payload)
106
+
107
+ def search_one(self, payload: SearchUsersRequest) -> int:
108
+ return self._user.search_one(payload)
109
+
110
+ @property
111
+ def auth(self) -> _Auth:
112
+ return self._auth
113
+
114
+ @property
115
+ def menu(self) -> _Menu:
116
+ return self._menu
117
+
118
+ @property
119
+ def role(self) -> _Role:
120
+ return self._role
121
+
122
+ @property
123
+ def user(self) -> _User:
124
+ return self._user
125
+
126
+
127
+ class _Auth:
128
+ _base = "/api/v1/auth"
129
+
130
+ def __init__(self, http: httpx.Client):
131
+ self._http = http
132
+
133
+ def login(self, req: LoginRequest) -> LoginResponse:
134
+ resp = self._http.post(
135
+ f"{self._base}/login",
136
+ json=req.model_dump(by_alias=True, exclude_none=True),
137
+ )
138
+ wrapper = APIWrapper[LoginResponse].model_validate(resp.json())
139
+ if wrapper.code != 0:
140
+ raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
141
+ return wrapper.data
142
+
143
+ def signup(self, req: SignupRequest) -> SignupResponse:
144
+ resp = self._http.post(
145
+ f"{self._base}/signup",
146
+ json=req.model_dump(by_alias=True, exclude_none=True),
147
+ )
148
+ wrapper = APIWrapper[SignupResponse].model_validate(resp.json())
149
+ if wrapper.code != 0:
150
+ raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
151
+ return wrapper.data
152
+
153
+
154
+ class _Menu:
155
+ _base = "/api/v1/menus"
156
+
157
+ def __init__(self, http: httpx.Client):
158
+ self._http = http
159
+
160
+ def list(self, need_roles: bool) -> ListMenusResponse:
161
+ resp = self._http.get(self._base, params={"need_roles": str(need_roles).lower()})
162
+ wrapper = APIWrapper[ListMenusResponse].model_validate(resp.json())
163
+ if wrapper.code != 0:
164
+ raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
165
+ return wrapper.data
166
+
167
+ def get(self, menu_id: int) -> Menu:
168
+ resp = self._http.get(f"{self._base}/{menu_id}")
169
+ wrapper = APIWrapper[Menu].model_validate(resp.json())
170
+ if wrapper.code != 0:
171
+ raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
172
+ return wrapper.data
173
+
174
+ def create(self, req: CreateMenuRequest) -> int:
175
+ resp = self._http.post(self._base, json=req.model_dump(by_alias=True, exclude_none=True))
176
+ wrapper = APIWrapper[CreateMenuResponse].model_validate(resp.json())
177
+ if wrapper.code != 0:
178
+ raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
179
+ return wrapper.data.id
180
+
181
+ def update(self, menu_id: int, req: UpdateMenuRequest) -> None:
182
+ resp = self._http.put(f"{self._base}/{menu_id}", json=req.model_dump(by_alias=True, exclude_none=True))
183
+ wrapper = APIWrapper[dict].model_validate(resp.json())
184
+ if wrapper.code != 0:
185
+ raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
186
+
187
+ def delete(self, menu_id: int) -> None:
188
+ resp = self._http.delete(f"{self._base}/{menu_id}")
189
+ wrapper = APIWrapper[dict].model_validate(resp.json())
190
+ if wrapper.code != 0:
191
+ raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
192
+
193
+ def get_roles(self, menu_id: int) -> List[int]:
194
+ resp = self._http.get(f"{self._base}/{menu_id}/roles")
195
+ wrapper = APIWrapper[GetMenuRolesResponse].model_validate(resp.json())
196
+ if wrapper.code != 0:
197
+ raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
198
+ return wrapper.data.role_ids
199
+
200
+ def set_roles(self, menu_id: int, role_ids: List[int]) -> None:
201
+ resp = self._http.put(f"{self._base}/{menu_id}/roles", json={"role_ids": role_ids})
202
+ wrapper = APIWrapper[dict].model_validate(resp.json())
203
+ if wrapper.code != 0:
204
+ raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
205
+
206
+
207
+ class _Role:
208
+ _base = "/api/v1/roles"
209
+ _search = "/api/v1/search-roles"
210
+
211
+ def __init__(self, http: httpx.Client):
212
+ self._http = http
213
+
214
+ def list(self, req: ListRolesRequest) -> ListRolesResponse:
215
+ resp = self._http.get(self._base, params=req.model_dump(by_alias=True, exclude_none=True))
216
+ wrapper = APIWrapper[ListRolesResponse].model_validate(resp.json())
217
+ if wrapper.code != 0:
218
+ raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
219
+ return wrapper.data
220
+
221
+ def get(self, role_id: int) -> Role:
222
+ resp = self._http.get(f"{self._base}/{role_id}")
223
+ wrapper = APIWrapper[Role].model_validate(resp.json())
224
+ if wrapper.code != 0:
225
+ raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
226
+ return wrapper.data
227
+
228
+ def create(self, req: CreateRoleRequest) -> int:
229
+ resp = self._http.post(self._base, json=req.model_dump(by_alias=True, exclude_none=True))
230
+ wrapper = APIWrapper[CreateRoleResponse].model_validate(resp.json())
231
+ if wrapper.code != 0:
232
+ raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
233
+ return wrapper.data.id
234
+
235
+ def update(self, role_id: int, req: UpdateRoleRequest) -> None:
236
+ resp = self._http.put(f"{self._base}/{role_id}", json=req.model_dump(by_alias=True, exclude_none=True))
237
+ wrapper = APIWrapper[dict].model_validate(resp.json())
238
+ if wrapper.code != 0:
239
+ raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
240
+
241
+ def delete(self, role_id: int) -> None:
242
+ resp = self._http.delete(f"{self._base}/{role_id}")
243
+ wrapper = APIWrapper[dict].model_validate(resp.json())
244
+ if wrapper.code != 0:
245
+ raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
246
+
247
+ def get_menus(self, role_id: int) -> List[int]:
248
+ resp = self._http.get(f"{self._base}/{role_id}/menus")
249
+ wrapper = APIWrapper[GetRoleMenusResponse].model_validate(resp.json())
250
+ if wrapper.code != 0:
251
+ raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
252
+ return wrapper.data.menu_ids
253
+
254
+ def set_menus(self, role_id: int, menu_ids: List[int]) -> None:
255
+ resp = self._http.put(f"{self._base}/{role_id}/menus", json={"menu_ids": menu_ids})
256
+ wrapper = APIWrapper[dict].model_validate(resp.json())
257
+ if wrapper.code != 0:
258
+ raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
259
+
260
+ def search(self, req: SearchRolesRequest) -> SearchRolesResponse:
261
+ resp = self._http.post(self._search, json=req.model_dump(by_alias=True, exclude_none=True))
262
+ wrapper = APIWrapper[SearchRolesResponse].model_validate(resp.json())
263
+ if wrapper.code != 0:
264
+ raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
265
+ return wrapper.data
266
+
267
+
268
+ class _User:
269
+ _base = "/api/v1/users"
270
+ _search = "/api/v1/search-users"
271
+
272
+ def __init__(self, http: httpx.Client):
273
+ self._http = http
274
+
275
+ def list(self, req: ListUsersRequest) -> ListUsersResponse:
276
+ resp = self._http.get(self._base, params=req.model_dump(by_alias=True, exclude_none=True))
277
+ wrapper = APIWrapper[ListUsersResponse].model_validate(resp.json())
278
+ if wrapper.code != 0:
279
+ raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
280
+ return wrapper.data
281
+
282
+ def get(self, user_id: int) -> User:
283
+ resp = self._http.get(f"{self._base}/{user_id}")
284
+ wrapper = APIWrapper[User].model_validate(resp.json())
285
+ if wrapper.code != 0:
286
+ raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
287
+ return wrapper.data
288
+
289
+ def create(self, req: CreateUserRequest) -> int:
290
+ resp = self._http.post(self._base, json=req.model_dump(by_alias=True, exclude_none=True))
291
+ wrapper = APIWrapper[CreateUserResponse].model_validate(resp.json())
292
+ if wrapper.code != 0:
293
+ raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
294
+ return wrapper.data.id
295
+
296
+ def update(self, user_id: int, req: UpdateUserRequest) -> None:
297
+ resp = self._http.put(f"{self._base}/{user_id}", json=req.model_dump(by_alias=True, exclude_none=True))
298
+ wrapper = APIWrapper[dict].model_validate(resp.json())
299
+ if wrapper.code != 0:
300
+ raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
301
+
302
+ def delete(self, user_id: int) -> None:
303
+ resp = self._http.delete(f"{self._base}/{user_id}")
304
+ wrapper = APIWrapper[dict].model_validate(resp.json())
305
+ if wrapper.code != 0:
306
+ raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
307
+
308
+ def set_roles(self, user_id: int, req: SetUserRolesRequest) -> None:
309
+ resp = self._http.put(f"{self._base}/{user_id}/roles", json=req.model_dump(by_alias=True, exclude_none=True))
310
+ wrapper = APIWrapper[dict].model_validate(resp.json())
311
+ if wrapper.code != 0:
312
+ raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
313
+
314
+ def get_menus(self, user_id: int, parent_id: int | None = None, auth: str | None = None) -> List[TreeMenu]:
315
+ params = {}
316
+ if parent_id is not None:
317
+ params["parent_id"] = parent_id
318
+ if auth:
319
+ params["auth"] = auth
320
+
321
+ resp = self._http.get(f"{self._base}/{user_id}/menus", params=params)
322
+ wrapper = APIWrapper[GetUserMenusResponse].model_validate(resp.json())
323
+ if wrapper.code != 0:
324
+ raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
325
+ return wrapper.data.menus
326
+
327
+ def search(self, req: SearchUsersRequest) -> SearchUsersResponse:
328
+ resp = self._http.post(self._search, json=req.model_dump(by_alias=True, exclude_none=True))
329
+ wrapper = APIWrapper[SearchUsersResponse].model_validate(resp.json())
330
+ if wrapper.code != 0:
331
+ raise APIError(f"backend code {wrapper.code}: {wrapper.msg}")
332
+ return wrapper.data
333
+
334
+ def search_one(self, req: SearchUsersRequest) -> int:
335
+ resp = self.search(req)
336
+ for user in resp.data:
337
+ if user.nickname == req.nickname:
338
+ return user.id
339
+ raise APIError("no user found")