mirror of
https://github.com/Chia-Network/chia-blockchain.git
synced 2026-08-29 02:24:35 -05:00
33 lines
954 B
Python
33 lines
954 B
Python
# Package: utils
|
|
|
|
from __future__ import annotations
|
|
|
|
import sqlite3
|
|
|
|
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
|
|
|
|
|
|
async def set_db_version_async(db: aiosqlite.Connection, version: int) -> None:
|
|
await db.execute("CREATE TABLE database_version(version int)")
|
|
await db.execute("INSERT INTO database_version VALUES (?)", (version,))
|
|
await db.commit()
|
|
|
|
|
|
def set_db_version(db: sqlite3.Connection, version: int) -> None:
|
|
db.execute("CREATE TABLE database_version(version int)")
|
|
db.execute("INSERT INTO database_version VALUES (?)", (version,))
|
|
db.commit()
|