starred/chia-blockchain
1
0
Fork 0
mirror of https://github.com/Chia-Network/chia-blockchain.git synced 2026-09-24 07:25:03 -05:00
Code Issues Packages Projects Releases Wiki Activity

simplify db connection helper (more simply) (#15959)

Browse Source
This commit is contained in:
Kyle Altendorf
2023-08-08 10:14:59 -05:00
committed by GitHub
parent 03248c9a88
commit 603e34e9b5
1 changed files with 7 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/util/db_connection.py
+7 -9
View File
@@ -10,12 +10,10 @@ from chia.util.db_wrapper import DBWrapper2
@asynccontextmanager
async def DBConnection(db_version: int) -> AsyncIterator[DBWrapper2]:
db_path = Path(tempfile.NamedTemporaryFile().name)
if db_path.exists():
db_path.unlink()
_db_wrapper = await DBWrapper2.create(database=db_path, reader_count=4, db_version=db_version)
try:
yield _db_wrapper
finally:
await _db_wrapper.close()
db_path.unlink()
with tempfile.TemporaryDirectory() as directory:
db_path = Path(directory).joinpath("db.sqlite")
_db_wrapper = await DBWrapper2.create(database=db_path, reader_count=4, db_version=db_version)
try:
yield _db_wrapper
finally:
await _db_wrapper.close()