mirror of
https://gitlab.com/allianceauth/allianceauth.git
synced 2025-07-09 12:30:15 +02:00
95 lines
3.8 KiB
Python
95 lines
3.8 KiB
Python
from models import EveCharacter
|
|
from models import EveApiKeyPair
|
|
from services.eveapi_manager import EveApiManager
|
|
|
|
|
|
class EveManager():
|
|
|
|
def __init__(self):
|
|
pass
|
|
|
|
def create_character(self, character_id, character_name, corporation_id,
|
|
corporation_name, corporation_ticker, alliance_id,
|
|
alliance_name, user, api_id):
|
|
|
|
if not EveCharacter.objects.filter(character_id=character_id).exists():
|
|
eve_char = EveCharacter();
|
|
eve_char.character_id = character_id
|
|
eve_char.character_name = character_name
|
|
eve_char.corporation_id = corporation_id
|
|
eve_char.corporation_name = corporation_name
|
|
eve_char.corporation_ticker = corporation_ticker
|
|
eve_char.alliance_id = alliance_id
|
|
eve_char.alliance_name = alliance_name
|
|
eve_char.user = user
|
|
eve_char.api_id = api_id
|
|
eve_char.save()
|
|
|
|
def create_characters_from_list(self, chars, user, api_id):
|
|
evemanager = EveApiManager()
|
|
|
|
for char in chars.result:
|
|
if not self.check_if_character_exist(chars.result[char]['name']):
|
|
self.create_character(chars.result[char]['id'],
|
|
chars.result[char]['name'],
|
|
chars.result[char]['corp']['id'],
|
|
chars.result[char]['corp']['name'],
|
|
evemanager.get_corporation_ticker_from_id(chars.result[char]['corp']['id']),
|
|
chars.result[char]['alliance']['id'],
|
|
chars.result[char]['alliance']['name'],
|
|
user, api_id)
|
|
|
|
def create_api_keypair(self, api_id, api_key, user_id):
|
|
if not EveApiKeyPair.objects.filter(api_id=api_id).exists():
|
|
api_pair = EveApiKeyPair()
|
|
api_pair.api_id = api_id
|
|
api_pair.api_key = api_key
|
|
api_pair.user = user_id
|
|
api_pair.save()
|
|
|
|
def get_api_key_pairs(self, user_id):
|
|
if EveApiKeyPair.objects.filter(user=user_id).exists():
|
|
return EveApiKeyPair.objects.filter(user=user_id)
|
|
|
|
def delete_api_key_pair(self, api_id, user_id):
|
|
if EveApiKeyPair.objects.filter(api_id=api_id).exists():
|
|
# Check that its owned by our user_id
|
|
apikeypair = EveApiKeyPair.objects.get(api_id=api_id)
|
|
if apikeypair.user.id == user_id:
|
|
apikeypair.delete()
|
|
|
|
def delete_characters_by_api_id(self, api_id, user_id):
|
|
if EveCharacter.objects.filter(api_id=api_id).exists():
|
|
# Check that its owned by our user_id
|
|
characters = EveCharacter.objects.filter(api_id=api_id)
|
|
|
|
for char in characters:
|
|
if char.user.id == user_id:
|
|
char.delete()
|
|
|
|
def check_if_character_exist(self, char_name):
|
|
return EveCharacter.objects.filter(character_name=char_name).exists()
|
|
|
|
def get_characters_by_owner_id(self, user):
|
|
if EveCharacter.objects.filter(user=user).exists():
|
|
return EveCharacter.objects.all().filter(user=user)
|
|
|
|
return None
|
|
|
|
def get_character_by_id(self, char_id):
|
|
if EveCharacter.objects.filter(character_id=char_id).exists():
|
|
return EveCharacter.objects.get(character_id=char_id)
|
|
|
|
return None
|
|
|
|
def get_charater_alliance_id_by_id(self, char_id):
|
|
if EveCharacter.objects.filter(character_id=char_id).exists():
|
|
return EveCharacter.objects.get(character_id=char_id).alliance_id
|
|
|
|
def check_if_character_owned_by_user(self, char_id, user):
|
|
character = EveCharacter.objects.get(character_id=char_id)
|
|
|
|
if character.user.id == user.id:
|
|
return True
|
|
|
|
return False |