46 lines
887 B
Python
46 lines
887 B
Python
from pydantic import BaseModel
|
|
from typing import List, Optional
|
|
from datetime import datetime
|
|
from .models import UserRole, JobStatus
|
|
|
|
class UserBase(BaseModel):
|
|
username: str
|
|
|
|
class UserCreate(UserBase):
|
|
password: str
|
|
role: UserRole = UserRole.user
|
|
|
|
class User(UserBase):
|
|
id: int
|
|
role: UserRole
|
|
created_at: datetime
|
|
class Config:
|
|
orm_mode = True
|
|
|
|
class JobBase(BaseModel):
|
|
branch_name: str
|
|
scenarios: List[str]
|
|
environment: str
|
|
test_mode: str
|
|
|
|
class JobCreate(JobBase):
|
|
pass
|
|
|
|
class Job(JobBase):
|
|
id: int
|
|
user_id: int
|
|
status: JobStatus
|
|
result_path: Optional[str]
|
|
duration: Optional[str]
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
class Config:
|
|
orm_mode = True
|
|
|
|
class Token(BaseModel):
|
|
access_token: str
|
|
token_type: str
|
|
|
|
class TokenData(BaseModel):
|
|
username: Optional[str] = None
|