better exceptions handling for demo backend

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

View File

@ -7,9 +7,11 @@
# This is a demo backend
from ldapcherry.exceptions import MissingParameter
from sets import Set
import ldapcherry.backend
from ldapcherry.exceptions import UserDoesntExist, \
GroupDoesntExist, MissingParameter, \
UserAlreadyExists
import re
@ -99,7 +101,10 @@ class Backend(ldapcherry.backend.Backend):
"""
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):
""" 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
"""
return self.users[username]
try:
return self.users[username]
except:
raise UserDoesntExist(username, self.backend_name)
def get_groups(self, username):
""" Get a user's groups
@ -178,4 +186,7 @@ class Backend(ldapcherry.backend.Backend):
:type username: string
:rtype: list of groups
"""
return self.users[username]['groups']
try:
return self.users[username]['groups']
except:
raise UserDoesntExist(username, self.backend_name)