summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorMartin Kosek <mkosek@redhat.com>2011-09-26 09:20:03 +0200
committerMartin Kosek <mkosek@redhat.com>2011-09-27 15:37:24 +0200
commitc0f3c318daca5a2a1404e829c16ab234bbaf7e9e (patch)
tree4efde4238d65cdc99e1d71754dcbec49bbd27748
parentd6d73fd8a8c0b5af02f6e12f1d6acea8bbc448c6 (diff)
downloadfreeipa-c0f3c318daca5a2a1404e829c16ab234bbaf7e9e.tar.gz
freeipa-c0f3c318daca5a2a1404e829c16ab234bbaf7e9e.tar.xz
freeipa-c0f3c318daca5a2a1404e829c16ab234bbaf7e9e.zip
Fix error messages in hbacrule
Fix NotFound error messages in hbacrule commands so that the text is consistent with the rest of the framework. https://fedorahosted.org/freeipa/ticket/1861
-rw-r--r--ipalib/plugins/hbacrule.py20
1 files changed, 16 insertions, 4 deletions
diff --git a/ipalib/plugins/hbacrule.py b/ipalib/plugins/hbacrule.py
index 22a6a0dc6..f9f04529f 100644
--- a/ipalib/plugins/hbacrule.py
+++ b/ipalib/plugins/hbacrule.py
@@ -424,7 +424,10 @@ class hbacrule_add_user(LDAPAddMember):
member_count_out = ('%i object added.', '%i objects added.')
def pre_callback(self, ldap, dn, found, not_found, *keys, **options):
- (dn, entry_attrs) = ldap.get_entry(dn, self.obj.default_attributes)
+ try:
+ (dn, entry_attrs) = ldap.get_entry(dn, self.obj.default_attributes)
+ except errors.NotFound:
+ self.obj.handle_not_found(*keys)
if 'usercategory' in entry_attrs and \
entry_attrs['usercategory'][0].lower() == 'all':
raise errors.MutuallyExclusiveError(reason="users cannot be added when user category='all'")
@@ -449,7 +452,10 @@ class hbacrule_add_host(LDAPAddMember):
member_count_out = ('%i object added.', '%i objects added.')
def pre_callback(self, ldap, dn, found, not_found, *keys, **options):
- (dn, entry_attrs) = ldap.get_entry(dn, self.obj.default_attributes)
+ try:
+ (dn, entry_attrs) = ldap.get_entry(dn, self.obj.default_attributes)
+ except errors.NotFound:
+ self.obj.handle_not_found(*keys)
if 'hostcategory' in entry_attrs and \
entry_attrs['hostcategory'][0].lower() == 'all':
raise errors.MutuallyExclusiveError(reason="hosts cannot be added when host category='all'")
@@ -474,7 +480,10 @@ class hbacrule_add_sourcehost(LDAPAddMember):
member_count_out = ('%i object added.', '%i objects added.')
def pre_callback(self, ldap, dn, found, not_found, *keys, **options):
- (dn, entry_attrs) = ldap.get_entry(dn, self.obj.default_attributes)
+ try:
+ (dn, entry_attrs) = ldap.get_entry(dn, self.obj.default_attributes)
+ except errors.NotFound:
+ self.obj.handle_not_found(*keys)
if 'sourcehostcategory' in entry_attrs and \
entry_attrs['sourcehostcategory'][0].lower() == 'all':
raise errors.MutuallyExclusiveError(reason="source hosts cannot be added when sourcehost category='all'")
@@ -552,7 +561,10 @@ class hbacrule_add_service(LDAPAddMember):
member_count_out = ('%i object added.', '%i objects added.')
def pre_callback(self, ldap, dn, found, not_found, *keys, **options):
- (dn, entry_attrs) = ldap.get_entry(dn, self.obj.default_attributes)
+ try:
+ (dn, entry_attrs) = ldap.get_entry(dn, self.obj.default_attributes)
+ except errors.NotFound:
+ self.obj.handle_not_found(*keys)
if 'servicecategory' in entry_attrs and \
entry_attrs['servicecategory'][0].lower() == 'all':
raise errors.MutuallyExclusiveError(reason="services cannot be added when service category='all'")