1- # -*- coding: utf-8 -*-
21import re
32from functools import wraps
43import ldap
5- import sys
64from ldap import filter as ldap_filter
75from flask import abort , current_app , g , make_response , redirect , url_for , \
86 request
@@ -235,12 +233,8 @@ def get_user_groups(self, user):
235233 if current_app .config ['LDAP_OPENLDAP' ]:
236234 group_member_filter = \
237235 current_app .config ['LDAP_GROUP_MEMBER_FILTER_FIELD' ]
238- if sys .version_info [0 ] > 2 :
239- groups = [record [1 ][group_member_filter ][0 ].decode (
240- 'utf-8' ) for record in records ]
241- else :
242- groups = [record [1 ][group_member_filter ][0 ] for
243- record in records ]
236+ groups = [record [1 ][group_member_filter ][0 ] for
237+ record in records ]
244238 return groups
245239 else :
246240 if current_app .config ['LDAP_USER_GROUPS_FIELD' ] in \
@@ -249,8 +243,6 @@ def get_user_groups(self, user):
249243 current_app .config ['LDAP_USER_GROUPS_FIELD' ]]
250244 result = [re .findall (b'(?:cn=|CN=)(.*?),' , group )[0 ]
251245 for group in groups ]
252- if sys .version_info [0 ] > 2 :
253- result = [r .decode ('utf-8' ) for r in result ]
254246 return result
255247 except ldap .LDAPError as e :
256248 raise LDAPException (self .error (e .args ))
@@ -275,8 +267,6 @@ def get_group_members(self, group):
275267 records [0 ][1 ]:
276268 members = records [0 ][1 ][
277269 current_app .config ['LDAP_GROUP_MEMBERS_FIELD' ]]
278- if sys .version_info [0 ] > 2 :
279- members = [m .decode ('utf-8' ) for m in members ]
280270 return members
281271 except ldap .LDAPError as e :
282272 raise LDAPException (self .error (e .args ))
0 commit comments