在命令行（CMD）環(huán)境中運(yùn)行 Python 腳本時(shí)，經(jīng)常需要讀取系統(tǒng)或用戶自定義的環(huán)境變量。環(huán)境變量是操作系統(tǒng)中存儲(chǔ)全局配置信息的鍵值對(duì)，例如 PATH、HOME 或自定義的 API_KEY。Python 提供了多種方式讀取環(huán)境變量，本文將詳細(xì)講解如何通過(guò)代碼實(shí)現(xiàn)這一功能，并涵蓋常見(jiàn)問(wèn)題的解決方案和最佳實(shí)踐。

一、環(huán)境變量的基本概念

1.1 什么是環(huán)境變量

環(huán)境變量是操作系統(tǒng)中用于配置程序運(yùn)行環(huán)境的全局變量。它們可以在命令行、腳本或程序中被訪問(wèn)和修改。常見(jiàn)的環(huán)境變量包括：

• PATH：指定可執(zhí)行文件的搜索路徑。
• HOME：用戶的主目錄路徑。
• USER：當(dāng)前登錄用戶名。
• 自定義變量：用戶手動(dòng)設(shè)置的變量（如 MY_VAR=123）。

1.2 環(huán)境變量的作用

配置程序行為：例如，通過(guò) PYTHONPATH 指定 Python 模塊的搜索路徑。

傳遞敏感信息：例如，數(shù)據(jù)庫(kù)密碼或 API 密鑰。

跨平臺(tái)兼容性：通過(guò)環(huán)境變量適配不同操作系統(tǒng)的配置。

二、Python 獲取環(huán)境變量的核心方法

Python 提供了 os 模塊來(lái)操作環(huán)境變量，其中最常用的方法是 os.environ 和 os.getenv。

2.1 使用 os.environ

os.environ 是一個(gè)字典（dict），包含所有當(dāng)前進(jìn)程的環(huán)境變量。

示例代碼：

import os

# 讀取所有環(huán)境變量
print("所有環(huán)境變量：", os.environ)

# 讀取特定環(huán)境變量
path = os.environ.get("PATH")
print("PATH 環(huán)境變量：", path)

# 檢查環(huán)境變量是否存在
if "MY_VAR" in os.environ:
    print("MY_VAR 存在，值為：", os.environ["MY_VAR"])
else:
    print("MY_VAR 未設(shè)置")

輸出示例：

所有環(huán)境變量： environ({'PATH': '/usr/bin:/bin', 'HOME': '/home/user', ...})
PATH 環(huán)境變量： /usr/bin:/bin
MY_VAR 未設(shè)置

注意事項(xiàng)：

直接訪問(wèn)風(fēng)險(xiǎn)：使用 os.environ["VARIABLE"] 訪問(wèn)不存在的變量會(huì)拋出 KeyError 異常。

推薦用法：建議使用 os.environ.get("VARIABLE", "默認(rèn)值") 避免異常。

2.2 使用 os.getenv

os.getenv 是 os.environ.get 的簡(jiǎn)化方法，直接返回指定變量的值。

示例代碼：

import os

# 讀取環(huán)境變量并設(shè)置默認(rèn)值
my_var = os.getenv("MY_VAR", "default_value")
print("MY_VAR 的值：", my_var)

輸出示例（假設(shè) MY_VAR 未設(shè)置）：

MY_VAR 的值： default_value

三、在 CMD 中設(shè)置環(huán)境變量

在 Python 腳本運(yùn)行前，需要在 CMD 中設(shè)置環(huán)境變量。以下是不同操作系統(tǒng)的設(shè)置方法：

3.1 Windows 系統(tǒng)

在 CMD 中使用 set 命令：

set MY_VAR=Hello_World
python script.py

永久設(shè)置環(huán)境變量：

通過(guò)系統(tǒng)屬性界面或使用 setx 命令：

setx MY_VAR "Hello_World"

3.2 Linux/macOS 系統(tǒng)

在終端中使用 export 命令：

export MY_VAR=Hello_World
python3 script.py

永久設(shè)置環(huán)境變量：

將 export 命令添加到 ~/.bashrc 或 ~/.zshrc 文件中。

四、完整示例：讀取 CMD 環(huán)境變量

以下是一個(gè)完整的 Python 腳本示例，演示如何讀取 CMD 中設(shè)置的環(huán)境變量：

4.1 腳本代碼（script.py）

import os

# 讀取環(huán)境變量并設(shè)置默認(rèn)值
my_var = os.getenv("MY_VAR", "default_value")
path_var = os.getenv("PATH", "未找到 PATH")

# 輸出結(jié)果
print(f"MY_VAR 的值: {my_var}")
print(f"PATH 的值: {path_var}")

4.2 運(yùn)行步驟（Windows 示例）

打開(kāi) CMD 并設(shè)置環(huán)境變量：

set MY_VAR=Custom_Value

運(yùn)行 Python 腳本：

python script.py

4.3 輸出結(jié)果

MY_VAR 的值: Custom_Value
PATH 的值: C:\Windows\system32;C:\Program Files\Python39\Scripts\;...

五、常見(jiàn)問(wèn)題與解決方案

5.1 問(wèn)題1：環(huán)境變量未生效

原因：環(huán)境變量未在 CMD 中設(shè)置，或設(shè)置后未運(yùn)行腳本。

解決：確保在運(yùn)行腳本前正確設(shè)置環(huán)境變量。

5.2 問(wèn)題2：跨平臺(tái)兼容性

原因：不同操作系統(tǒng)的環(huán)境變量格式不同（如路徑分隔符）。

解決：使用 os.pathsep 處理路徑分隔符：

import os
paths = os.getenv("PATH", "").split(os.pathsep)
print("PATH 分割后的路徑：", paths)

5.3 問(wèn)題3：敏感信息泄露

原因：在代碼中直接打印或記錄環(huán)境變量值。

解決：避免輸出敏感信息（如 API 密鑰），使用日志過(guò)濾或加密存儲(chǔ)。

六、進(jìn)階技巧

6.1 修改環(huán)境變量

Python 可臨時(shí)修改當(dāng)前進(jìn)程的環(huán)境變量：

import os

# 修改環(huán)境變量
os.environ["NEW_VAR"] = "New_Value"
print("NEW_VAR:", os.getenv("NEW_VAR"))

注意：修改僅對(duì)當(dāng)前進(jìn)程及其子進(jìn)程生效，不會(huì)影響系統(tǒng)全局設(shè)置。

6.2 清除環(huán)境變量

刪除當(dāng)前進(jìn)程的環(huán)境變量：

import os

if "TEMP_VAR" in os.environ:
    del os.environ["TEMP_VAR"]
    print("TEMP_VAR 已刪除")

6.3 多級(jí)嵌套環(huán)境變量

某些環(huán)境變量可能依賴其他變量（如 PYTHONPATH）。通過(guò) os.path.expandvars 解析嵌套變量：

import os

# 假設(shè) HOME=/home/user
path = os.path.expandvars("$HOME/.config")
print("解析后的路徑：", path)

七、最佳實(shí)踐

使用默認(rèn)值：避免因環(huán)境變量缺失導(dǎo)致程序崩潰。

db_password = os.getenv("DB_PASSWORD", "local_db_password")

加密敏感信息：不要在代碼中硬編碼敏感數(shù)據(jù)，優(yōu)先通過(guò)環(huán)境變量傳遞。

跨平臺(tái)兼容性：使用 os.path 模塊處理路徑，避免硬編碼斜杠（/ 或 \）。

測(cè)試環(huán)境變量：在開(kāi)發(fā)階段通過(guò) .env 文件模擬環(huán)境變量：

from dotenv import load_dotenv
load_dotenv()  # 從 .env 文件加載環(huán)境變量

八、總結(jié)

通過(guò) os.environ 和 os.getenv，Python 可以高效讀取 CMD 或系統(tǒng)設(shè)置的環(huán)境變量，為程序提供靈活的配置能力。掌握環(huán)境變量的讀取、修改和跨平臺(tái)處理方法，不僅能提升開(kāi)發(fā)效率，還能增強(qiáng)程序的安全性和可維護(hù)性。

到此這篇關(guān)于Python獲取CMD環(huán)境變量值的完整實(shí)現(xiàn)指南的文章就介紹到這了,更多相關(guān)Python獲取CMD環(huán)境變量值內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論