security-and-privacy/ldapcherry
security-and-privacy/ldapcherry
1
0
Fork 0
mirror of https://github.com/kakwa/ldapcherry synced 2024-11-25 18:54:29 +01:00
Code Releases Activity

better exceptions handling for demo backend

Browse Source
This commit is contained in:
kakwa 2015-10-20 22:17:00 +02:00
parent 27089f68ef
commit 1985408324
1 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
ldapcherry/backend/backendDemo.py
View File

@ -7,9 +7,11 @@
# This is a demo backend # This is a demo backend
from ldapcherry.exceptions import MissingParameter
from sets import Set from sets import Set
import ldapcherry.backend import ldapcherry.backend
from ldapcherry.exceptions import UserDoesntExist, \
GroupDoesntExist, MissingParameter, \
UserAlreadyExists
import re import re
@ -99,7 +101,10 @@ class Backend(ldapcherry.backend.Backend):
""" """
self._check_fix_users(username) self._check_fix_users(username)
del self.users[username] try:
del self.users[username]
except:
raise UserDoesntExist(username, self.backend_name)
def set_attrs(self, username, attrs): def set_attrs(self, username, attrs):
""" Set a list of attributes for a given user """ Set a list of attributes for a given user
@ -169,7 +174,10 @@ class Backend(ldapcherry.backend.Backend):
.. warning:: raise UserDoesntExist if user doesn't exist .. warning:: raise UserDoesntExist if user doesn't exist
""" """
return self.users[username] try:
return self.users[username]
except:
raise UserDoesntExist(username, self.backend_name)
def get_groups(self, username): def get_groups(self, username):
""" Get a user's groups """ Get a user's groups
@ -178,4 +186,7 @@ class Backend(ldapcherry.backend.Backend):
:type username: string :type username: string
:rtype: list of groups :rtype: list of groups
""" """
return self.users[username]['groups'] try:
return self.users[username]['groups']
except:
raise UserDoesntExist(username, self.backend_name)