使用datetime获取上月指定工作日日期，如何优化含大量if的代码逻辑

优化的核心思路是字典查表替换多分支判断，将字符串参数直接映射为计算所需的数字偏移量，既能简化代码，又能提升执行效率，同时后续维护扩展也更方便。

优化后的完整实现代码如下：

from datetime import date, timedelta

# 提前定义好参数与偏移量的映射关系
WEEK_OFFSET_MAP = {
    "this_week": 0,
    "last_week": 1,
    "two_weeks_ago": 2,
    "three_weeks_ago": 3
}

WEEKDAY_OFFSET_MAP = {
    "monday": 0,
    "tuesday": 1,
    "wednesday": 2,
    "thursday": 3,
    "friday": 4,
    "saturday": 5,
    "sunday": 6
}

def set_specific_date_based_on_given_date(week, day_of_the_week):
    # 先校验参数合法性，避免非法输入导致报错
    if week not in WEEK_OFFSET_MAP:
        raise ValueError(f"非法周标识，可选值为：{list(WEEK_OFFSET_MAP.keys())}")
    if day_of_the_week not in WEEKDAY_OFFSET_MAP:
        raise ValueError(f"非法工作日名称，可选值为：{list(WEEKDAY_OFFSET_MAP.keys())}")
    
    today = date.today()
    # 直接查表获取周偏移量计算周一日期
    week_offset = WEEK_OFFSET_MAP[week]
    monday = today - timedelta(weeks=week_offset, days=today.weekday())
    # 查表获取工作日偏移量返回目标日期
    day_offset = WEEKDAY_OFFSET_MAP[day_of_the_week]
    return monday + timedelta(days=day_offset)

优化后的优势：

• 去掉了全部冗余的if分支，代码精简度提升50%以上，可读性更高
• 后续如果需要新增周标识（比如four_weeks_ago）或者修改偏移规则，只需要更新映射字典即可，不需要修改核心逻辑
• 字典查表的时间复杂度为O(1)，参数越多比多分支判断的效率优势越明显
• 新增了参数合法性校验，非法输入会抛出明确的错误提示，便于调试

内容的提问来源于stack exchange，提问作者user964503