def get_group(self, name):
with ldap_context(self.ldap_settings):
group_dn, group_data = get_group_by_id(name,
[self.ldap_settings['gid']])
if not group_dn:
return None
group_name = to_unicode(group_data[self.ldap_settings['gid']][0])
return self.group_class(self, group_name, group_dn)
def get_group(self, name):
with ldap_context(self.ldap_settings):
group_dn, group_data = get_group_by_id(name,
[self.ldap_settings['gid']])
if not group_dn:
return None
return self.group_class(self,
group_data.get(self.ldap_settings['gid'])[0],
group_dn)
def has_member(self, user_identifier):
with ldap_context(self.ldap_settings):
user_dn, user_data = get_user_by_id(user_identifier, attributes=[self.ldap_settings['member_of_attr']])
if not user_dn:
return False
if self.ldap_settings['ad_group_style']:
group_dn, group_data = get_group_by_id(self.name, attributes=['objectSid'])
group_sids = group_data.get('objectSid')
token_groups = get_token_groups_from_user_dn(user_dn)
return any(group_sid in token_groups for group_sid in group_sids)
else:
return self.dn in user_data.get(self.ldap_settings['member_of_attr'], [])
def has_member(self, user_identifier):
with ldap_context(self.ldap_settings):
user_dn, user_data = get_user_by_id(
user_identifier,
attributes=[self.ldap_settings['member_of_attr']])
if not user_dn:
return False
if self.ldap_settings['ad_group_style']:
group_dn, group_data = get_group_by_id(
self.name, attributes=['objectSid'])
group_sids = group_data.get('objectSid')
token_groups = get_token_groups_from_user_dn(user_dn)
return any(group_sid in token_groups
for group_sid in group_sids)
else:
return self.dn in user_data.get(
self.ldap_settings['member_of_attr'], [])
def test_get_group_by_id_handles_none_id():
with pytest.raises(GroupRetrievalFailed) as excinfo:
get_group_by_id(None)
assert excinfo.value.message == 'No identifier specified'
def get_group(self, name):
with ldap_context(self.ldap_settings):
group_dn, group_data = get_group_by_id(name, [self.ldap_settings['gid']])
if not group_dn:
return None
return self.group_class(self, group_data.get(self.ldap_settings['gid'])[0], group_dn)
def test_get_group_by_id_handles_none_id():
with pytest.raises(GroupRetrievalFailed) as excinfo:
get_group_by_id(None)
assert excinfo.value.message == 'No identifier specified'
