Adding fields to context records (closes #375)

1 files changed, 23 insertions, 0 deletions

diff --git a/ishtar/furnitures/models.py b/ishtar/furnitures/models.py
index f9e9080df..44e0e9391 100644
--- a/ishtar/furnitures/models.py
+++ b/ishtar/furnitures/models.py
@@ -681,6 +681,25 @@ class Unit(GeneralType):
     def __unicode__(self):
         return self.label
 
+class ActivityType(GeneralType):
+    order = models.IntegerField(_(u"Order"))
+
+    class Meta:
+        verbose_name = _(u"Type Activity")
+        verbose_name_plural = _(u"Types Activity")
+
+    def __unicode__(self):
+        return self.label
+
+class IdentificationType(GeneralType):
+    order = models.IntegerField(_(u"Order"))
+    class Meta:
+        verbose_name = _(u"Type Identification")
+        verbose_name_plural = _(u"Types Identification")
+
+    def __unicode__(self):
+        return self.label
+
 class ContextRecord(BaseHistorizedItem, OwnPerms):
     TABLE_COLS = ['parcel.town', 'parcel.operation.year',
                   'parcel.operation.operation_code',
@@ -716,6 +735,10 @@ class ContextRecord(BaseHistorizedItem, OwnPerms):
                  " created before this date"))
     tpq_estimated = models.IntegerField(_(u"Estimated TPQ"), blank=True,
      null=True, help_text=_("Estimation of a \"Terminus Post Quem\""))
+    identification = models.ForeignKey(IdentificationType, blank=True,
+                            null=True, verbose_name=_(u"Identification"),)
+    activity = models.ForeignKey(ActivityType,blank=True, null=True,
+                                 verbose_name=_(u"Activity"),)
     history = HistoricalRecords()
 
     class Meta: