forked from LBRYCommunity/lbry-sdk
58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
from sqlalchemy import text
|
|
from sqlalchemy.future import select
|
|
|
|
from ..query_context import context
|
|
from ..tables import SCHEMA_VERSION, metadata, Version, Claim, Support, Block, TX
|
|
|
|
|
|
def execute(sql):
|
|
return context().execute(text(sql))
|
|
|
|
|
|
def execute_fetchall(sql):
|
|
return context().fetchall(text(sql))
|
|
|
|
|
|
def has_claims():
|
|
return context().has_records(Claim)
|
|
|
|
|
|
def has_supports():
|
|
return context().has_records(Support)
|
|
|
|
|
|
def get_best_block_height():
|
|
context().fetchmax(Block.c.height, -1)
|
|
|
|
|
|
def insert_block(block):
|
|
context().get_bulk_loader().add_block(block).flush()
|
|
|
|
|
|
def insert_transaction(block_hash, tx):
|
|
context().get_bulk_loader().add_transaction(block_hash, tx).flush(TX)
|
|
|
|
|
|
def check_version_and_create_tables():
|
|
with context("db.connecting") as ctx:
|
|
if ctx.has_table('version'):
|
|
version = ctx.fetchone(select(Version.c.version).limit(1))
|
|
if version and version['version'] == SCHEMA_VERSION:
|
|
return
|
|
metadata.drop_all(ctx.engine)
|
|
metadata.create_all(ctx.engine)
|
|
ctx.execute(Version.insert().values(version=SCHEMA_VERSION))
|
|
for table in metadata.sorted_tables:
|
|
disable_trigger_and_constraints(table.name)
|
|
|
|
|
|
def disable_trigger_and_constraints(table_name):
|
|
ctx = context()
|
|
if ctx.is_postgres:
|
|
ctx.execute(text(f"ALTER TABLE {table_name} DISABLE TRIGGER ALL;"))
|
|
if table_name in ('tag', 'stake'):
|
|
return
|
|
if ctx.is_postgres:
|
|
ctx.execute(text(
|
|
f"ALTER TABLE {table_name} DROP CONSTRAINT {table_name}_pkey CASCADE;"
|
|
))
|