mptt.fields: 22 total statements, 0.0% covered

Generated: Wed 2013-03-13 10:33 CET

Source file: /media/Envs/Envs/filer-gallery/lib/python2.7/site-packages/mptt/fields.py

Stats: 0 executed, 19 missed, 3 excluded, 24 ignored

1. """
2. Model fields for working with trees.
3. """
5. __all__ = ('TreeForeignKey', 'TreeOneToOneField', 'TreeManyToManyField')
7. from django.db import models
8. from mptt.forms import TreeNodeChoiceField, TreeNodeMultipleChoiceField
11. class TreeForeignKey(models.ForeignKey):
12. """
13. Extends the foreign key, but uses mptt's ``TreeNodeChoiceField`` as
14. the default form field.
16. This is useful if you are creating models that need automatically
17. generated ModelForms to use the correct widgets.
18. """
20. def formfield(self, **kwargs):
21. """
22. Use MPTT's ``TreeNodeChoiceField``
23. """
24. kwargs.setdefault('form_class', TreeNodeChoiceField)
25. return super(TreeForeignKey, self).formfield(**kwargs)
28. class TreeOneToOneField(models.OneToOneField):
29. def formfield(self, **kwargs):
30. kwargs.setdefault('form_class', TreeNodeChoiceField)
31. return super(TreeOneToOneField, self).formfield(**kwargs)
34. class TreeManyToManyField(models.ManyToManyField):
35. def formfield(self, **kwargs):
36. kwargs.setdefault('form_class', TreeNodeMultipleChoiceField)
37. return super(TreeManyToManyField, self).formfield(**kwargs)
39. # South integration
40. try:
41. from south.modelsinspector import add_introspection_rules
42. add_introspection_rules([], ["^mptt\.fields\.TreeForeignKey"])
43. add_introspection_rules([], ["^mptt\.fields\.TreeOneToOneField"])
44. add_introspection_rules([], ["^mptt\.fields\.TreeManyToManyField"])
45. except ImportError:
46. pass