mirror of
https://github.com/nikdoof/test-auth.git
synced 2025-12-14 14:52:15 +00:00
123 lines
3.6 KiB
Python
123 lines
3.6 KiB
Python
from django.http import HttpResponseRedirect
|
|
from django.shortcuts import render_to_response
|
|
from django.core.urlresolvers import reverse
|
|
from django.contrib.auth.models import User
|
|
from django.contrib.auth.decorators import login_required
|
|
|
|
from eve_api.api_exceptions import APIAuthException, APINoUserIDException
|
|
from eve_api.api_puller.accounts import import_eve_account
|
|
from eve_api.models.api_player import EVEAccount
|
|
|
|
from sso.models import ServiceAccount, SSOUser
|
|
from sso.forms import EveAPIForm, ServiceAccountForm
|
|
|
|
import settings
|
|
|
|
def index(request):
|
|
return render_to_response('sso/index.html')
|
|
|
|
@login_required
|
|
def profile(request):
|
|
|
|
user = request.user
|
|
try:
|
|
profile = request.user.get_profile()
|
|
except SSOUser.DoesNotExist:
|
|
profile = SSOUser(user=request.user)
|
|
profile.save()
|
|
|
|
try:
|
|
srvaccounts = ServiceAccount.objects.filter(user=request.user).all()
|
|
except ServiceAccount.DoesNotExist:
|
|
srvaccounts = None
|
|
|
|
try:
|
|
eveaccounts = EVEAccount.objects.filter(user=request.user).all()
|
|
except EVEAccount.DoesNotExist:
|
|
eveaccounts = None
|
|
|
|
return render_to_response('profile.html', locals())
|
|
|
|
@login_required
|
|
def eveapi_add(request):
|
|
if request.method == 'POST':
|
|
form = EveAPIForm(request.POST)
|
|
if form.is_valid():
|
|
try:
|
|
acc = import_eve_account(form.cleaned_data['api_key'], form.cleaned_data['user_id'])
|
|
except APIAuthException:
|
|
return HttpResponseRedirect(reverse('sso.views.profile'))
|
|
|
|
acc.user = request.user
|
|
acc.description = form.cleaned_data['description']
|
|
acc.save()
|
|
|
|
for eacc in EVEAccount.objects.filter(user=request.user):
|
|
if acc.api_status == 1 and acc.in_corp(settings.EVE_CORP_ID):
|
|
profile = request.user.get_profile()
|
|
profile.corp_user = True
|
|
profile.save()
|
|
break
|
|
|
|
return HttpResponseRedirect(reverse('sso.views.profile')) # Redirect after POST
|
|
else:
|
|
form = EveAPIForm() # An unbound form
|
|
|
|
return render_to_response('sso/eveapi.html', {
|
|
'form': form,
|
|
})
|
|
|
|
@login_required
|
|
def eveapi_del(request, userid=0):
|
|
|
|
if userid > 0 :
|
|
|
|
try:
|
|
acc = EVEAccount.objects.get(id=userid)
|
|
except EVEAccount.DoesNotExist:
|
|
return HttpResponseRedirect(reverse('sso.views.profile'))
|
|
|
|
if acc.user == request.user:
|
|
acc.delete()
|
|
|
|
return HttpResponseRedirect(reverse('sso.views.profile'))
|
|
|
|
@login_required
|
|
def service_add(request):
|
|
if request.method == 'POST':
|
|
form = ServiceAccountForm(request.POST)
|
|
if form.is_valid():
|
|
|
|
acc = ServiceAccount()
|
|
|
|
acc.user = request.user
|
|
acc.service = form.cleaned_data['service']
|
|
acc.username = form.cleaned_data['username']
|
|
acc.password = form.cleaned_data['password']
|
|
|
|
acc.save()
|
|
return HttpResponseRedirect(reverse('sso.views.profile')) # Redirect after POST
|
|
else:
|
|
form = ServiceAccountForm() # An unbound form
|
|
|
|
return render_to_response('sso/serviceaccount.html', {
|
|
'form': form,
|
|
})
|
|
|
|
@login_required
|
|
def service_del(request, serviceid=0):
|
|
if serviceid > 0 :
|
|
|
|
try:
|
|
acc = ServiceAccount.objects.get(id=serviceid)
|
|
except ServiceAccount.DoesNotExist:
|
|
return HttpResponseRedirect(reverse('sso.views.profile'))
|
|
|
|
if acc.user == request.user:
|
|
acc.delete()
|
|
|
|
return HttpResponseRedirect(reverse('sso.views.profile'))
|
|
|
|
|
|
|