From 40bb871af86c913ffaa4235052424cc0d69211a0 Mon Sep 17 00:00:00 2001 From: Filipp Lepalaan Date: Fri, 25 Sep 2015 20:01:46 +0300 Subject: Added type field to note --- servo/models/note.py | 27 ++++++++++++++++++++++++--- 1 file changed, 24 insertions(+), 3 deletions(-) (limited to 'servo/models/note.py') diff --git a/servo/models/note.py b/servo/models/note.py index 0981530..da9e08a 100644 --- a/servo/models/note.py +++ b/servo/models/note.py @@ -58,6 +58,11 @@ def validate_phone_number(number): class Note(MPTTModel): + + T_NOTE = 0 + T_PROBLEM = 1 + T_ESCALATION = 2 + subject = models.CharField( blank=True, max_length=255, @@ -102,16 +107,32 @@ class Note(MPTTModel): sent_at = models.DateTimeField(null=True, editable=False) order = models.ForeignKey(Order, null=True, blank=True) - is_reported = models.BooleanField(default=False, verbose_name=_("report")) + is_reported = models.BooleanField( + default=False, + verbose_name=_("Report") + ) is_read = models.BooleanField( default=True, editable=False, - verbose_name=_("read") + verbose_name=_("Read") ) is_flagged = models.BooleanField( default=False, editable=False, - verbose_name=_("flagged") + verbose_name=_("Flagged") + ) + + TYPES = ( + (T_NOTE, _('Note')), + (T_PROBLEM, _('Problem')), + (T_ESCALATION, _('Escalation')), + ) + + type = models.IntegerField( + blank=True, + default=T_NOTE, + choices=TYPES, + verbose_name=_('Type') ) objects = TreeManager() -- cgit v1.2.3