From f05fe8a0d5fbb00b3e39fcbb2779c8b950cfe57b Mon Sep 17 00:00:00 2001 From: Blallo Date: Sat, 6 Jul 2019 21:25:23 +0200 Subject: [PATCH] Improve User remove. --- src/phi/async_ldap/model.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/phi/async_ldap/model.py b/src/phi/async_ldap/model.py index 42fb2ef..4671cea 100644 --- a/src/phi/async_ldap/model.py +++ b/src/phi/async_ldap/model.py @@ -4,7 +4,7 @@ import asyncio import logging import typing as T -from bonsai import LDAPEntry, LDAPModOp +from bonsai import LDAPEntry, LDAPModOp, NoSuchObjectError from phi.logging import get_logger @@ -286,7 +286,10 @@ class User(Hackers): async def remove(self): async with self.client.connect(is_async=True) as conn: self._entry.connection = conn - await self._entry.delete() + try: + await self._entry.delete() + except NoSuchObjectError: + raise PhiUserDoesNotExist class PhiUserExists(Exception):