【Python进阶】常量和枚举类型

有时我们的程序里会出现一些不会变化的变量，这时一般需要把它们设置为常量。但有时候常量过多，会显得很乱，不容易管理，这时候就可以使用枚举类型来管理常量。
比如下面的代码：

from datetime import datetime

DAY_DATA_PAHT1 = "/data_path_20240101"
DAY_DATA_PATH2 = "/data_path_20240201"
MONTH_DATA_PATH1 = "/month_data_path_202406"
MONTH_DATA_PATH2 = "/month_data_path_202407"
DAY_DATA_TIME_FORMAT = "%Y%m%d"
MONTH_DATA_TIME_FORMAT = "%Y%m"
DAY_DATA_TIME_LOCATION = 2
MONTH_DATA_TIME_LOCATION = 3

def foo():
    day_data_time1 = DAY_DATA_PAHT1.split('_')[DAY_DATA_TIME_LOCATION]
    dt_day_time1 = datetime.strptime(day_data_time1, DAY_DATA_TIME_FORMAT)
    return dt_day_time1

这样写在小项目里还可以应付的过来，但一旦需要管理的常量过多，就容易出现混乱。
可以改写成这样，并加上注释：

from enum import Enum
from datetime import datetime

class DayDataPath(str, Enum):
    """
    每日数据地址
    """
    PATH1 = "/data_path_20240101"
    PATH2 = "/data_path_20240201"

class MonthDataPath(str, Enum):
    """
    每月数据地址
    """
    PATH1 = "/month_data_path_202406"
    PATH2 = "/month_data_path_202407"

class DataTimeFormat(str, Enum):
    """
    时间格式
    """
    DAY_TIME_FORMAT = "%Y%m%d"
    MONTH_TIME_FORMAT = "%Y%m"

class DataTimeLocation(int, Enum):
    """
    时间在文件中的位置
    """
    DAY_TIME_LOCATION = 2
    MONTH_TIME_LOCATION = 3

def foo():
    day_data_time1 = DayDataPath.PATH1.split('_')[DataTimeLocation.DAY_TIME_LOCATION]
    dt_day_time1 = datetime.strptime(day_data_time1, DataTimeFormat.DAY_TIME_FORMAT)
    return dt_day_time1

这段代码把常量分类并抽象化，我这里只是举个简单的例子，实际上不复杂的项目不需要这么抽象化，直接设计常量也可以胜任。
但复杂的代码可以采用这种方式抽象化常量，这可以帮助我们很好地规避输入错误问题，让我们的程序更加健壮。

如有错误欢迎评论指正，
如有问题欢迎评论提问。