60 lines
859 B
Python
60 lines
859 B
Python
# schemas.py
|
|
from pydantic import BaseModel
|
|
from typing import Optional
|
|
from core.enums import ConnectionTypes, UserRole
|
|
|
|
|
|
class ConnectionBase(BaseModel):
|
|
db_name: str
|
|
type: ConnectionTypes
|
|
host: str
|
|
port: int
|
|
username: str
|
|
password: str
|
|
|
|
|
|
class ConnectionCreate(ConnectionBase):
|
|
pass
|
|
|
|
|
|
class ConnectionUpdate(ConnectionBase):
|
|
pass
|
|
|
|
|
|
class ConnectionInDBBase(ConnectionBase):
|
|
id: int
|
|
owner_id: int
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class Connection(ConnectionInDBBase):
|
|
pass
|
|
|
|
|
|
class UserBase(BaseModel):
|
|
username: str
|
|
|
|
|
|
class UserCreate(UserBase):
|
|
role: UserRole
|
|
|
|
|
|
class UserOut(UserBase):
|
|
id: int
|
|
role: UserRole
|
|
|
|
|
|
class UserInDBBase(UserBase):
|
|
id: int
|
|
role: UserRole
|
|
api_key: str
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class User(UserInDBBase):
|
|
pass
|