Fixed SSOUser snafu

This commit is contained in:
2010-02-25 19:21:56 +00:00
parent 982223f4aa
commit 78cb8409b4
4 changed files with 51 additions and 17 deletions

View File

@@ -1,7 +1,9 @@
from django.shortcuts import render_to_response
from django.contrib.auth.models import User
from django.contrib.auth.decorators import login_required
from sso.models import ServiceAccount
from eve_api.models.api_player import EVEAccount
from sso.models import ServiceAccount, SSOUser
def index(request):
pass
@@ -10,10 +12,23 @@ def index(request):
def profile(request):
user = request.user
profile = request.user.get_profile()
srvaccounts = ServiceAccounts.objects.get(user=request.user)
try:
profile = request.user.get_profile()
except SSOUser.DoesNotExist:
profile = SSOUser(user=request.user)
profile.save()
try:
srvaccounts = ServiceAccount.objects.get(user=request.user)
except ServiceAccount.DoesNotExist:
srvaccounts = None
return render_to_response('sso/profile.html', locals())
try:
eveaccounts = EVEAccount.objects.get(user=request.user)
except EVEAccount.DoesNotExist:
eveaccounts = None
return render_to_response('profile.html', locals())
def service_add(request):