前言
從環境變量和配置文件中獲取配置參數,相關庫:
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獲取valueParameters----------config: ConfigParser(), 配置實例對象section: str, 配置文件中的區域key: str, 配置的參數名"""# 優先從環境變量中獲取配置參數, 沒有的話再從配置文件中獲取value = os_env.get(key, "")if not value:try:value = config.get(section, key)except (NoOptionError, NoSectionError):# 沒有的話就返回Nonevalue = Nonereturn value
if __name__ == '__main__':config = read_config("test.ini")print(get_config(config, "mysql", "host"))