chia-blockchain/chia/util/db_version.py
Arvid Norberg 70d31a8c68
V2 database framework (#9442)
* fix block_store benchmark linting issues

* basic support for database versioning

* update benchmark to run v1 and v2
2021-12-03 13:52:10 -08:00

15 lines
422 B
Python

import aiosqlite
async def lookup_db_version(db: aiosqlite.Connection) -> int:
try:
cursor = await db.execute("SELECT * from database_version")
row = await cursor.fetchone()
if row is not None and row[0] == 2:
return 2
else:
return 1
except aiosqlite.OperationalError:
# expects OperationalError('no such table: database_version')
return 1