From da3d694a4a118540d1f326d349060603cd26edf0 Mon Sep 17 00:00:00 2001 From: Andrew Williams Date: Thu, 14 Oct 2010 09:44:35 +0100 Subject: [PATCH] Reference the first API key for the character --- sso/services/miningbuddy/__init__.py | 3 ++- sso/services/postracker/__init__.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/sso/services/miningbuddy/__init__.py b/sso/services/miningbuddy/__init__.py index 6c3270f..74b2565 100644 --- a/sso/services/miningbuddy/__init__.py +++ b/sso/services/miningbuddy/__init__.py @@ -49,7 +49,8 @@ class MiningBuddyService(BaseDBService): self.db.connection.commit() userid = self.dbcursor.lastrowid - self.dbcursor.execute(self.SQL_ADD_API, [userid, int(time.time()), kwargs['character'].eveaccount.api_user_id, kwargs['character'].eveaccount.api_key, kwargs['character'].id]) + api = kwargs['character'].eveaccount_set.all()[0] + self.dbcursor.execute(self.SQL_ADD_API, [userid, int(time.time()), api.api_user_id, api.api_key, kwargs['character'].id]) self.db.connection.commit() return { 'username': self._clean_username(username), 'password': password } diff --git a/sso/services/postracker/__init__.py b/sso/services/postracker/__init__.py index c3f8859..da70fe6 100644 --- a/sso/services/postracker/__init__.py +++ b/sso/services/postracker/__init__.py @@ -34,7 +34,7 @@ class POSTrackerService(BaseDBService): """ Add a user """ email = kwargs['user'].email pwhash, salt = self._gen_pwhash(password) - eveid = kwargs['character'].eveaccount.api_user_id + eveid = kwargs['character'].eveaccount_set.all()[0].api_user_id self.dbcursor.execute(self.SQL_ADD_USER, [eveid, username, "%s%s" % (salt, pwhash) , email]) self.db.connection.commit()