from flask_restful_swagger_2 import Schema


class AnnotationModel(Schema):
    type = "object"
    description = "Filter by annotation key: value"
    properties = {
        "name": {
            "type": "string"
        },
        # TODO update to OpenAPI v3.0 when a library is available that supports it
        # Unfortunately 2.0 doesn't have a way to have a schema that accepts multiple types
        # Overloading the type key with a list seems to work ok and makes it to the page
        "values": {
            "type": "array",
            "items": {
                "type": ["float32", "string", "int32", "bool"]
            }
        },
        "min": {
            "type": ["int32", "float32"],
        },
        "max": {
            "type": ["int32", "float32"],
        }
    }
    required = ["name"]


class IndexModel(Schema):
    type = "object"
    description = "Filter by index of observation/variable ex. [0, 5, 15]"
    properties = {
        "index": {
            "type": "array",
            "items": {
                "format": "int32",
                "type": "integer"
            }

        }
    }


class AxisModel(Schema):
    type = "object"
    description = "Axis of data -- obs or var"
    properties = {
        "index": IndexModel,
        "annotation_value": AnnotationModel.array()
    }


class FilterModel(Schema):
    type = "object"
    description = "Complex filter"
    properties = {
        "filter": {
            "type": "object",
            "properties": {
                "obs": AxisModel,
                "var": AxisModel
            }
        }
    }