updated auto init of app
This commit is contained in:
+33
-84
@@ -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")
|
||||||
Reference in New Issue
Block a user