Reorganise the file structure into a project tree

app/eve_api/tasks/character.py

@@ -0,0 +1,158 @@
+from datetime import datetime, timedelta
+from xml.dom import minidom
+import logging
+
+from celery.decorators import task
+from celery.task.sets import subtask
+
+from eve_proxy.exceptions import *
+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.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, **kwargs):
+    """ 
+    Imports a character from the API, providing a API key will populate 
+    further details. Returns a single EVEPlayerCharacter object
+
+    """
+
+    log = import_eve_character.get_logger()
+    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=[character_id, api_key, user_id, callback], exc=exc, kwargs=kwargs)
+
+    if not pchar:
+        log.error('Error importing character %s' % character_id)
+    else:
+        if callback:
+            subtask(callback).delay(character=pchar.id)
+        else:
+            return pchar
+
+
+@task()
+def import_eve_characters(character_list, api_key=None, user_id=None, callback=None, **kwargs):
+    """
+    Imports characters from the API, providing a API key will populate
+    further details. Returns a list of EVEPlayerCharacter objects
+
+    """
+
+    log = import_eve_characters.get_logger()
+    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=[character_list, api_key, user_id, callback], exc=exc, kwargs=kwargs)
+    if callback:
+        subtask(callback).delay(characters=results)
+    else:
+        return results
+
+
+def import_eve_character_func(character_id, api_key=None, user_id=None, logger=logging.getLogger(__name__)):
+
+    try:
+        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' % (character_id, exc))
+        raise APIAccessException
+
+    d = basic_xml_parse_doc(char_doc)['eveapi']
+    if 'error' in d:
+        logger.debug('EVE API Error enountered in API document')
+        return
+
+    values = d['result']
+    pchar, created = EVEPlayerCharacter.objects.get_or_create(id=character_id)
+    if not values['characterName'] == {}:
+        pchar.name = values['characterName']
+    else:
+        pchar.name = ""
+    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 or corp.api_last_updated < (datetime.utcnow() - timedelta(hours=12)):
+        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 }
+        try:
+            char_doc = CachedDocument.objects.api_query('/char/CharacterSheet.xml.aspx', params=auth_params, 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']
+        if not 'error' in doc:
+
+            values = doc['result']
+            pchar.name = values['name']
+            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 pchar.id in acc.characters.all().values_list('id', flat=True):
+            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
+
+    return pchar