前言
从环境变量和配置文件中获取配置参数,相关库:
python-dotenv
:第三方库,需要使用pip安装configparser
:标准库
示例代码
- test.ini
[mysql]
host = "192.168.0.10"
port = 3306
user = "root"
password = "123456"
[postgresql]
host = "192.168.0.11"
port = 5432
user = "postgres"
password = "123456"
- demo.py
from configparser import ConfigParser, NoSectionError, NoOptionError
from dotenv import load_dotenv
import os
# 如果存在环境变量的文件,则加载配置到环境变量
if os.path.exists("settings.env"):
load_dotenv("settings.env")
os_env = os.environ
def read_config(filename: str) -> ConfigParser:
"""
从文件中读取配置信息
Parameters
----------
filename : str, 配置文件
"""
# 实例化对象
config = ConfigParser()
if not os.path.exists(filename):
raise FileNotFoundError(f"配置文件 {filename} 不存在")
config.read(filename, encoding="utf-8")
return config
def get_config(config: ConfigParser, section: str, key: str):
"""
根据指定section和key获取value
Parameters
----------
config: ConfigParser(), 配置实例对象
section: str, 配置文件中的区域
key: str, 配置的参数名
"""
# 优先从环境变量中获取配置参数, 没有的话再从配置文件中获取
value = os_env.get(key, "")
if not value:
try:
value = config.get(section, key)
except (NoOptionError, NoSectionError):
# 没有的话就返回None
value = None
return value
if __name__ == '__main__':
config = read_config("test.ini")
print(get_config(config, "mysql", "host"))
原创文章,作者:506227337,如若转载,请注明出处:https://blog.ytso.com/278186.html