openapi-core/openapi_core/unmarshalling/schemas/util.py

"""OpenAPI core schemas util module"""
from base64 import b64decode
from copy import copy
import datetime
from distutils.util import strtobool
from six import string_types, text_type, integer_types
from uuid import UUID

from openapi_schema_validator import oas30_format_checker

from openapi_core.compat import lru_cache


def forcebool(val):
    if isinstance(val, string_types):
        val = strtobool(val)

    return bool(val)


def format_date(value):
    return datetime.datetime.strptime(value, '%Y-%m-%d').date()


def format_uuid(value):
    if isinstance(value, UUID):
        return value
    return UUID(value)


def format_byte(value, encoding='utf8'):
    return text_type(b64decode(value), encoding)


def format_number(value):
    if isinstance(value, integer_types + (float, )):
        return value

    return float(value)


@lru_cache()
def build_format_checker(**custom_formatters):
    if not custom_formatters:
        return oas30_format_checker

    fc = copy(oas30_format_checker)
    for name, formatter in custom_formatters.items():
        fc.checks(name)(formatter.validate)
    return fc