Example #1
0
class BISearchSchema(OneOfSchema):
    type_field = "type"
    type_field_remove = False
    type_schemas = dict((k, v.schema()) for k, v in bi_search_registry.items())

    def get_obj_type(self, obj: ABCBISearch) -> str:
        return obj.type()
Example #2
0
class BISearchSchema(OneOfSchema):
    type_field = "type"
    type_field_remove = False
    type_schemas = dict((k, v.schema()) for k, v in bi_search_registry.items())

    def get_obj_type(self, obj: Union[ABCBISearch, dict]) -> str:
        if isinstance(obj, dict):
            return obj["type"]
        return obj.type()