lodel2_mirror/lodel/leapi/datahandlers/reference.py

# -*- coding: utf-8 -*-
from lodel.leapi.datahandlers.field_data_handler import FieldDataHandler
from lodel.editorial_model.components import EmClass


class Reference(FieldDataHandler):

    ## @brief Instanciation
    # @param allowed_classes list | None : list of allowed em classes if None no restriction
    # @param internal bool : if False, the field is not internal
    # @param **kwargs : other arguments
    def __init__(self, allowed_classes = None, internal=False, **kwargs):
        self.allowed_classes = None if allowed_classes is None else set(allowed_classes)
        super().__init__(internal=internal, **kwargs)

    ## @brief gets the target of the reference
    def get_relateds(self):
        return self._refs

    ## @brief checks if the target is valid

    def check_data_value(self, value):

        if not isinstance(value, self._refs_class):
            return (value, "The reference should be an instance of %s, %s gotten" % (self._refs_class, value.__class__))

        if isinstance(value, EmClass):
            value = [value]

        if isinstance(value, dict):
            ref_values = value.values()

        for related in value:
            if not isinstance(related, EmClass):
                return (value, "The reference %s should be an instance of EmClass, %s gotten" % (related.display_name, related.__class__))

        return (value, None)