diff --git a/datum/database.py b/datum/database.py index 9447938..24c77a9 100644 --- a/datum/database.py +++ b/datum/database.py @@ -28,6 +28,12 @@ def __getitem__(self, key): """Alternate notation for getting a table: db['table']""" return self.table(key) + def __enter__(self): + return self + + def __exit__(self, exception_type, exception_value, traceback): + self.close() + @property def name(self): return self._child.name