aswi2022bug-thugs-gitlab/server/sql_app/schemas.py @ aba60b70

from typing import List, Optional

from pydantic import BaseModel



class USBLogBase(BaseModel):
    timestamp: str
    status: str


class USBLogCreate(USBLogBase):
    pass


class USBLog(USBLogBase):
    id: int
    device_id: int

    class Config:
        orm_mode = True


class DeviceBase(BaseModel):
    vendor_id: int
    product_id: int
    serial_number: str


class DeviceCreate(DeviceBase):
    pass


class Device(DeviceBase):
    id: int
    logs: List[USBLog] = []

    class Config:
        orm_mode = True


class USBTempBase(BaseModel):
    username: str
    hostname: str
    timestamp: str
    device: DeviceBase
    status: str


class USBTempCreate(USBTempBase):
    pass


class USBTemp(USBTempBase):
    id: int
    device_id: int

    class Config:
        orm_mode = True