lodel2_mirror/EditorialModel/fields.py

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

import importlib

from EditorialModel.components import EmComponent
from EditorialModel.exceptions import EmComponentCheckError
import EditorialModel
from django.db import models

## EmField (Class)
#
# Represents one data for a lodel2 document
class EmField(EmComponent):

    ranked_in = 'fieldgroup_id'

    ftype = None

    fieldtypes = {
        'int': models.IntegerField,
        'integer': models.IntegerField,
        'bigint': models.BigIntegerField,
        'smallint': models.SmallIntegerField,
        'boolean': models.BooleanField,
        'bool': models.BooleanField,
        'float': models.FloatField,
        'char': models.CharField,
        'varchar': models.CharField,
        'text': models.TextField,
        'time': models.TimeField,
        'date': models.DateField,
        'datetime': models.DateTimeField,
    }

    ## Instanciate a new EmField
    # @todo define and test type for icon and fieldtype
    # @warning nullable == True by default
    def __init__(self, model, uid, name, fieldgroup_id, optional=False, internal=False, rel_field_id=None, icon='0', string=None, help_text=None, date_update=None, date_create=None, rank=None, nullable = True, default = None, **kwargs):

        self.fieldgroup_id = fieldgroup_id
        self.check_type('fieldgroup_id', int)
        self.optional = optional
        self.check_type('optional', bool)
        self.internal = internal
        self.check_type('internal', bool)
        self.rel_field_id = rel_field_id
        self.check_type('rel_field_id', (int, type(None)))
        self.icon = icon

        self.nullable = nullable
        self.default = default

        self.options = kwargs

        super(EmField, self).__init__(model=model, uid=uid, name=name, string=string, help_text=help_text, date_update=date_update, date_create=date_create, rank=rank)

    @staticmethod
    def get_field_class(ftype, **kwargs):
        ftype_module = importlib.import_module('EditorialModel.fieldtypes.%s'%ftype)
        return ftype_module.fclass

    ## @brief Abstract method that should return a validation function
    # @param raise_e Exception : if not valid raise this exception
    # @param ret_valid : if valid return this value
    # @param ret_invalid : if not valid return this value
    def validation_function(self, raise_e = None, ret_valid = None, ret_invalid = None):
        if self.__class__ == EmField:
            raise NotImplementedError("Abstract method")
        if raise_e is None and ret_valid is None:
            raise AttributeError("Behavior doesn't allows to return a valid validation function")

        return False
            

    ## @brief Return the list of relation fields for a rel_to_type
    # @return None if the field is not a rel_to_type else return a list of EmField
    def rel_to_type_fields(self):
        if not self.rel_to_type_id:
            return None
        
        return [ f for f in self.model.components(EmField) if f.rel_field_id == self.uid ]

    ## Check if the EmField is valid
    # @return True if valid False if not
    def check(self):
        super(EmField, self).check()
        em_fieldgroup = self.model.component(self.fieldgroup_id)
        if not em_fieldgroup:
            raise EmComponentCheckError("fieldgroup_id contains a non existing uid : '%d'" % self.fieldgroup_id)
        if not isinstance(em_fieldgroup, EditorialModel.fieldgroups.EmFieldGroup):
            raise EmComponentCheckError("fieldgroup_id contains an uid from a component that is not an EmFieldGroup but a %s" % str(type(em_fieldgroup)))

    ## @brief Delete a field if it's not linked
    # @return bool : True if deleted False if deletion aborded
    # @todo Check if unconditionnal deletion is correct
    def delete_check(self):
        return True

    """
    def to_django(self):
        if self.fieldtype in ('varchar', 'char'):
            max_length = None if 'max_length' not in self.options else self.options['max_length']
            return self.fieldtypes[self.fieldtype](max_length=max_length, **self.options)

        if self.fieldtype in ('time', 'datetime', 'date'):
            auto_now = False if 'auto_now' not in self.options else self.options['auto_now']
            auto_now_add = False if 'auto_now_add' not in self.options else self.options['auto_now_add']
            return self.fieldtypes[self.fieldtype](auto_now=auto_now, auto_now_add=auto_now_add, **self.options)

        if self.fieldtype == 'boolean' and ('nullable' in self.options and self.options['nullable'] == 1):
            return models.NullBooleanField(**self.options)

        return self.fieldtypes[self.fieldtype](**self.options)
    """