This commit is contained in:
Kovid Goyal 2025-03-08 14:14:46 +05:30
parent 9bc711bd04
commit e6f0c0f6e3
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C

View File

@ -52,10 +52,10 @@ class Database:
return self.conn.__enter__() return self.conn.__enter__()
def __exit__(self, exc_type, exc_value, tb) -> bool | None: def __exit__(self, exc_type, exc_value, tb) -> bool | None:
suppress_exception = self.conn.__exit__(exc_type, exc_value, tb) with closing(self.conn):
self.conn.close() suppress_exception = self.conn.__exit__(exc_type, exc_value, tb)
if self.needs_copy and (suppress_exception or (exc_type is None and exc_value is None and tb is None)): if self.needs_copy and (suppress_exception or (exc_type is None and exc_value is None and tb is None)):
self.copy_db() self.copy_db()
return suppress_exception return suppress_exception
def copy_db(self): def copy_db(self):