-
Notifications
You must be signed in to change notification settings - Fork 221
Expand file tree
/
Copy path_users.py
More file actions
82 lines (73 loc) · 3.07 KB
/
_users.py
File metadata and controls
82 lines (73 loc) · 3.07 KB
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
import json
from datetime import datetime
from typing import Any, List, Optional
from uuid import UUID
from pydantic import parse_obj_as
from pydantic.json import pydantic_encoder
from dstack._internal.core.models.users import (
GlobalRole,
User,
UsersInfoList,
UsersInfoListOrUsersList,
UserWithCreds,
)
from dstack._internal.server.schemas.users import (
CreateUserRequest,
GetUserRequest,
RefreshTokenRequest,
UpdateUserRequest,
)
from dstack.api.server._group import APIClientGroup
class UsersAPIClient(APIClientGroup):
def list(
self,
return_total_count: Optional[bool] = None,
name_pattern: Optional[str] = None,
prev_created_at: Optional[datetime] = None,
prev_id: Optional[UUID] = None,
limit: Optional[int] = None,
ascending: Optional[bool] = None,
) -> UsersInfoListOrUsersList:
# Passing only non-None fields for backward compatibility with 0.20 servers.
body: dict[str, Any] = {}
if return_total_count is not None:
body["return_total_count"] = return_total_count
if name_pattern is not None:
body["name_pattern"] = name_pattern
if prev_created_at is not None:
body["prev_created_at"] = prev_created_at
if prev_id is not None:
body["prev_id"] = prev_id
if limit is not None:
body["limit"] = limit
if ascending is not None:
body["ascending"] = ascending
if body:
resp = self._request(
"/api/users/list", body=json.dumps(body, default=pydantic_encoder)
)
else:
resp = self._request("/api/users/list")
resp_json = resp.json()
if isinstance(resp_json, list):
return parse_obj_as(List[User.__response__], resp_json)
return parse_obj_as(UsersInfoList, resp_json)
def get_my_user(self) -> UserWithCreds:
resp = self._request("/api/users/get_my_user")
return parse_obj_as(UserWithCreds.__response__, resp.json())
def get_user(self, username: str) -> User:
body = GetUserRequest(username=username)
resp = self._request("/api/users/get_user", body=body.json())
return parse_obj_as(User.__response__, resp.json())
def create(self, username: str, global_role: GlobalRole) -> User:
body = CreateUserRequest(username=username, global_role=global_role, email=None)
resp = self._request("/api/users/create", body=body.json())
return parse_obj_as(User.__response__, resp.json())
def update(self, username: str, global_role: GlobalRole) -> User:
body = UpdateUserRequest(username=username, global_role=global_role, email=None)
resp = self._request("/api/users/update", body=body.json())
return parse_obj_as(User.__response__, resp.json())
def refresh_token(self, username: str) -> UserWithCreds:
body = RefreshTokenRequest(username=username)
resp = self._request("/api/users/refresh_token", body=body.json())
return parse_obj_as(UserWithCreds.__response__, resp.json())