Templates: list all dependant documents

New template tags for displaying table of documents

1 files changed, 28 insertions, 1 deletions

diff --git a/archaeological_operations/models.py b/archaeological_operations/models.py
index c805f9a3d..3d6eaa15a 100644
--- a/archaeological_operations/models.py
+++ b/archaeological_operations/models.py
@@ -22,6 +22,7 @@ from itertools import groupby
 
 from django.conf import settings
 from django.contrib.gis.db import models
+from django.core.urlresolvers import reverse
 from django.db.models import Q, Count, Sum, Max, Avg
 from django.db.models.signals import post_save, m2m_changed
 from django.utils.translation import ugettext_lazy as _, ugettext
@@ -157,6 +158,16 @@ class Operation(BaseHistorizedItem, OwnPerms):
         return _(u"OPE")
 
     @property
+    def short_label(self):
+        if settings.COUNTRY == 'fr':
+            return self.code_patriarche
+        return unicode(self)
+
+    @property
+    def show_url(self):
+        return reverse('show-operation', args=[self.pk, ''])
+
+    @property
     def reference(self):
         if self.code_patriarche:
             return unicode(self.code_patriarche)
@@ -187,7 +198,19 @@ class Operation(BaseHistorizedItem, OwnPerms):
     def grouped_parcels(self):
         return Parcel.grouped_parcels(list(self.parcels.all()))
 
-    associated_file_short_label_lbl = _(u"Archaelogical file")
+    def context_record_docs_q(self):
+        from archaeological_context_records.models import ContextRecordSource
+        return ContextRecordSource.objects.filter(
+                                               context_record__operation=self)
+
+    def find_docs_q(self):
+        from archaeological_finds.models import FindSource
+        print FindSource.objects.filter(
+                             find__base_finds__context_record__operation=self).query
+        return FindSource.objects.filter(
+                             find__base_finds__context_record__operation=self)
+
+    associated_file_short_label_lbl = _(u"Archaeological file")
     @property
     def associated_file_short_label(self):
         if not self.associated_file:
@@ -309,6 +332,10 @@ class OperationSource(Source):
     TABLE_COLS = ['operation.year', 'operation.operation_code'] + \
                  Source.TABLE_COLS
 
+    @property
+    def owner(self):
+        return self.operation
+
 class ActType(GeneralType):
     TYPE = (('F', _(u'Archaelogical file')),
             ('O', _(u'Operation')),