python3: add detection of xml encoding
This commit is contained in:
parent
7df3a6f5e0
commit
239f39c097
|
@ -3,6 +3,7 @@ import datetime
|
|||
import importlib
|
||||
from functools import wraps
|
||||
import isodate
|
||||
from xml.parsers import expat
|
||||
|
||||
from django.contrib import auth
|
||||
from django.core.urlresolvers import reverse
|
||||
|
@ -246,3 +247,13 @@ def login(request, user):
|
|||
break
|
||||
else:
|
||||
auth.login(request, user)
|
||||
|
||||
|
||||
def get_xml_encoding(content):
|
||||
xml_encoding = 'utf-8'
|
||||
def xmlDeclHandler(version, encoding, standalone):
|
||||
xml_encoding = encoding
|
||||
parser = expat.ParserCreate()
|
||||
parser.XmlDeclHandler = xmlDeclHandler
|
||||
parser.Parse(content, True)
|
||||
return xml_encoding
|
||||
|
|
|
@ -272,8 +272,9 @@ class LoginView(ProfileMixin, LogMixin, View):
|
|||
return self.sso_failure(request, login, idp_message, status_codes)
|
||||
|
||||
self.log.info('Got SAML Artifact Response', extra={'saml_response': result.content})
|
||||
result.encoding = utils.get_xml_encoding(result.content)
|
||||
try:
|
||||
login.processResponseMsg(result.content)
|
||||
login.processResponseMsg(result.text)
|
||||
login.acceptSso()
|
||||
except lasso.ProfileMissingResponseError:
|
||||
# artifact is invalid, idp returned no response
|
||||
|
|
Loading…
Reference in New Issue