mirror of
https://gitlab.com/allianceauth/allianceauth.git
synced 2025-07-09 12:30:15 +02:00
02f2968ee5
# Conflicts: # alliance_auth/__init__.py # corputils/models.py # corputils/views.py # eveonline/tasks.py # fleetactivitytracking/views.py # hrapplications/admin.py # requirements.txt
184 lines
7.3 KiB
Python
184 lines
7.3 KiB
Python
from __future__ import unicode_literals
|
|
from django.utils.encoding import python_2_unicode_compatible
|
|
from django.db import models
|
|
from eveonline.models import EveCorporationInfo, EveCharacter
|
|
from esi.models import Token
|
|
from esi.errors import TokenError
|
|
from notifications import notify
|
|
from authentication.models import CharacterOwnership, UserProfile
|
|
from bravado.exception import HTTPForbidden
|
|
from corputils.managers import CorpStatsManager
|
|
import logging
|
|
import os
|
|
|
|
|
|
SWAGGER_SPEC_PATH = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'swagger.json')
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@python_2_unicode_compatible
|
|
class CorpStats(models.Model):
|
|
token = models.ForeignKey(Token, on_delete=models.CASCADE)
|
|
corp = models.OneToOneField(EveCorporationInfo)
|
|
last_update = models.DateTimeField(auto_now=True)
|
|
|
|
class Meta:
|
|
permissions = (
|
|
('view_corp_corpstats', 'Can view corp stats of their corporation.'),
|
|
('view_alliance_corpstats', 'Can view corp stats of members of their alliance.'),
|
|
('view_state_corpstats', 'Can view corp stats of members of their auth state.'),
|
|
)
|
|
verbose_name = "corp stats"
|
|
verbose_name_plural = "corp stats"
|
|
|
|
objects = CorpStatsManager()
|
|
|
|
def __str__(self):
|
|
return "%s for %s" % (self.__class__.__name__, self.corp)
|
|
|
|
def update(self):
|
|
try:
|
|
c = self.token.get_esi_client(spec_file=SWAGGER_SPEC_PATH)
|
|
assert c.Character.get_characters_character_id(character_id=self.token.character_id).result()[
|
|
'corporation_id'] == int(self.corp.corporation_id)
|
|
members = c.Corporation.get_corporations_corporation_id_members(
|
|
corporation_id=self.corp.corporation_id).result()
|
|
member_ids = [m['character_id'] for m in members]
|
|
|
|
# requesting too many ids per call results in a HTTP400
|
|
# the swagger spec doesn't have a maxItems count
|
|
# manual testing says we can do over 350, but let's not risk it
|
|
member_id_chunks = [member_ids[i:i + 255] for i in range(0, len(member_ids), 255)]
|
|
member_name_chunks = [c.Character.get_characters_names(character_ids=id_chunk).result() for id_chunk in
|
|
member_id_chunks]
|
|
member_list = {}
|
|
for name_chunk in member_name_chunks:
|
|
member_list.update({m['character_id']: m['character_name'] for m in name_chunk})
|
|
|
|
# bulk create new member models
|
|
missing_members = [m_id for m_id in member_ids if
|
|
not CorpMember.objects.filter(corpstats=self, character_id=m_id).exists()]
|
|
CorpMember.objects.bulk_create(
|
|
[CorpMember(character_id=m_id, character_name=member_list[m_id], corpstats=self) for m_id in
|
|
missing_members])
|
|
|
|
# purge old members
|
|
self.members.exclude(character_id__in=member_ids).delete()
|
|
|
|
# update the timer
|
|
self.save()
|
|
|
|
except TokenError as e:
|
|
logger.warning("%s failed to update: %s" % (self, e))
|
|
if self.token.user:
|
|
notify(self.token.user, "%s failed to update with your ESI token." % self,
|
|
message="Your token has expired or is no longer valid. Please add a new one to create a new CorpStats.",
|
|
level="error")
|
|
self.delete()
|
|
except HTTPForbidden as e:
|
|
logger.warning("%s failed to update: %s" % (self, e))
|
|
if self.token.user:
|
|
notify(self.token.user, "%s failed to update with your ESI token." % self,
|
|
message="%s: %s" % (e.status_code, e.message), level="error")
|
|
self.delete()
|
|
except AssertionError:
|
|
logger.warning("%s token character no longer in corp." % self)
|
|
if self.token.user:
|
|
notify(self.token.user, "%s cannot update with your ESI token." % self,
|
|
message="%s cannot update with your ESI token as you have left corp." % self, level="error")
|
|
self.delete()
|
|
|
|
@property
|
|
def member_count(self):
|
|
return self.members.count()
|
|
|
|
@property
|
|
def user_count(self):
|
|
return len(set([m.main_character for m in self.members.all() if m.main_character]))
|
|
|
|
@property
|
|
def registered_member_count(self):
|
|
return len(self.registered_members)
|
|
|
|
@property
|
|
def registered_members(self):
|
|
return self.members.filter(pk__in=[m.pk for m in self.members.all() if m.registered])
|
|
|
|
@property
|
|
def unregistered_member_count(self):
|
|
return self.member_count - self.registered_member_count
|
|
|
|
@property
|
|
def unregistered_members(self):
|
|
return self.members.filter(pk__in=[m.pk for m in self.members.all() if not m.registered])
|
|
|
|
@property
|
|
def main_count(self):
|
|
return len(self.mains)
|
|
|
|
@property
|
|
def mains(self):
|
|
return self.members.filter(pk__in=[m.pk for m in self.members.all() if
|
|
m.main_character and int(m.main_character.character_id) == int(
|
|
m.character_id)])
|
|
|
|
def can_update(self, user):
|
|
return user.is_superuser or user == self.token.user
|
|
|
|
def corp_logo(self, size=128):
|
|
return "https://image.eveonline.com/Corporation/%s_%s.png" % (self.corp.corporation_id, size)
|
|
|
|
def alliance_logo(self, size=128):
|
|
if self.corp.alliance:
|
|
return "https://image.eveonline.com/Alliance/%s_%s.png" % (self.corp.alliance.alliance_id, size)
|
|
else:
|
|
return "https://image.eveonline.com/Alliance/1_%s.png" % size
|
|
|
|
|
|
class CorpMember(models.Model):
|
|
character_id = models.PositiveIntegerField()
|
|
character_name = models.CharField(max_length=37)
|
|
corpstats = models.ForeignKey(CorpStats, on_delete=models.CASCADE, related_name='members')
|
|
|
|
class Meta:
|
|
# not making character_id unique in case a character moves between two corps while only one updates
|
|
unique_together = ('corpstats', 'character_id')
|
|
ordering = ['character_name']
|
|
|
|
def __str__(self):
|
|
return self.character_name
|
|
|
|
@property
|
|
def character(self):
|
|
try:
|
|
return EveCharacter.objects.get(character_id=self.character_id)
|
|
except EveCharacter.DoesNotExist:
|
|
return None
|
|
|
|
@property
|
|
def main_character(self):
|
|
try:
|
|
return self.character.character_ownership.user.profile.main_character
|
|
except (CharacterOwnership.DoesNotExist, UserProfile.DoesNotExist, AttributeError):
|
|
return None
|
|
|
|
@property
|
|
def alts(self):
|
|
if self.main_character:
|
|
return [co.character for co in self.main_character.character_ownership.user.character_ownerships.all()]
|
|
else:
|
|
return []
|
|
|
|
@property
|
|
def registered(self):
|
|
return CharacterOwnership.objects.filter(character__character_id=self.character_id).exists()
|
|
|
|
def portrait_url(self, size=32):
|
|
return "https://image.eveonline.com/Character/%s_%s.jpg" % (self.character_id, size)
|
|
|
|
def __getattr__(self, item):
|
|
if item.startswith('portrait_url_'):
|
|
size = item.strip('portrait_url_')
|
|
return self.portrait_url(size)
|
|
return super(CorpMember, self).__getattr__(item) |