Thông báo: Download 4 khóa học Python từ cơ bản đến nâng cao tại đây.
Cách truy cập và thiết lập biến môi trường trong Python
Biến môi trường (environment variables) được lưu trữ ở cấp độ hệ điều hành dưới dạng các cặp key-value. Chúng rất hữu ích trong việc trừu tượng hóa thông tin nhạy cảm và cho phép cấu hình ứng dụng mà không cần thay đổi mã nguồn. Bài viết này sẽ hướng dẫn bạn cách lấy và thiết lập giá trị của biến môi trường bằng Python.
Lấy giá trị của biến môi trường trong Python
Module os
trong Python cung cấp các công cụ để thao tác với hệ điều hành, bao gồm việc truy cập các biến môi trường. Bạn có thể sử dụng đoạn mã sau để lấy giá trị của chúng:
import os # In tất cả các biến môi trường dưới dạng dictionary print(os.environ) # Lấy giá trị của biến môi trường "LANG" mà không gây lỗi nếu không tồn tại print(os.environ.get("LANG"))
Giải thích:
os.environ
: Trả về tất cả các biến môi trường dưới dạng một đối tượng tương tự như dictionary.os.environ.get("KEY")
: Lấy giá trị của biến môi trường với tên "KEY". Nếu biến không tồn tại, nó trả vềNone
thay vì gây lỗi như khi dùng cú phápos.environ["KEY"]
.
Ví dụ, đầu ra có thể như sau:
{'PATH': '/usr/bin:/bin:/usr/sbin:/sbin', 'LANG': 'en_US.UTF-8', ...} en_US.UTF-8
Thiết lập biến môi trường
Bạn có thể thiết lập biến môi trường bằng cách gán giá trị cho một key trong os.environ
. Lưu ý rằng cả key và value phải là chuỗi.
Bài viết này được đăng tại [free tuts .net]
import os # Thiết lập một biến môi trường os.environ["MY_CUSTOM_KEY"] = "secret_value" # Kiểm tra giá trị vừa thiết lập print(os.environ.get("MY_CUSTOM_KEY"))
Giải thích:
os.environ["KEY"] = "VALUE"
: Gán giá trị"VALUE"
cho biến môi trường"KEY"
.- Biến vừa thiết lập sẽ tồn tại trong quá trình thực thi của chương trình Python, nhưng sẽ không tồn tại sau khi chương trình kết thúc (chỉ ảnh hưởng cục bộ).
Kết quả:
secret_value
Một số lưu ý quan trọng:
-
Tính chất tạm thời:
Các biến môi trường được thiết lập bằng Python chỉ tồn tại trong thời gian chương trình Python đang chạy. Nếu cần thiết lập biến môi trường vĩnh viễn, bạn phải chỉnh sửa trực tiếp vào cấu hình hệ thống (ví dụ: file.bashrc
,.zshrc
hoặc thiết lập trong IDE). -
Quy ước:
- Tên biến môi trường (key) thường được viết in hoa (ví dụ:
DATABASE_URL
,SECRET_KEY
). - Giá trị biến môi trường nên được trừu tượng hóa thông tin nhạy cảm như mật khẩu, API key.
- Tên biến môi trường (key) thường được viết in hoa (ví dụ:
-
Sử dụng với các thư viện:
Biến môi trường thường được sử dụng để cấu hình ứng dụng thông qua các thư viện nhưdotenv
(nạp từ file.env
) để quản lý dễ dàng hơn.
Ví dụ sử dụng biến môi trường trong ứng dụng Flask trong Python
Trong ứng dụng Flask, bạn có thể sử dụng biến môi trường để lưu trữ thông tin nhạy cảm như API key:
import os from flask import Flask app = Flask(__name__) # Lấy giá trị API key từ biến môi trường api_key = os.environ.get("API_KEY", "default_key") @app.route("/") def index(): return f"API Key: {api_key}"
Trong ví dụ này:
- Biến môi trường
API_KEY
được lấy từ hệ điều hành. - Nếu biến
API_KEY
không tồn tại, giá trị mặc định là"default_key"
sẽ được sử dụng.
Kết bài
- Biến môi trường là cách hiệu quả để quản lý các thông tin nhạy cảm và cấu hình ứng dụng.
- Module
os
trong Python cung cấp các phương pháp đơn giản để truy cập và thiết lập biến môi trường. - Bạn nên sử dụng thêm các công cụ như
python-dotenv
để quản lý biến môi trường một cách thuận tiện và an toàn hơn.
Nếu cần thêm thông tin chi tiết, hãy tham khảo tài liệu chính thức của Python.