mirror of
https://github.com/Chia-Network/chia-blockchain.git
synced 2026-08-24 02:24:23 -05:00
Matt HauffandGitHub
40db4635a8
* Enable PEP604 Ruff rules * Fix harcoded signature in test * Hack CLVMStreamable test with note to fast follow
54 lines
1.4 KiB
Python
54 lines
1.4 KiB
Python
from __future__ import annotations
|
|
|
|
import contextlib
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
from collections.abc import AsyncIterator
|
|
from pathlib import Path
|
|
|
|
from chia.util.db_wrapper import DBWrapper2
|
|
|
|
|
|
@contextlib.asynccontextmanager
|
|
async def setup_db(name: str | os.PathLike[str], db_version: int) -> AsyncIterator[DBWrapper2]:
|
|
db_filename = Path(name)
|
|
try:
|
|
os.unlink(db_filename)
|
|
except FileNotFoundError:
|
|
pass
|
|
|
|
log_path: Path | None
|
|
if "--sql-logging" in sys.argv:
|
|
log_path = Path("sql.log")
|
|
else:
|
|
log_path = None
|
|
|
|
async with DBWrapper2.managed(
|
|
database=db_filename,
|
|
log_path=log_path,
|
|
db_version=db_version,
|
|
reader_count=1,
|
|
journal_mode="wal",
|
|
synchronous="full",
|
|
) as db_wrapper:
|
|
yield db_wrapper
|
|
|
|
|
|
def get_commit_hash() -> str:
|
|
try:
|
|
os.chdir(Path(os.path.realpath(__file__)).parent)
|
|
commit_hash = (
|
|
subprocess.run(["git", "rev-parse", "--short", "HEAD"], check=True, stdout=subprocess.PIPE)
|
|
.stdout.decode("utf-8")
|
|
.strip()
|
|
)
|
|
except Exception:
|
|
sys.exit("Failed to get the commit hash")
|
|
try:
|
|
if len(subprocess.run(["git", "status", "-s"], check=True, stdout=subprocess.PIPE).stdout) > 0:
|
|
raise Exception
|
|
except Exception:
|
|
commit_hash += "-dirty"
|
|
return commit_hash
|