mirror of
https://github.com/nikdoof/test-auth.git
synced 2025-12-14 06:42:16 +00:00
114 lines
4.3 KiB
Python
114 lines
4.3 KiB
Python
from datetime import datetime, timedelta
|
|
from xml.dom import minidom
|
|
|
|
from celery.decorators import task
|
|
|
|
from eve_proxy.models import CachedDocument
|
|
|
|
from eve_api.models import EVEPlayerCorporation, EVEPlayerCharacter, EVEPlayerCharacterRole, EVEPlayerCharacterSkill, EVESkill, EVEAccount
|
|
from eve_api.app_defines import *
|
|
from eve_api.utils import basic_xml_parse, basic_xml_parse_doc
|
|
|
|
|
|
@task()
|
|
def import_eve_character(character_id, api_key=None, user_id=None, callback=None):
|
|
"""
|
|
Imports a character from the API, providing a API key will populate
|
|
further details
|
|
|
|
"""
|
|
|
|
char_doc = CachedDocument.objects.api_query('/eve/CharacterInfo.xml.aspx',
|
|
params={'characterID': character_id},
|
|
no_cache=False)
|
|
|
|
d = basic_xml_parse_doc(char_doc)['eveapi']
|
|
if 'error' in d:
|
|
return
|
|
values = d['result']
|
|
pchar, created = EVEPlayerCharacter.objects.get_or_create(id=character_id)
|
|
|
|
pchar.name = values['characterName']
|
|
pchar.security_status = values['securityStatus']
|
|
|
|
corp, created = EVEPlayerCorporation.objects.get_or_create(id=values['corporationID'])
|
|
|
|
from eve_api.tasks.corporation import import_corp_details
|
|
|
|
if created or not corp.name:
|
|
import_corp_details.delay(values['corporationID'])
|
|
|
|
pchar.corporation = corp
|
|
pchar.corporation_date = values['corporationDate']
|
|
|
|
for v in API_RACES_CHOICES:
|
|
val, race = v
|
|
if race == values['race']:
|
|
pchar.race = val
|
|
break
|
|
|
|
if api_key and user_id:
|
|
auth_params = {'userID': user_id, 'apiKey': api_key, 'characterID': character_id }
|
|
char_doc = CachedDocument.objects.api_query('/char/CharacterSheet.xml.aspx',
|
|
params=auth_params,
|
|
no_cache=False)
|
|
|
|
doc = basic_xml_parse_doc(char_doc)['eveapi']
|
|
if not 'error' in doc:
|
|
|
|
values = doc['result']
|
|
pchar.balance = values['balance']
|
|
pchar.attrib_intelligence = values['attributes']['intelligence']
|
|
pchar.attrib_charisma = values['attributes']['charisma']
|
|
pchar.attrib_perception = values['attributes']['perception']
|
|
pchar.attrib_willpower = values['attributes']['willpower']
|
|
pchar.attrib_memory = values['attributes']['memory']
|
|
|
|
# Process the character's skills
|
|
pchar.total_sp = 0
|
|
for skill in values.get('skills', None):
|
|
skillobj, created = EVESkill.objects.get_or_create(id=skill['typeID'])
|
|
charskillobj, created = EVEPlayerCharacterSkill.objects.get_or_create(skill=skillobj, character=pchar)
|
|
if created or not charskillobj.level == int(skill['level']) or not charskillobj.skillpoints == int(skill['skillpoints']):
|
|
charskillobj.level = int(skill['level'])
|
|
charskillobj.skillpoints = int(skill['skillpoints'])
|
|
charskillobj.save()
|
|
pchar.total_sp = pchar.total_sp + int(skill['skillpoints'])
|
|
|
|
# Process the character's roles
|
|
pchar.director = False
|
|
pchar.roles.clear()
|
|
roles = values.get('corporationRoles', None)
|
|
if roles and len(roles):
|
|
for r in roles:
|
|
role, created = EVEPlayerCharacterRole.objects.get_or_create(roleid=r['roleID'], name=r['roleName'])
|
|
pchar.roles.add(role)
|
|
if r['roleName'] == 'roleDirector':
|
|
pchar.director = True
|
|
|
|
if values['gender'] == 'Male':
|
|
pchar.gender = API_GENDER_MALE
|
|
else:
|
|
pchar.gender = API_GENDER_FEMALE
|
|
|
|
|
|
pchar.api_last_updated = datetime.utcnow()
|
|
pchar.save()
|
|
|
|
try:
|
|
acc = EVEAccount.objects.get(api_user_id=user_id)
|
|
|
|
if not char in acc.characters.all():
|
|
acc.characters.add(pchar)
|
|
|
|
if pchar.director and acc.api_keytype == API_KEYTYPE_FULL:
|
|
from eve_api.tasks.corporation import import_corp_members
|
|
import_corp_members.delay(api_key=api_key, api_userid=user_id, character_id=pchar.id)
|
|
except EVEAccount.DoesNotExist:
|
|
pass
|
|
|
|
if callback:
|
|
callback.delay(character=pchar.id)
|
|
else:
|
|
return pchar
|