Source code for luna_handlers.classes.schemas.extract_policies

"""
Module contains schemas for extract policy
"""
from classes.schemas import types
from classes.schemas.base_schema import BaseSchema


class BaseExtractPolicy(BaseSchema):
    """Base extract policy"""

    # whether to extract basic attributes
    extractBasicAttributes: types.Int01 = 0
    # descriptor quality score threshold
    fdScoreThreshold: types.StrictFloat01 = 0.0


class HandlerExtractPolicy(BaseExtractPolicy):
    """Handler extract policy schema"""

    # whether to extract face descriptor
    extractFaceDescriptor: types.Int01 = 1
    # whether to extract body descriptor
    extractBodyDescriptor: types.Int01 = 0