main.py

# encoding: utf-8
import logging
import os

from starlette.responses import RedirectResponse

from dbsession import create_all
from endpoints import (
    get_balance,
    get_utxos,
    get_blocks,
    get_blockdag,
    get_circulating_supply,
    get_spectred_info,
    get_network,
    get_fee_estimate,
    get_price,
)
from endpoints.get_address_transactions import get_transactions_for_address
from endpoints.get_blockreward import get_blockreward
from endpoints.get_halving import get_halving
from endpoints.get_hashrate import get_hashrate
from endpoints.get_health import health_state
from endpoints.get_marketcap import get_marketcap
from endpoints.get_transaction_mass import calculate_transaction_mass
from endpoints.get_transactions import get_transaction
from endpoints.get_virtual_chain_blue_score import (
    get_virtual_selected_parent_blue_score,
)
from endpoints.spectred_requests.submit_transaction_request import (
    submit_a_new_transaction,
)
from helper import get_spr_market_data
from server import app, spectred_client

IS_SQL_DB_CONFIGURED = os.getenv("SQL_URI") is not None

print(
    f"Loaded: {get_balance}, {get_utxos}, {get_blocks}, {get_blockdag}, {get_circulating_supply}, "
    f"{get_spectred_info}, {get_network}, {get_fee_estimate}, {get_marketcap}, {get_hashrate}, {get_blockreward}"
    f"{get_halving} {health_state} {get_transaction}"
    f"{get_virtual_selected_parent_blue_score} {get_transactions_for_address}"
    f"{submit_a_new_transaction} {calculate_transaction_mass} {get_price}"
)

if os.getenv("VSPC_REQUEST") == "true":
    from endpoints.get_vspc import get_virtual_selected_parent_chain_from_block

    print(get_virtual_selected_parent_chain_from_block)


@app.on_event("startup")
async def startup():
    # create db if needed
    if IS_SQL_DB_CONFIGURED:
        await create_all(drop=False)
    # get spectred
    await get_spr_market_data()

    # find spectred before staring webserver
    await spectred_client.initialize_all()


@app.get("/", include_in_schema=False)
async def docs_redirect():
    return RedirectResponse(url="/docs")


logging.basicConfig(
    format="%(asctime)s::%(levelname)s::%(name)s::%(message)s",
    level=logging.DEBUG if os.getenv("DEBUG", False) else logging.INFO,
    handlers=[logging.StreamHandler()],
)

if __name__ == "__main__":
    if os.getenv("DEBUG"):
        import uvicorn

        uvicorn.run(app)