使用python-dotenv进行高效的环境管理

发布时间 2023-09-13 14:16:17作者: 醉一心

1 python-dotenv 项目简介

  • 项目地址https://github.com/theskumar/python-dotenv
  • 功能描述:从 .env 文件读取键值对,并将它们添加到环境变量中。
  • 核心理念:在项目中,可以将所有使用的环境变量写入 .env 文件,然后以键值对的方式读取作为环境变量。

2 用法

2.1 .env文件内容示例

# 这是一个注释,将被忽略
REDIS_ADDRESS=localhost:6379
MEANING_OF_LIFE=42
MULTILINE_VAR="hello\nworld"
  • 注意:使用 export 作为每行的前缀,python-dotenv 会忽略该变量,但可以使用 source 命令运行该文件。

2.2 项目中如何使用

安装

pip install -U python-dotenv

基本用法

确保项目目录下有 .env 文件,然后在 settings.py 中添加以下代码:

# settings.py
from dotenv import load_dotenv, find_dotenv
from pathlib import Path

# 自动搜索 .env 文件
load_dotenv(verbose=True)

# 或者指定 .env 文件位置
env_path = Path('.') / '.env'
load_dotenv(dotenv_path=env_path, verbose=True)

通过 load_dotenv,可以像访问系统环境变量一样使用 .env 文件中的变量,例如 os.getenv(key, default=None)

参数

  • dotenv_path: 指定.env文件路径,当然如果不传该参数的话(默认为None)也会自定调用dotenv.find_dotenv()去查找文件位置的,但是你的文件名如果不是.env那就必须传递该参数了
  • override: 当.env 文件中有变量与系统中原来的环境变量有冲突时,按照上面的取值顺序,默认使用系统变量,如果要用.env中的变量覆盖系统变量,可以给load_dotenv() 传递参数override=True此时只是临时使用了.env 中的变量值
  • encoding: load_dotenv() 也可以传递encoding 参数指定文件的编码方式。

从流对象加载

从可读的流对象中加载,记得把读指针调到要开始读取的位置。

>>> from io import StringIO     # Python2: from StringIO import StringIO
>>> from dotenv import dotenv_values
>>> filelike = StringIO('SPAM=EGGS\\n')
>>> filelike.seek(0)
>>> parsed = dotenv_values(stream=filelike)
>>> parsed['SPAM']
'EGGS'

注意:此时返回值是个字典


3 常见问题及解决方法

  1. 问题:在 virtualenv 中尝试通过 pip 安装时遇到的问题。
  • 解决方案:应该安装 python-dotenv,使用 pip3 install python-dotenvpip install python-dotenv
  1. 坑一:load_dotenv函数默认不会更新的配置项。

    官方文档的例子简单易懂。

    from dotenv import load_dotenv, find_dotenv
    load_dotenv(find_dotenv())
    

    然而这个例子里面缺隐藏了一个大坑。此时当用户在.env中更新配置项的值时,是不会生效的。原因是load_dotenv默认不会更新已经存在的配置项。推荐使用override参数,推荐代码如下:

    from dotenv import load_dotenv, find_dotenv
    load_dotenv(find_dotenv(), override=True)
    
  2. 坑二:env文件中# 的处理。

    考虑.env中的如下写法:

    BASEURL=http://codehub.com/#/python
    

    暂且不讨论在 url 中带#是否优雅。#在 url 中表示锚点,的确是会经常用到的;然而在Python中却是表示注释开始。此时BASEURL的值会是http://codehub.com/。以下写法均符合预期。

    BASEURL="<http://codehub.com/#/python>"
    BASEURL='<http://codehub.com/#/python>'
    

    需要使用双引号 / 单引号括起来。