mirror of
https://gitlab.com/allianceauth/allianceauth.git
synced 2025-07-09 12:30:15 +02:00
8360371ab7
Alter user field to OneToOneField Migration to enforce uniqueness pre-change Migration to ensure all users have an AuthServicesInfo Receiver to automatically create one upon user creation Replace AuthServicesInfo.get_or_create with get Prevent deletion of AuthServicesInfo from admin site Remove add and delete permissions from model. Get character names in chunks on corpstats update to prevent HTTP400 when requesting >350(ish) names Include corpstats docs. Update settings docs.
43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
from django.db import models
|
|
from authentication.models import AuthServicesInfo
|
|
from eveonline.models import EveCharacter
|
|
|
|
|
|
class CorpStatsQuerySet(models.QuerySet):
|
|
def visible_to(self, user):
|
|
# superusers get all visible
|
|
if user.is_superuser:
|
|
return self
|
|
|
|
auth = AuthServicesInfo.objects.get(user=user)
|
|
try:
|
|
char = EveCharacter.objects.get(character_id=auth.main_char_id)
|
|
# build all accepted queries
|
|
queries = []
|
|
if user.has_perm('corputils.view_corp_corpstats'):
|
|
queries.append(models.Q(corp__corporation_id=char.corporation_id))
|
|
if user.has_perm('corputils.view_alliance_corpstats'):
|
|
queries.append(models.Q(corp__alliance__alliance_id=char.alliance_id))
|
|
if user.has_perm('corputils.view_blue_corpstats'):
|
|
queries.append(models.Q(corp__is_blue=True))
|
|
|
|
# filter based on queries
|
|
if queries:
|
|
query = queries.pop()
|
|
for q in queries:
|
|
query |= q
|
|
return self.filter(query)
|
|
else:
|
|
# not allowed to see any
|
|
return self.none()
|
|
except EveCharacter.DoesNotExist:
|
|
return self.none()
|
|
|
|
|
|
class CorpStatsManager(models.Manager):
|
|
def get_queryset(self):
|
|
return CorpStatsQuerySet(self.model, using=self._db)
|
|
|
|
def visible_to(self, user):
|
|
return self.get_queryset().visible_to(user)
|