SoulBook/soulbook/config/__init__.py
2024-08-01 19:38:07 +08:00

44 lines
1013 B
Python

#!/usr/bin/env python
import os
import logging
from .rules import *
logging_format = "[%(asctime)s] %(process)d-%(levelname)s "
logging_format += "%(module)s::%(funcName)s():l%(lineno)d: "
logging_format += "%(message)s"
logging.basicConfig(
format=logging_format,
level=logging.DEBUG
)
LOGGER = logging.getLogger()
def load_config():
"""
Load a config class
"""
os.environ['MODE'] = 'PRO'
mode = os.environ.get('MODE', 'DEV')
LOGGER.info('soulbook 启动模式:{}'.format(mode))
try:
# if mode == 'PRO':
# from .pro_config import ProConfig
# return ProConfig
# elif mode == 'DEV':
# from .dev_config import DevConfig
# return DevConfig
# else:
# from .dev_config import DevConfig
# return DevConfig
from .dev_config import DevConfig
return DevConfig
except ImportError:
from .config import Config
return Config
CONFIG = load_config()