from datetime import datetime, timedelta
from xml.dom import minidom

from celery.decorators import task
from celery.task.set import subtask

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. Returns a single EVEPlayerCharacter object

    """

    pchar = import_eve_character_func(character_id, api_key, user_id)
    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):
    """
    Imports characters from the API, providing a API key will populate
    further details. Returns a list of EVEPlayerCharacter objects

    """

    results = [import_eve_character_func(char, api_key, user_id) for char in character_list]
    if callback:
        subtask(callback).delay(characters=results)
    else:
        return results


def import_eve_character_func(character_id, api_key=None, user_id=None):
    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 pchar 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

    return pchar