动态引入模块并注入模块参数【python】【2022-12-30】

项目里有一些配置信息,是以模块参数的形式存在的(形如django的settings.py)。因为开发环境和生产环境的区别,这些配置信息也有区别。如果每次发布时都要手动地将这些配置信息修改成生产环境的内容,然后再发布,很容易出问题,能不能通过一种方式,让项目能自动识别当前的环境,获取正确的参数呢??

首先,将配置文件拆分成四个,统一放在一个package下:

见名知意,大部分的配置都放在base.py里。对于不同环境要有不同配置的参数,要放到相应的develop.py、pilot.py或production.py中,后面三个的配置信息会覆盖掉base.py的信息。

接下来的工作有两项:A 让系统得知自己当前所处的环境 B 将对应的python模块中的参数读取出来,注入到package中。
对于A,很简单,使用socket.gethostname()获取当前主机名即可。

难点在于B。分两步: 读取模块中的参数;注入到package中。
做法如下: 在app_settings.init中:

import socket
from loguru import logger
import traceback
import importlib

from .base import *


host_name = socket.gethostname().upper()
try:
    # 获取模块参数
    if host_name.startswith("POR"):
        logger.debug("### POR ENV ###")
        settings = importlib.import_module('config.app_settings.production')
    elif host_name.startswith("PILOT"):
        logger.debug("### PILOT ENV ###")
        settings = importlib.import_module('config.app_settings.pilot')
    else:
        logger.debug("### DEVELOP ENV ###")
        settings = importlib.import_module('config.app_settings.develop')
    
    # 参数注入。注意跳过内置参数(以_开头)
    for param in dir(settings):
        if param.startswith("_"):
            pass
        else:
            globals()[param] = getattr(settings, param)

except Exception as e:
    err_msg = "Error happened during handle config items in init of app_settings: " + str(e)
    logger.error(err_msg)
    logger.error(traceback.format_exc())

如何在项目的其他位置使用参数:先引入from config import app_settings,然后通过app_settings.XXX来获取参数值。

测试:
base.py:

AAA = 'baseAAA'

develop.py:

AAA = "developAAA"

项目的其他位置:

from config import app_settings

# ...

print(app_settings.AAA)

结果:


最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容