validator-node-api/src/validator_node_api/validator_node.py at main · orcfax/validator-node-api · GitHub

History

133 lines (113 loc) · 3.52 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

"""Entry-point for validator node api."""

import argparse

import base64

import importlib

import logging

import sys

from pathlib import Path

import uvicorn

try:

import config

import log_handler

from helpers import get_version

except ModuleNotFoundError:

try:

from src.validator_node_api import config, log_handler

from src.validator_node_api.helpers import get_version

except ModuleNotFoundError:

from validator_node_api import config, log_handler

from validator_node_api.helpers import get_version

# Initialize application logging.

log_handler.log_init()

logger = logging.getLogger(__name__)

def main() -> None:

"""Primary entry-point for this script."""

init_ctx = config.partial_app_init()

keyfile = Path(init_ctx[config.CTX_CERTS]) / Path("server.key")

cert = Path(init_ctx[config.CTX_CERTS]) / Path("server.crt")

ca_certs = Path(init_ctx[config.CTX_CERTS]) / Path("cacerts.pem")

parser = argparse.ArgumentParser(

prog="validator-node",

description="runs an orcfax validator node",

epilog="for more information visit https://orcfax.io",

)

parser.add_argument(

"--announce",

help="return the server's public key file for connecting nodes (base64)",

required=False,

action="store_true",

)

parser.add_argument(

"--secure",

help="run validator in production",

required=False,

action="store_true",

)

parser.add_argument(

"--version",

help="return script version",

required=False,

action="store_true",

)

args = parser.parse_args()

if args.version:

print(get_version())

sys.exit()

if args.announce:

cert_data = None

with open(cert, "rb") as cert_file:

cert_data = cert_file.read()

print(base64.b64encode(cert_data).decode())

sys.exit()

import_str = "api"

try:

importlib.import_module(import_str)

import_str = "api:app"

except ModuleNotFoundError as err:

try:

logger.error(

"uvicorn import string not correctly configured: %s (%s)",

import_str,

err,

)

import_str = "src.validator_node_api.api"

importlib.import_module(import_str)

import_str = f"{import_str}:app"

logger.info("trying: %s", import_str)

except ModuleNotFoundError:

logger.error(

"uvicorn import string not correctly configured: %s (%s)",

import_str,

err,

)

import_str = "validator_node_api.api:app"

logger.info("trying: %s", import_str)

logger.info(

"starting validator-node-api: %s", init_ctx[config.CTX_SOURCE_SYSTEM_URN]

)

if args.secure:

logger.info("keyfile: %s, certs: %s", keyfile, cert)

logger.info("running validator securely")

uvicorn.run(

import_str,

host="0.0.0.0",

port=8001,

access_log=False,

log_level="info",

reload=False,

ssl_keyfile=keyfile,

ssl_certfile=cert,

ssl_ca_certs=ca_certs,

)

sys.exit()

logger.info("running validator in development mode -- do not use in production!")

uvicorn.run(

import_str,

host="0.0.0.0",

port=8001,

access_log=False,

log_level="info",

reload=True,

)

if __name__ == "__main__":

main()

validator-node-api/src/validator_node_api/validator_node.py at main · orcfax/validator-node-api · GitHub

Expand file tree

Breadcrumbs

validator_node.py

Latest commit

History

Breadcrumbs

validator_node.py

File metadata and controls