perf: SAML2属性映射去掉域名前缀

pull/7498/head
jiangweidong 2022-01-14 15:48:15 +08:00 committed by Jiangjie.Bai
parent 451e61bb61
commit 106bb9b63d
1 changed files with 1 additions and 4 deletions

View File

@ -76,11 +76,10 @@ class PrepareRequestMixin:
@staticmethod @staticmethod
def get_attribute_consuming_service(): def get_attribute_consuming_service():
attr_mapping = settings.SAML2_RENAME_ATTRIBUTES attr_mapping = settings.SAML2_RENAME_ATTRIBUTES
name_prefix = settings.SITE_URL
if attr_mapping and isinstance(attr_mapping, dict): if attr_mapping and isinstance(attr_mapping, dict):
attr_list = [ attr_list = [
{ {
"name": '{}/{}'.format(name_prefix, sp_key), "name": sp_key,
"friendlyName": idp_key, "isRequired": True "friendlyName": idp_key, "isRequired": True
} }
for idp_key, sp_key in attr_mapping.items() for idp_key, sp_key in attr_mapping.items()
@ -168,12 +167,10 @@ class PrepareRequestMixin:
def get_attributes(self, saml_instance): def get_attributes(self, saml_instance):
user_attrs = {} user_attrs = {}
real_key_index = len(settings.SITE_URL) + 1
attrs = saml_instance.get_attributes() attrs = saml_instance.get_attributes()
valid_attrs = ['username', 'name', 'email', 'comment', 'phone'] valid_attrs = ['username', 'name', 'email', 'comment', 'phone']
for attr, value in attrs.items(): for attr, value in attrs.items():
attr = attr[real_key_index:]
if attr not in valid_attrs: if attr not in valid_attrs:
continue continue
user_attrs[attr] = self.value_to_str(value) user_attrs[attr] = self.value_to_str(value)