Gustavo Picon is sharing code with you
Bitbucket is a code hosting site. Unlimited public and private repositories. Free for small teams.
Don't show this againdjango-treebeard / treebeard / forms.py
- Tag
- 1.5
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 | # -*- coding: utf-8 -*-
"""
treebeard.forms
---------------
Forms for treebeard.
:copyright: 2008-2009 by Gustavo Picon
:license: Apache License 2.0
Original contribution by aleh.fl
"""
from django.forms.models import model_to_dict, ErrorList, BaseModelForm
from django import forms
class MoveNodeForm(forms.ModelForm):
"""Form to handle moving a node in a tree.
Handles sorted/unsorted trees.
Read the `Django Form objects documentation`_ for reference.
.. _`Django Form objects documentation`:
http://docs.djangoproject.com/en/dev/topics/forms/#form-objects
"""
__position_choices_sorted = (
('sorted-child', 'Child of'),
('sorted-sibling', 'Sibling of'),
)
__position_choices_unsorted = (
('first-child', 'First child of'),
('left', 'Before'),
('right', 'After'),
)
_position = forms.ChoiceField(required=True, label="Position")
_ref_node_id = forms.TypedChoiceField(required=False,
coerce=int,
label="Relative to")
class Meta:
exclude = ('path',
'depth',
'numchild',
'lft',
'rgt',
'tree_id',
'parent',
'sib_order')
def __init__(self, data=None, files=None, auto_id='id_%s', prefix=None,
initial=None, error_class=ErrorList, label_suffix=':',
empty_permitted=False, instance=None):
opts = self._meta
if instance:
opts.model = type(instance)
self.is_sorted = (hasattr(opts.model, 'node_order_by') and
opts.model.node_order_by)
#self.is_sorted = (len(opts.model.node_order_by) > 0)
if self.is_sorted:
self.declared_fields['_position'].choices = \
self.__class__.__position_choices_sorted
else:
self.declared_fields['_position'].choices = \
self.__class__.__position_choices_unsorted
def mk_dropdown_tree(for_node=None):
""" Creates a tree-like list of choices """
is_loop_safe = lambda(possible_parent): True
# Do actual check only if for_node is provided
if for_node is not None:
is_loop_safe = lambda(possible_parent): not (\
possible_parent == for_node) or \
possible_parent.is_descendant_of(for_node)
mk_indent = lambda(level): '. . ' * (level - 1)
def add_subtree(node, options):
""" Recursively build options tree. """
if is_loop_safe(node):
options.append(
(node.pk, mk_indent(node.get_depth())+str(node)))
for subnode in node.get_children():
add_subtree(subnode, options)
options = [(0, '-- root --')]
for node in opts.model.get_root_nodes():
add_subtree(node, options)
return options
if instance is None:
# if we didn't get an instance, instantiate a new one
self.instance = opts.model()
object_data = {}
self.declared_fields['_ref_node_id'].choices = mk_dropdown_tree()
else:
object_data = model_to_dict(instance, opts.fields, opts.exclude)
if self.is_sorted:
node_parent = instance.get_parent()
if node_parent is None:
object_data.update({'_ref_node_id': '',
'_position': 'sorted-child',
})
else:
object_data.update({'_ref_node_id': node_parent.pk,
'_position': 'sorted-child',
})
else:
prev_sibling = instance.get_prev_sibling()
if prev_sibling is None:
if(instance.is_root()):
object_data.update({'_ref_node_id': '',
'_position': 'first-child',
})
else:
object_data.update(
{'_ref_node_id': instance.get_parent().pk,
'_position': 'first-child',
})
else:
object_data.update({'_ref_node_id': prev_sibling.pk,
'_position': 'right',
})
self.declared_fields['_ref_node_id'].choices = mk_dropdown_tree(
for_node=instance)
self.instance = instance
# if initial was provided, it should override the values from instance
if initial is not None:
object_data.update(initial)
super(BaseModelForm, self).__init__(data, files, auto_id, prefix,
object_data, error_class,
label_suffix, empty_permitted)
def save(self, commit=True):
reference_node_id = 0
if '_ref_node_id' in self.cleaned_data:
reference_node_id = self.cleaned_data['_ref_node_id']
position_type = self.cleaned_data['_position']
# delete auxilary fields not belonging to node model
del self.cleaned_data['_ref_node_id']
del self.cleaned_data['_position']
if self.instance.pk is None:
if reference_node_id:
reference_node = self.Meta.model.objects.get(
pk=reference_node_id)
self.instance = reference_node.add_child(** self.cleaned_data)
self.instance.move(reference_node, pos=position_type)
else:
self.instance = self.Meta.model.add_root(** self.cleaned_data)
else:
if reference_node_id:
reference_node = self.Meta.model.objects.get(
pk=reference_node_id)
self.instance.move(reference_node, pos=position_type)
else:
if self.is_sorted:
self.instance.move(self.Meta.model.get_first_root_node(),
pos='sorted-sibling')
else:
self.instance.move(self.Meta.model.get_first_root_node(),
pos='first-sibling')
# Reload the instance
self.instance = self.Meta.model.objects.get(pk=self.instance.pk)
super(MoveNodeForm, self).save(commit=commit)
return self.instance
|