mirror of
https://github.com/nikdoof/test-auth.git
synced 2025-12-13 22:32:15 +00:00
73 lines
2.6 KiB
Python
Executable File
73 lines
2.6 KiB
Python
Executable File
#!/usr/bin/env python
|
|
"""
|
|
This module abstracts the pulling of account data from the EVE API.
|
|
"""
|
|
from xml.dom import minidom
|
|
from datetime import datetime
|
|
|
|
if __name__ == "__main__":
|
|
# Only mess with the environmental stuff if this is being ran directly.
|
|
from importer_path import fix_environment
|
|
fix_environment()
|
|
|
|
from django.conf import settings
|
|
from eve_proxy.models import CachedDocument
|
|
from eve_api.api_exceptions import APIAuthException, APINoUserIDException
|
|
from eve_api.models import EVEAccount, EVEPlayerCharacter, EVEPlayerCorporation
|
|
|
|
def import_eve_account(api_key, user_id):
|
|
"""
|
|
Imports an account from the API into the EVEAccount model.
|
|
"""
|
|
print user_id, ":", api_key
|
|
auth_params = {'userID': user_id, 'apiKey': api_key}
|
|
account_doc = CachedDocument.objects.api_query('/account/Characters.xml.aspx',
|
|
params=auth_params,
|
|
no_cache=False)
|
|
#print account_doc.body
|
|
|
|
dom = minidom.parseString(account_doc.body)
|
|
characters_node_children = dom.getElementsByTagName('rowset')[0].childNodes
|
|
|
|
# Create or retrieve the account last to make sure everything
|
|
# before here is good to go.
|
|
try:
|
|
account = EVEAccount.objects.get(id=user_id)
|
|
except EVEAccount.DoesNotExist:
|
|
account = EVEAccount(id=user_id)
|
|
|
|
account.api_key = api_key
|
|
account.api_user_id = user_id
|
|
account.api_last_updated = datetime.now()
|
|
account.save()
|
|
|
|
for node in characters_node_children:
|
|
try:
|
|
# Get this first, as it's safe.
|
|
corporation_id = node.getAttribute('corporationID')
|
|
corp, created = EVEPlayerCorporation.objects.get_or_create(id=corporation_id)
|
|
# Do this last, since the things we retrieved above are used
|
|
# on the EVEPlayerCharacter object's fields.
|
|
character_id = node.getAttribute('characterID')
|
|
pchar, created = EVEPlayerCharacter.objects.get_or_create(id=character_id)
|
|
name = node.getAttribute('name')
|
|
# Save these for last to keep the save count low.
|
|
pchar.name = name
|
|
pchar.corporation = corp
|
|
pchar.save()
|
|
account.characters.add(pchar)
|
|
except AttributeError:
|
|
# This must be a Text node, ignore it.
|
|
continue
|
|
|
|
return account
|
|
|
|
if __name__ == "__main__":
|
|
"""
|
|
Test import.
|
|
"""
|
|
api_key = settings.EVE_API_USER_KEY
|
|
#api_key += "1"
|
|
user_id = settings.EVE_API_USER_ID
|
|
import_eve_account(api_key, user_id)
|