2014-10-05 18:34:02 -07:00

51 lines
2.0 KiB
Python

from models import EveCharacter
class EveCharacterManager():
def __init__(self):
pass
def create_character(self, character_id, character_name, corporation_id,
corporation_name, alliance_id,
alliance_name, allianceuser_owner):
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.alliance_id = alliance_id
eve_char.alliance_name = alliance_name
eve_char.allianceuser_owner = allianceuser_owner
eve_char.save()
def create_characters_from_list(self, chars, owner):
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'],
chars.result[char]['alliance']['id'], chars.result[char]['alliance']['name'],
owner)
def check_if_character_exist(self, char_name):
return EveCharacter.objects.filter(character_name=char_name).exists()
def get_characters_by_owner_id(self, owner_id):
return EveCharacter.objects.all().filter(allianceuser_owner=owner_id)
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 check_if_character_owned_by_user(self, char_id, user_id):
character = EveCharacter.objects.get(character_id = char_id)
if character.allianceuser_owner.id == user_id:
return True
return False