updated auto init of app

This commit is contained in:
Brian Bjarke Jensen
2024-02-23 23:29:42 +01:00
parent 131edc9913
commit cac816606a
+33 -84
View File
@@ -1,99 +1,48 @@
import logging import logging
from discord_logging.handler import DiscordHandler
from dotenv import load_dotenv from dotenv import load_dotenv
from configparser import ConfigParser
from pathlib import Path from pathlib import Path
import logging import logging
import os import os
from src.web import server # prepare optional local setup
env_path = Path(__file__).parent.parent / "local.env"
load_dotenv(env_path)
# load default values # ensure env vars set
config = ConfigParser() necesasary_var_list = {
config.read(Path(__file__).parent / 'defaults.ini') "MONGO_HOST",
LOGGER_LEVEL = config.get('main', 'logger_level') "MONGO_DB",
DISCORD_SERVICE_NAME = config.get('discord', 'service_name') "MONGO_COLLECTION",
DISCORD_LOGGER_LEVEL = config.get('discord', 'logger_level') "MONGO_USER",
"MONGO_PASSWORD",
}
def setup_logging( for env_var in necesasary_var_list:
discord_webhook_url: str, # ensure env var set
discord_service_name: str = DISCORD_SERVICE_NAME, assert (
discord_logger_level: str = DISCORD_LOGGER_LEVEL, env_var in os.environ
logger_level: str = LOGGER_LEVEL, ), (
) -> None: f"environment variable not set: {env_var}"
# setup stream handler
fmt = (
'%(asctime)s | '
'%(levelname)s | '
'%(filename)s | '
'%(funcName)s | '
'%(message)s'
) )
datefmt = '%Y-%m-%d %H:%M:%S'
level = getattr(logging, logger_level.upper())
logging.basicConfig(format=fmt, datefmt=datefmt, level=level)
logger = logging.getLogger()
# add discord handler
discord_handler = DiscordHandler(
service_name=discord_service_name,
webhook_url=discord_webhook_url,
)
discord_handler.setFormatter(logging.Formatter('%(message)s'))
level = getattr(logging, discord_logger_level.upper())
discord_handler.setLevel(level=level)
logger.addHandler(discord_handler)
logging.debug('finished')
# setup logging stream handler
fmt = (
'%(asctime)s | '
'%(levelname)s | '
'%(filename)s | '
'%(funcName)s | '
'%(message)s'
)
datefmt = '%Y-%m-%d %H:%M:%S'
logging.basicConfig(format=fmt, datefmt=datefmt, level=logging.INFO)
def initialise_app() -> None: logging.info("initialized app")
"""
Ensure all necessary environment variables are provided
"""
# load env vars
load_dotenv()
# ensure env vars set
necesasary_var_map = {
'LOGGER_LEVEL': False,
'DISCORD_SERVICE_NAME': False,
'DISCORD_WEBHOOK_URL': True,
'DISCORD_LOGGER_LEVEL': False,
'DATABASE_HOST': True,
'DATABASE_PORT': True,
'DATABASE_ENGINE': True,
'DATABASE_DATABASE': True,
'DATABASE_USERNAME': True,
'DATABASE_PASSWORD': True,
}
for env_var, must_be_set in necesasary_var_map.items():
if must_be_set:
# ensure env var set
assert (
env_var in os.environ
), (
f"environment variable not set: {env_var}"
)
# set variable from env var
locals()[env_var.lower()] = os.getenv(key=env_var)
else:
# ensure default value set
assert env_var in globals(), f"default variable not set: {env_var}"
# set variable from env var with backup from default value
locals()[env_var.lower()] = os.getenv(
key=env_var,
default=globals()[env_var]
)
setup_logging(
discord_webhook_url=locals()['discord_webhook_url'],
discord_service_name=locals()['discord_service_name'],
discord_logger_level=locals()['discord_logger_level'],
logger_level=locals()['logger_level'],
)
logging.debug('finished')
from src.web import app
server = app.server
if __name__ == "__main__": if __name__ == "__main__":
from src.web import app # prepare local env vars
# initialise_app() os.environ["MONGO_HOST"] = "localhost"
# run app
app.run(debug=True) app.run(debug=True)
logging.info("started app") logging.info("started app")