diff options
Diffstat (limited to 'ishtar_common/models_imports.py')
| -rw-r--r-- | ishtar_common/models_imports.py | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/ishtar_common/models_imports.py b/ishtar_common/models_imports.py index 63a5b92fc..33d793333 100644 --- a/ishtar_common/models_imports.py +++ b/ishtar_common/models_imports.py @@ -144,6 +144,7 @@ class ImporterModel(models.Model): IMPORT_TYPES = ( ("tab", _("Table")), ("gis", _("GIS")), + ("qgs", _("QGIS")), ) IMPORT_TYPES_DICT = dict(IMPORT_TYPES) @@ -222,6 +223,14 @@ class ImporterType(models.Model): verbose_name = _("Importer - Type") verbose_name_plural = _("Importer - Types") ordering = ("name",) + # own is defined by users + # change permission need view permission + permissions = ( + ("view_gis_importer", "Can export to QGIS"), + ("view_own_gis_importer", "Can export own to QGIS"), + ("change_gis_importer", "Can import from QGIS"), + ("change_own_gis_importer", "Can import own to QGIS"), + ) ADMIN_SECTION = _("Imports") def natural_key(self): @@ -230,10 +239,29 @@ class ImporterType(models.Model): def __str__(self): return self.name + def clean(self): + if self.type == "qgs" and not self.is_template: + raise ValidationError( + "QGIS importers should at least can be exported. Check the \"Can be " + "exported\" case.") + @classmethod def is_own(cls, ishtar_user): return bool(cls.objects.filter(users__pk=ishtar_user.pk).count()) + @classmethod + def q_qgis_importers(cls, ishtaruser): + """ + Return QGIS importer query + """ + # filter available according to permissions + q = cls.objects.filter(type="qgs", is_template=True, available=True) + if not ishtaruser.is_superuser and not ishtaruser.has_permission("change_gis_importer"): + if not ishtaruser.has_permission("change_own_gis_importer"): + return q.filter(pk__isnull=True) + q = q.filter(users__pk=ishtaruser.pk) + return q + @property def type_label(self): if self.type in IMPORT_TYPES_DICT: |