Administrative act for treatment and treatment files

1 files changed, 13 insertions, 1 deletions

diff --git a/archaeological_operations/models.py b/archaeological_operations/models.py
index c54a060cb..307d02e97 100644
--- a/archaeological_operations/models.py
+++ b/archaeological_operations/models.py
@@ -882,8 +882,10 @@ class OperationSource(Source):
 class ActType(GeneralType):
     TYPE = (('F', _(u'Archaeological file')),
             ('O', _(u'Operation')),
+            ('TF', _(u'Treatment file')),
+            ('T', _(u'Treatment')),
             )
-    intented_to = models.CharField(_(u"Intended to"), max_length=1,
+    intented_to = models.CharField(_(u"Intended to"), max_length=2,
                                    choices=TYPE)
     code = models.CharField(_(u"Code"), max_length=10, blank=True, null=True)
     associated_template = models.ManyToManyField(
@@ -988,6 +990,16 @@ class AdministrativeAct(BaseHistorizedItem, OwnPerms, ValueGetter):
         blank=True, null=True,
         related_name='administrative_act',
         verbose_name=_(u"Archaeological file"))
+    treatment_file = models.ForeignKey(
+        'archaeological_finds.TreatmentFile',
+        blank=True, null=True,
+        related_name='administrative_act',
+        verbose_name=_(u"Treatment file"))
+    treatment = models.ForeignKey(
+        'archaeological_finds.Treatment',
+        blank=True, null=True,
+        related_name='administrative_act',
+        verbose_name=_(u"Treatment"))
     signature_date = models.DateField(_(u"Signature date"), blank=True,
                                       null=True)
     year = models.IntegerField(_(u"Year"), blank=True, null=True)