pillar/attract/application/modules/nodes/forms.py

from flask_wtf import Form
from wtforms import TextField
from wtforms import BooleanField
from wtforms import SelectField
from wtforms import TextAreaField
from wtforms import IntegerField
from wtforms import HiddenField
from wtforms import FieldList
from wtforms import FormField
from wtforms import Form as BasicForm

from application.modules.nodes.models import CustomFields
from wtforms.validators import DataRequired

from application import db

from application.modules.nodes.models import Node, NodeType, NodeProperties

class CustomFieldForm(BasicForm):
    id = HiddenField()
    field_type = TextField('Field Type', validators=[DataRequired()])
    name = TextField('Name', validators=[DataRequired()])
    name_url = TextField('Url', validators=[DataRequired()])
    description = TextAreaField('Description')
    is_required = BooleanField('Is extended')
    def __init__(self, csrf_enabled=False, *args, **kwargs):
        super(CustomFieldForm, self).__init__(csrf_enabled=False, *args, **kwargs)

class ModelFieldList(FieldList):
    def __init__(self, *args, **kwargs):         
        self.model = kwargs.pop("model", None)
        super(ModelFieldList, self).__init__(*args, **kwargs)
        if not self.model:
            raise ValueError("ModelFieldList requires model to be set")
 
    def populate_obj(self, obj, name):
        while len(getattr(obj, name)) < len(self.entries):
            newModel = self.model()
            db.session.add(newModel)
            getattr(obj, name).append(newModel)
        while len(getattr(obj, name)) > len(self.entries):
            db.session.delete(getattr(obj, name).pop())
        super(ModelFieldList, self).populate_obj(obj, name)

class ChildInline(Form):
    title = TextField('Title',)

class NodeTypeForm(Form):
    name = TextField('Name', validators=[DataRequired()])
    url = TextField('Url', validators=[DataRequired()])
    description = TextAreaField('Description', validators=[DataRequired()])
    is_extended = BooleanField('Is extended')
    custom_fields = ModelFieldList(FormField(CustomFieldForm), model=CustomFields)


class IMForm(Form):
    protocol = SelectField(choices=[('aim', 'AIM'), ('msn', 'MSN')])
    username = TextField()

class ContactForm(Form):
    first_name  = TextField()
    last_name   = TextField()
    im_accounts = FieldList(BooleanField('Is extended'),)


def get_node_form(node_type):
    node_type = NodeType.query.filter_by(url=node_type).first()
    class ProceduralForm(Form):
        pass

    setattr(ProceduralForm,
        'name',
        TextField('Name', validators=[DataRequired()]))
    setattr(ProceduralForm,
        'url',
        TextField('Url'))
    setattr(ProceduralForm,
        'description',
        TextAreaField('Description', validators=[DataRequired()]))
    setattr(ProceduralForm,
        'node_type_id',
        HiddenField(default=node_type.id))

    for custom_field in CustomFields.query\
        .join(NodeType)\
        .filter(NodeType.url == node_type.url):
        
        if custom_field.field_type == 'text':
            field_properties = TextAreaField(custom_field.name, 
                validators=[DataRequired()])
        elif custom_field.field_type == 'string':
            field_properties = TextField(custom_field.name, 
                validators=[DataRequired()])
        elif custom_field.field_type == 'integer':
            field_properties = IntegerField(custom_field.name, 
                validators=[DataRequired()])
        elif custom_field.field_type == 'select':
            options = Node.query\
                .join(NodeType)\
                .filter(NodeType.url==custom_field.name_url)\
                .all()
            field_properties = SelectField(custom_field.name, 
                coerce=int,
                choices=[(option.id, option.name) for option in options] )
        
        setattr(ProceduralForm, custom_field.name_url, field_properties)

    return ProceduralForm()


def process_node_form(form, node_id=None):
    """Generic function used to process new nodes, as well as edits
    """
    if form.validate_on_submit():
        node_type = NodeType.query.get(form.node_type_id.data)
        if node_id:
            node = Node.query.get(node_id)
            node.name = form.name.data
            node.description = form.description.data
        else:
            node = Node(
                name=form.name.data,
                description=form.description.data,
                node_type_id=form.node_type_id.data)
            db.session.add(node)
        db.session.commit()

        for custom_field in CustomFields.query\
            .join(NodeType)\
            .filter(NodeType.url == node_type.url):

            for field in form:
                if field.name == custom_field.name_url:
                    if node_id:
                        # Query for the indivitual property
                        # TODO: collect all properties and loop through them
                        node_property = NodeProperties.query\
                            .filter_by(node_id=node_id)\
                            .filter_by(custom_field_id=custom_field.id)\
                            .first()
                        if node_property:
                            # Update the value of the property
                            node_property.value = field.data
                        else:
                            # If the property is missing we add it
                            node_property = NodeProperties(
                                node_id=node.id,
                                custom_field_id=custom_field.id,
                                value=field.data)
                            db.session.add(node_property)
                    else:
                        node_property = NodeProperties(
                            node_id=node.id,
                            custom_field_id=custom_field.id,
                            value=field.data)
                        db.session.add(node_property)
                    db.session.commit()
        return True
    else:
        return False