在 Python 中创建环境变量

在编程中，环境变量是一种常用的配置方式。Python 也提供了多种方式来创建和使用环境变量。本文将介绍在 Python 中创建环境变量的几种方法。

1. 使用 os 模块创建环境变量

os 模块是 Python 中操作系统相关功能的标准库，其中包含了访问环境变量的函数。可以使用 os 模块中的 putenv 函数来创建环境变量。

import os

os.putenv('ENV_VAR_NAME', 'ENV_VAR_VALUE')

2. 使用 os.environ 创建环境变量

os.environ 是一个字典，包含了当前 Python 解释器的环境变量。可以通过该字典操作环境变量。

import os

os.environ['ENV_VAR_NAME'] = 'ENV_VAR_VALUE'

3. 使用 dotenv 库创建环境变量

dotenv 是一个 Python 库，可以从 .env 文件中读取环境变量并将其注入到 os.environ 中。可以使用 pip 命令安装该库。

from dotenv import load_dotenv
import os

load_dotenv()

ENV_VAR_VALUE = os.getenv('ENV_VAR_NAME')

4. 使用 argparse 库创建环境变量

argparse 是 Python 中命令行参数解析的标准库。可以使用 argparse 来接收命令行参数，并将其转换为环境变量。

import argparse
import os

parser = argparse.ArgumentParser()
parser.add_argument("--env-var", "-e", help="set environment variable")

args = parser.parse_args()

if args.env_var:
    os.environ['ENV_VAR_NAME'] = args.env_var

以上是在 Python 中创建环境变量的几种方法。根据自己的需要，可以选择合适的方式进行环境变量的创建和使用。