summaryrefslogtreecommitdiff
path: root/archaeological_finds/models_finds.py
diff options
context:
space:
mode:
Diffstat (limited to 'archaeological_finds/models_finds.py')
-rw-r--r--archaeological_finds/models_finds.py32
1 files changed, 23 insertions, 9 deletions
diff --git a/archaeological_finds/models_finds.py b/archaeological_finds/models_finds.py
index 3d379d2a7..5f4183f0c 100644
--- a/archaeological_finds/models_finds.py
+++ b/archaeological_finds/models_finds.py
@@ -69,15 +69,28 @@ post_save.connect(post_save_cache, sender=ConservatoryState)
post_delete.connect(post_save_cache, sender=ConservatoryState)
-class PreservationType(GeneralType):
+class TreatmentType(GeneralType):
+ order = models.IntegerField(_(u"Order"), default=10)
+ parent = models.ForeignKey("TreatmentType", verbose_name=_(u"Parent type"),
+ blank=True, null=True)
+ virtual = models.BooleanField(_(u"Virtual"))
+ upstream_is_many = models.BooleanField(
+ _(u"Upstream is many"), default=False,
+ help_text=_(
+ u"Check this if for this treatment from many finds you'll get "
+ u"one."))
+ downstream_is_many = models.BooleanField(
+ _(u"Downstream is many"), default=False,
+ help_text=_(
+ u"Check this if for this treatment from one find you'll get "
+ u"many."))
+
class Meta:
- verbose_name = _(u"Preservation type")
- verbose_name_plural = _(u"Preservation types")
+ verbose_name = _(u"Treatment type")
+ verbose_name_plural = _(u"Treatment types")
ordering = ('label',)
-
-
-post_save.connect(post_save_cache, sender=PreservationType)
-post_delete.connect(post_save_cache, sender=PreservationType)
+post_save.connect(post_save_cache, sender=TreatmentType)
+post_delete.connect(post_save_cache, sender=TreatmentType)
class IntegrityType(GeneralType):
@@ -714,8 +727,9 @@ class Find(BulkUpdatedItem, ValueGetter, BaseHistorizedItem, ImageModel,
conservatory_comment = models.TextField(_(u"Conservatory comment"),
blank=True, null=True)
preservation_to_considers = models.ManyToManyField(
- PreservationType, verbose_name=_(u"Type of preservation to consider"),
- related_name='finds', blank=True)
+ TreatmentType,
+ verbose_name=_(u"Recommended treatments"),
+ related_name='finds_recommended', blank=True)
alterations = models.ManyToManyField(
AlterationType, verbose_name=_(u"Alteration"), blank=True,
related_name='finds'