diff --git a/groups/app_defines.py b/groups/app_defines.py index 9a4b973..76ade14 100644 --- a/groups/app_defines.py +++ b/groups/app_defines.py @@ -2,11 +2,13 @@ GROUP_TYPE_BUILTIN = 0 GROUP_TYPE_PERMISSION = 1 GROUP_TYPE_CORPORATE = 2 +GROUP_TYPE_MANAGED = 3 GROUP_TYPE_CHOICES = ( (GROUP_TYPE_BUILTIN, 'Built-In'), (GROUP_TYPE_PERMISSION, 'Permission'), (GROUP_TYPE_CORPORATE, 'Corporate'), + (GROUP_TYPE_MANAGED, 'Managed'), ) diff --git a/groups/models.py b/groups/models.py index 4626d5b..0b97d32 100644 --- a/groups/models.py +++ b/groups/models.py @@ -16,6 +16,11 @@ class GroupInformation(models.Model): description = models.TextField() + def save(self): + if self.group and (self.group.eveplayercorporation_set.count() or self.group.eveplayeralliance_set.count()): + self.type = GROUP_TYPE_MANAGED + models.Model.save(self) + @staticmethod def create_group(sender, instance, created, **kwargs): if created: