API/schemas.py at main · PharmaML/API · 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
from pydantic import BaseModel
from typing import Optional

# Define Pydantic models for data validation
class UserCreate(BaseModel):
    username: str
    password: str
    center: int
    permission: int
    employeeid: int
    islocked: bool

class GetUser(BaseModel):
    username: Optional[str] = None
    id: Optional[int] = None

class DeleteUser(BaseModel):
    id: int

class Login(BaseModel):
    username: str
    password: str

class SessionCookie(BaseModel):
    session_cookie: str

class DrugReplacements(BaseModel):
    session_cookie: str
    drugid: int
    isMultiple: bool
    w1: float
    w2: float
    w3: float

class Favorite(BaseModel):
    session_cookie: str
    history_id: int

class UploadSnapshot(BaseModel):
    session_cookie: str
    file: str