diff options
Diffstat (limited to 'ishtar_common')
| -rw-r--r-- | ishtar_common/templatetags/window_tables.py | 7 | 
1 files changed, 6 insertions, 1 deletions
| diff --git a/ishtar_common/templatetags/window_tables.py b/ishtar_common/templatetags/window_tables.py index 05592557a..960bc7184 100644 --- a/ishtar_common/templatetags/window_tables.py +++ b/ishtar_common/templatetags/window_tables.py @@ -18,7 +18,8 @@ from archaeological_context_records.models import ContextRecord, \      ContextRecordSource, RecordRelationView, \      RecordRelations as CRRecordRelations  from archaeological_finds.models import Find, FindSource, \ -    FindUpstreamTreatments, FindDownstreamTreatments, FindTreatments +    FindUpstreamTreatments, FindDownstreamTreatments, FindTreatments, \ +    TreatmentSource, TreatmentFileSource  from archaeological_warehouse.models import Container  register = template.Library() @@ -59,6 +60,10 @@ ASSOCIATED_MODELS['finds_downstreamtreatments'] = (      FindDownstreamTreatments, 'get-downstreamtreatment', '')  ASSOCIATED_MODELS['treatments'] = (      FindTreatments, 'get-treatment', '') +ASSOCIATED_MODELS['treatments_docs'] = ( +    TreatmentSource, 'get-treatmentsource', '') +ASSOCIATED_MODELS['treatmentfiles_docs'] = ( +    TreatmentFileSource, 'get-treatmentfilesource', '')  ASSOCIATED_MODELS['containers'] = (      Container, 'get-container', '') | 
