lodel2_mirror/EditorialModel/fieldtypes/leo.py

#-*- coding: utf-8 -*-

import leapi.letype as letype
import leapi.leclass as leclass

from .generic import ReferenceFieldType, FieldTypeError

class EmFieldType(ReferenceFieldType):
    
    help = 'Fieldtypes designed to handle pk of LeObject in LeRelations'

    _construct_data_deps = []
    
    ## @todo Replace hardcoded string for reference initialisation
    def __init__(self, superior=True, **kwargs):
        super().__init__(superior = superior, reference='object', **kwargs)

    def _check_data_value(self, value):
        if not isinstance(value, int):
            if not letype._LeType.implements_leobject():
                return (None, ValueError("An instance of a child class of LeType was expected"))
            if not hasattr(value, 'lodel_id'):
                return (None, ValueError("The LeType instance given has no lodel_id !"))
        return (value, None)
    
    ## @brief If field value is an integer, returns a partially instanciated LeObject (only with an ID)
    # @todo what should we do if the get fails ? Raise ?
    def construct_data(self, lec, fname, datas, cur_value):
        if isinstance(cur_value, str):
            # Cast to int
            try:
                cur_value = int(cur_value)
            except (ValueError, TypeError) as e:
                raise e # Raise Here !?
        if hasattr(cur_value, 'is_leobject') and cur_value.is_leobject():
            # Its an object not populated (we dont now its type)
            cur_value = cur_value.lodel_id #Optimize here giving only class_id and type_id to populate ?
        if isinstance(cur_value, int):
            # Get instance with id
            resget = lec.name2class('LeObject').get(['lodel_id = %d' % cur_value])
            if resget is None or len(resget) != 1:
                # Bad filter or bad id... raise ?
                raise Exception("BAAAAAD")
        return cur_value
    
    ## @brief checks datas consistency
    # @param lec LeCrud : A LeCrud child instance
    # @param fname str : concerned field name
    # @param datas dict : Datas of lec
    # @return True if ok else an Exception instance
    def check_data_consistency(self, lec, fname, datas):
        if self.superior:
            return self.check_sup_consistency(lec, fname, datas)
        else:
            return self.check_sub_consistency(lec, fname, datas)

    def check_sup_consistency(self, lec, fname, datas):
        if lec.implements_lerel2type():
            # Checking consistency for a rel2type relation
            lesup = datas[lec._superior_field_name]
            lesub = datas[lec._subordinate_field_name]
            if lesub.__class__ not in lesup._linked_types.values():
                return FieldTypeError("Rel2type not authorized between %s and %s"%(lesup, lesub))
        pass
            

    def check_sub_consistency(self, lec, fname, datas):
        pass