15427c96c0
- Change file mode from 644 to 755 for all project files - Add chain_id parameter to get_balance RPC endpoint with default "ait-devnet" - Rename Miner.extra_meta_data to extra_metadata for consistency
32 lines
941 B
Python
Executable File
32 lines
941 B
Python
Executable File
from __future__ import annotations
|
|
|
|
from contextlib import contextmanager
|
|
|
|
from sqlmodel import Session, SQLModel, create_engine
|
|
from sqlalchemy import event
|
|
|
|
from .config import settings
|
|
|
|
_engine = create_engine(f"sqlite:///{settings.db_path}", echo=False)
|
|
|
|
@event.listens_for(_engine, "connect")
|
|
def set_sqlite_pragma(dbapi_connection, connection_record):
|
|
cursor = dbapi_connection.cursor()
|
|
cursor.execute("PRAGMA journal_mode=WAL")
|
|
cursor.execute("PRAGMA synchronous=NORMAL")
|
|
cursor.execute("PRAGMA cache_size=-64000")
|
|
cursor.execute("PRAGMA temp_store=MEMORY")
|
|
cursor.execute("PRAGMA mmap_size=30000000000")
|
|
cursor.execute("PRAGMA busy_timeout=5000")
|
|
cursor.close()
|
|
|
|
def init_db() -> None:
|
|
settings.db_path.parent.mkdir(parents=True, exist_ok=True)
|
|
SQLModel.metadata.create_all(_engine)
|
|
|
|
|
|
@contextmanager
|
|
def session_scope() -> Session:
|
|
with Session(_engine) as session:
|
|
yield session
|