mirror of
https://github.com/nikdoof/test-auth.git
synced 2025-12-14 06:42:16 +00:00
Better handling of character API import and error handling
This commit is contained in:
@@ -14,3 +14,11 @@ class APINoUserIDException(Exception):
|
|||||||
"""
|
"""
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
return "This query requires a valid userID, but yours is either missing or invalid."
|
return "This query requires a valid userID, but yours is either missing or invalid."
|
||||||
|
|
||||||
|
class APIAccessException(Exception)
|
||||||
|
"""
|
||||||
|
Raised on Access errors of any kind, network and EVE API failures
|
||||||
|
"""
|
||||||
|
|
||||||
|
def __str__(self):
|
||||||
|
return "An error was encountered while accessing the EVE API."
|
||||||
|
|||||||
@@ -1,11 +1,14 @@
|
|||||||
from datetime import datetime, timedelta
|
from datetime import datetime, timedelta
|
||||||
from xml.dom import minidom
|
from xml.dom import minidom
|
||||||
|
import logging
|
||||||
|
|
||||||
from celery.decorators import task
|
from celery.decorators import task
|
||||||
from celery.task.sets import subtask
|
from celery.task.sets import subtask
|
||||||
|
|
||||||
|
from eve_proxy.exceptions import *
|
||||||
from eve_proxy.models import CachedDocument
|
from eve_proxy.models import CachedDocument
|
||||||
|
|
||||||
|
from eve_api.api_exceptions import *
|
||||||
from eve_api.models import EVEPlayerCorporation, EVEPlayerCharacter, EVEPlayerCharacterRole, EVEPlayerCharacterSkill, EVESkill, EVEAccount
|
from eve_api.models import EVEPlayerCorporation, EVEPlayerCharacter, EVEPlayerCharacterRole, EVEPlayerCharacterSkill, EVESkill, EVEAccount
|
||||||
from eve_api.app_defines import *
|
from eve_api.app_defines import *
|
||||||
from eve_api.utils import basic_xml_parse, basic_xml_parse_doc
|
from eve_api.utils import basic_xml_parse, basic_xml_parse_doc
|
||||||
@@ -19,7 +22,13 @@ def import_eve_character(character_id, api_key=None, user_id=None, callback=None
|
|||||||
|
|
||||||
"""
|
"""
|
||||||
|
|
||||||
pchar = import_eve_character_func(character_id, api_key, user_id)
|
log = import_eve_character.getLogger()
|
||||||
|
try:
|
||||||
|
pchar = import_eve_character_func(character_id, api_key, user_id, log)
|
||||||
|
except APIAccessException, exc:
|
||||||
|
log.error('Error importing character - flagging for retry')
|
||||||
|
import_eve_character.retry(args=[char, api_key, user_id, callback], exc=exc)
|
||||||
|
|
||||||
if callback:
|
if callback:
|
||||||
subtask(callback).delay(character=pchar.id)
|
subtask(callback).delay(character=pchar.id)
|
||||||
else:
|
else:
|
||||||
@@ -34,31 +43,38 @@ def import_eve_characters(character_list, api_key=None, user_id=None, callback=N
|
|||||||
|
|
||||||
"""
|
"""
|
||||||
|
|
||||||
results = [import_eve_character_func(char, api_key, user_id) for char in character_list]
|
log = import_eve_character.getLogger()
|
||||||
|
try:
|
||||||
|
results = [import_eve_character_func(char, api_key, user_id, log) for char in character_list]
|
||||||
|
except APIAccessException, exc:
|
||||||
|
log.error('Error importing characters - flagging for retry')
|
||||||
|
import_eve_characters.retry(args=[char, api_key, user_id, callback], exc=exc)
|
||||||
if callback:
|
if callback:
|
||||||
subtask(callback).delay(characters=results)
|
subtask(callback).delay(characters=results)
|
||||||
else:
|
else:
|
||||||
return results
|
return results
|
||||||
|
|
||||||
|
|
||||||
def import_eve_character_func(character_id, api_key=None, user_id=None):
|
def import_eve_character_func(character_id, api_key=None, user_id=None, logger=logging.getLogger(__name__)):
|
||||||
char_doc = CachedDocument.objects.api_query('/eve/CharacterInfo.xml.aspx',
|
|
||||||
params={'characterID': character_id},
|
try:
|
||||||
no_cache=False)
|
char_doc = CachedDocument.objects.api_query('/eve/CharacterInfo.xml.aspx', params={'characterID': character_id}, no_cache=False)
|
||||||
|
except DocumentRetrievalError, exc:
|
||||||
|
logger.error('Error retrieving CharacterInfo.xml.aspx for Character ID %s - %s' % (user_id, character_id, exc))
|
||||||
|
raise APIAccessException
|
||||||
|
|
||||||
d = basic_xml_parse_doc(char_doc)['eveapi']
|
d = basic_xml_parse_doc(char_doc)['eveapi']
|
||||||
if 'error' in d:
|
if 'error' in d:
|
||||||
|
logger.debug('EVE API Error enountered in API document')
|
||||||
return
|
return
|
||||||
|
|
||||||
values = d['result']
|
values = d['result']
|
||||||
pchar, created = EVEPlayerCharacter.objects.get_or_create(id=character_id)
|
pchar, created = EVEPlayerCharacter.objects.get_or_create(id=character_id)
|
||||||
|
|
||||||
pchar.name = values['characterName']
|
pchar.name = values['characterName']
|
||||||
pchar.security_status = values['securityStatus']
|
pchar.security_status = values['securityStatus']
|
||||||
|
|
||||||
corp, created = EVEPlayerCorporation.objects.get_or_create(id=values['corporationID'])
|
corp, created = EVEPlayerCorporation.objects.get_or_create(id=values['corporationID'])
|
||||||
|
|
||||||
from eve_api.tasks.corporation import import_corp_details
|
from eve_api.tasks.corporation import import_corp_details
|
||||||
|
|
||||||
if created or not corp.name:
|
if created or not corp.name:
|
||||||
import_corp_details.delay(values['corporationID'])
|
import_corp_details.delay(values['corporationID'])
|
||||||
|
|
||||||
@@ -73,9 +89,11 @@ def import_eve_character_func(character_id, api_key=None, user_id=None):
|
|||||||
|
|
||||||
if api_key and user_id:
|
if api_key and user_id:
|
||||||
auth_params = {'userID': user_id, 'apiKey': api_key, 'characterID': character_id }
|
auth_params = {'userID': user_id, 'apiKey': api_key, 'characterID': character_id }
|
||||||
char_doc = CachedDocument.objects.api_query('/char/CharacterSheet.xml.aspx',
|
try:
|
||||||
params=auth_params,
|
char_doc = CachedDocument.objects.api_query('/char/CharacterSheet.xml.aspx', params=auth_params, no_cache=False)
|
||||||
no_cache=False)
|
except DocumentRetrievalError, exc:
|
||||||
|
logger.error('Error retrieving CharacterSheet.xml.aspx for User ID %s, Character ID %s - %s' % (user_id, character_id, exc))
|
||||||
|
raise APIAccessException
|
||||||
|
|
||||||
doc = basic_xml_parse_doc(char_doc)['eveapi']
|
doc = basic_xml_parse_doc(char_doc)['eveapi']
|
||||||
if not 'error' in doc:
|
if not 'error' in doc:
|
||||||
@@ -121,7 +139,6 @@ def import_eve_character_func(character_id, api_key=None, user_id=None):
|
|||||||
|
|
||||||
try:
|
try:
|
||||||
acc = EVEAccount.objects.get(api_user_id=user_id)
|
acc = EVEAccount.objects.get(api_user_id=user_id)
|
||||||
|
|
||||||
if not pchar in acc.characters.all():
|
if not pchar in acc.characters.all():
|
||||||
acc.characters.add(pchar)
|
acc.characters.add(pchar)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user