mirror of
https://gitlab.com/allianceauth/allianceauth.git
synced 2025-07-09 12:30:15 +02:00
57 lines
2.1 KiB
Python
Executable File
57 lines
2.1 KiB
Python
Executable File
from __future__ import unicode_literals
|
|
from django.db.models import Manager, QuerySet, Q
|
|
from eveonline.managers import EveManager
|
|
from eveonline.models import EveCharacter
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class CharacterOwnershipManager(Manager):
|
|
def create_by_token(self, token):
|
|
if not EveCharacter.objects.filter(character_id=token.character_id).exists():
|
|
EveManager.create_character(token.character_id)
|
|
return self.create(character=EveCharacter.objects.get(characte_id=token.character_id), user=token.user,
|
|
owner_hash=token.character_owner_hash)
|
|
|
|
|
|
class StateQuerySet(QuerySet):
|
|
def available_to_character(self, character):
|
|
query = Q(member_characters__character_id=character.character_id)
|
|
query |= Q(member_corporations__corporation_id=character.corporation_id)
|
|
query |= Q(member_alliances__alliance_id=character.alliance_id)
|
|
query |= Q(public=True)
|
|
return self.filter(query)
|
|
|
|
def available_to_user(self, user):
|
|
if user.profile.main_character:
|
|
return self.available_to_character(user.profile.main_character)
|
|
else:
|
|
return self.none()
|
|
|
|
|
|
class StateManager(Manager):
|
|
def get_queryset(self):
|
|
return StateQuerySet(self.model, using=self._db)
|
|
|
|
def available_to_character(self, character):
|
|
return self.get_queryset().available_to_character(character)
|
|
|
|
def available_to_user(self, user):
|
|
return self.get_queryset().available_to_user(user)
|
|
|
|
def get_for_character(self, character):
|
|
states = self.get_queryset().available_to_character(character).order_by('priority')
|
|
if states.exists():
|
|
return states[0]
|
|
else:
|
|
from authentication.models import get_guest_state
|
|
return get_guest_state()
|
|
|
|
def get_for_user(self, user):
|
|
states = self.get_queryset().available_to_user(user)
|
|
if states.exists():
|
|
return states[0]
|
|
else:
|
|
from authentication.models import get_guest_state
|
|
return get_guest_state() |