forked from CGM_Public/pretix_original
Raphael MichelandGitHub
af23d6e4bf
* Upgrade Django to 3.0 and other dependencies to recent versions * Fix otp version contsraint * Remove six dependency * Resolve some warnings * Fix failing tests * Update django-countries * Resolve all RemovedInDjango31Warnings in test suite * Run isort * Fix import * Update PostgreSQL version on travis
93 lines
3.2 KiB
Python
93 lines
3.2 KiB
Python
import logging
|
|
|
|
from django import forms
|
|
from django.conf import settings
|
|
from django.utils.translation import gettext as _
|
|
from oauth2_provider.exceptions import OAuthToolkitError
|
|
from oauth2_provider.forms import AllowForm
|
|
from oauth2_provider.views import (
|
|
AuthorizationView as BaseAuthorizationView,
|
|
RevokeTokenView as BaseRevokeTokenView, TokenView as BaseTokenView,
|
|
)
|
|
|
|
from pretix.api.models import OAuthApplication
|
|
from pretix.base.models import Organizer
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class OAuthAllowForm(AllowForm):
|
|
organizers = forms.ModelMultipleChoiceField(
|
|
queryset=Organizer.objects.none(),
|
|
widget=forms.CheckboxSelectMultiple
|
|
)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
user = kwargs.pop('user')
|
|
super().__init__(*args, **kwargs)
|
|
self.fields['organizers'].queryset = Organizer.objects.filter(
|
|
pk__in=user.teams.values_list('organizer', flat=True))
|
|
|
|
|
|
class AuthorizationView(BaseAuthorizationView):
|
|
template_name = "pretixcontrol/auth/oauth_authorization.html"
|
|
form_class = OAuthAllowForm
|
|
|
|
def get_form_kwargs(self):
|
|
kwargs = super().get_form_kwargs()
|
|
kwargs['user'] = self.request.user
|
|
return kwargs
|
|
|
|
def get_context_data(self, **kwargs):
|
|
ctx = super().get_context_data(**kwargs)
|
|
ctx['settings'] = settings
|
|
return ctx
|
|
|
|
def create_authorization_response(self, request, scopes, credentials, allow, organizers):
|
|
credentials["organizers"] = organizers
|
|
return super().create_authorization_response(request, scopes, credentials, allow)
|
|
|
|
def form_valid(self, form):
|
|
client_id = form.cleaned_data["client_id"]
|
|
application = OAuthApplication.objects.get(client_id=client_id)
|
|
credentials = {
|
|
"client_id": form.cleaned_data.get("client_id"),
|
|
"redirect_uri": form.cleaned_data.get("redirect_uri"),
|
|
"response_type": form.cleaned_data.get("response_type", None),
|
|
"state": form.cleaned_data.get("state", None),
|
|
}
|
|
scopes = form.cleaned_data.get("scope")
|
|
allow = form.cleaned_data.get("allow")
|
|
|
|
try:
|
|
uri, headers, body, status = self.create_authorization_response(
|
|
request=self.request, scopes=scopes, credentials=credentials, allow=allow,
|
|
organizers=form.cleaned_data.get("organizers")
|
|
)
|
|
except OAuthToolkitError as error:
|
|
return self.error_response(error, application)
|
|
|
|
self.success_url = uri
|
|
logger.debug("Success url for the request: {0}".format(self.success_url))
|
|
|
|
msgs = [
|
|
_('The application "{application_name}" has been authorized to access your account.').format(
|
|
application_name=application.name
|
|
)
|
|
]
|
|
self.request.user.send_security_notice(msgs)
|
|
self.request.user.log_action('pretix.user.oauth.authorized', user=self.request.user, data={
|
|
'application_id': application.pk,
|
|
'application_name': application.name,
|
|
})
|
|
|
|
return self.redirect(self.success_url, application)
|
|
|
|
|
|
class TokenView(BaseTokenView):
|
|
pass
|
|
|
|
|
|
class RevokeTokenView(BaseRevokeTokenView):
|
|
pass
|