Enum
Enum là một kiểu dữ liệu cho phép bạn định nghĩa tập hợp các hằng số có tên. Mỗi giá trị trong enum có một tên rõ ràng và không thể thay đổi.
Khi nào dùng Enum
?
-
Khi bạn có một tập giá trị cố định, ví dụ:
"coffee"
,"tea"
,"unknown"
-
Để tránh dùng magic strings hoặc số không rõ ý nghĩa
-
Khi bạn muốn code dễ đọc, dễ bảo trì, và hỗ trợ autocomplete
Cách dùng Enum trong Python
from enum import Enum
class DrinkPreference(Enum):
COFFEE = "coffee"
TEA = "tea"
UNKNOWN = "unknown"
Bây giờ, bạn có thể sử dụng:
choice = DrinkPreference.COFFEE
if choice == DrinkPreference.TEA:
print("Bạn chọn trà.")
else:
print("Không phải trà.")
Lấy danh sách giá trị enum
for drink in DrinkPreference:
print(drink.name, "=", drink.value)
COFFEE = coffee
TEA = tea
UNKNOWN = unknown
Enum kết hợp với TypedDict
from typing import TypedDict
from enum import Enum
class DrinkPreference(Enum):
COFFEE = "coffee"
TEA = "tea"
UNKNOWN = "unknown"
class DrinkState(TypedDict):
preference: DrinkPreference
Tác giả: Đỗ Ngọc Tú
Công Ty Phần Mềm VHTSoft
Không có bình luận