thrown DatabaseNotFoundError exception if document is not found

jaedsonpys · Jan 2, 2023 · c915a28 · c915a28
1 parent 6acff98
commit c915a28
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/cookiedb/_document.py b/cookiedb/_document.py
@@ -74,7 +74,7 @@ def get_document(self, database: str) -> Union[None, dict]:
             with open(document_path, 'rb') as reader:
                 data = secpickle.load(reader, self._key)
         except FileNotFoundError:
-            document = None
+            raise exceptions.DatabaseNotFoundError(f'Database "{database}" not found')
         except sp_exceptions.IntegrityUnconfirmedError:
             raise exceptions.InvalidDatabaseKeyError(f'Invalid key to "{database}" database')
         else:

diff --git a/cookiedb/cookiedb.py b/cookiedb/cookiedb.py
@@ -128,7 +128,12 @@ class instance.
                 raise exceptions.DatabaseExistsError(f'Database {database_name} already exists.')
 
     def _get_database_items(self):
-        database = self._document.get_document(self._open_database)
+        try:
+            database = self._document.get_document(self._open_database)
+        except exceptions.DatabaseNotFoundError:
+            self._open_database = None
+            raise exceptions.DatabaseNotFoundError
+
         return database.get('items')
 
     def _filter_path_list(self, path_list: list) -> str: