Preservation module: add fields to models (refs #3639)

2 files changed, 38 insertions, 0 deletions

diff --git a/ishtar_common/migrations/0014_ishtarsiteprofile_preservation.py b/ishtar_common/migrations/0014_ishtarsiteprofile_preservation.py
new file mode 100644
index 000000000..90e6571ac
--- /dev/null
+++ b/ishtar_common/migrations/0014_ishtarsiteprofile_preservation.py
@@ -0,0 +1,20 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.11 on 2017-10-04 11:25
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('ishtar_common', '0013_importertype_available'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='ishtarsiteprofile',
+            name='preservation',
+            field=models.BooleanField(default=False, verbose_name='Preservation module'),
+        ),
+    ]
diff --git a/ishtar_common/models.py b/ishtar_common/models.py
index b0b050c6c..28a24115b 100644
--- a/ishtar_common/models.py
+++ b/ishtar_common/models.py
@@ -765,6 +765,22 @@ class GeneralType(Cached, models.Model):
             item.generate_key()
 
 
+class HierarchicalType(GeneralType):
+    parent = models.ForeignKey('self', blank=True, null=True,
+                               verbose_name=_(u"Parent"))
+
+    class Meta:
+        abstract = True
+
+    def full_label(self):
+        lbls = [self.label]
+        item = self
+        while item.parent:
+            item = item.parent
+            lbls.append(item.label)
+        return u" > ".join(reversed(lbls))
+
+
 class ItemKey(models.Model):
     key = models.CharField(_(u"Key"), max_length=100)
     content_type = models.ForeignKey(ContentType)
@@ -1255,6 +1271,8 @@ class IshtarSiteProfile(models.Model, Cached):
     warehouse_color = models.CharField(
         _(u"CSS code for warehouse module"), default=u'rgba(10,20,200,0.15)',
         max_length=200)
+    preservation = models.BooleanField(_(u"Preservation module"),
+                                       default=False)
     mapping = models.BooleanField(_(u"Mapping module"), default=False)
     mapping_color = models.CharField(
         _(u"CSS code for mapping module"), default=u'rgba(72, 236, 0, 0.15)',