Mathis/allianceauth
1
0
Fork 0
mirror of https://gitlab.com/allianceauth/allianceauth.git synced 2025-07-09 20:40:17 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update autogroups.

Browse Source
This commit is contained in:
colcrunch 2021-07-13 08:41:36 -04:00
parent 7c1d1074f9
commit eaa1cde01a
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
allianceauth/eveonline/autogroups/models.py
View File

@ -6,6 +6,8 @@ from django.core.exceptions import ObjectDoesNotExist
from allianceauth.authentication.models import State from allianceauth.authentication.models import State
from allianceauth.eveonline.models import EveCorporationInfo, EveAllianceInfo from allianceauth.eveonline.models import EveCorporationInfo, EveAllianceInfo
from allianceauth.groupmanagement.models import AuthGroup
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -179,12 +181,16 @@ class AutogroupsConfig(models.Model):
@transaction.atomic @transaction.atomic
def create_alliance_group(self, alliance: EveAllianceInfo) -> Group: def create_alliance_group(self, alliance: EveAllianceInfo) -> Group:
group, created = Group.objects.get_or_create(name=self.get_alliance_group_name(alliance)) group, created = Group.objects.get_or_create(name=self.get_alliance_group_name(alliance))
if created:
AuthGroup.objects.create(group=group)
ManagedAllianceGroup.objects.get_or_create(group=group, config=self, alliance=alliance) ManagedAllianceGroup.objects.get_or_create(group=group, config=self, alliance=alliance)
return group return group
@transaction.atomic @transaction.atomic
def create_corp_group(self, corp: EveCorporationInfo) -> Group: def create_corp_group(self, corp: EveCorporationInfo) -> Group:
group, created = Group.objects.get_or_create(name=self.get_corp_group_name(corp)) group, created = Group.objects.get_or_create(name=self.get_corp_group_name(corp))
if created:
AuthGroup.objects.create(group=group)
ManagedCorpGroup.objects.get_or_create(group=group, config=self, corp=corp) ManagedCorpGroup.objects.get_or_create(group=group, config=self, corp=corp)
return group return group
@ -241,6 +247,7 @@ class ManagedGroup(models.Model):
def __str__(self): def __str__(self):
return "Managed Group: %s" % self.group.name return "Managed Group: %s" % self.group.name
class ManagedCorpGroup(ManagedGroup): class ManagedCorpGroup(ManagedGroup):
corp = models.ForeignKey(EveCorporationInfo, on_delete=models.CASCADE) corp = models.ForeignKey(EveCorporationInfo, on_delete=models.CASCADE)