Source code for luna_handlers.classes.schemas.simple_schemas

"""
Module contains simple schemas for multiple usage
"""

from pydantic import conlist

from classes.schemas import types
from classes.schemas.base_schema import BaseSchema
from classes.schemas.types import OptionalNotNullable


class BoundingBoxSchema(BaseSchema):
    """Bounding box schema"""

    # bounding box x coordinate
    x: types.SDKInt
    # bounding box y coordinate
    y: types.SDKInt
    # bounding box width
    width: types.SDKInt
    # bounding box height
    height: types.SDKInt


class UrlWithFaceBbox(BaseSchema):
    """Url with face bounding boxes schema"""

    # url
    url: str
    # face bounding box list
    faceBoundingBoxes: conlist(BoundingBoxSchema, min_items=1) = OptionalNotNullable()


class UrlBboxSchema(UrlWithFaceBbox):
    """Url with face and body bounding boxes schema"""

    # body bounding box list
    bodyBoundingBoxes: conlist(BoundingBoxSchema, min_items=1) = OptionalNotNullable()