from datetime import datetime, timedelta


def 取下周结束时间(时间字符串: str | None = None) -> str | None:
    """返回指定日期所在“下周”的结束时间。

    功能说明:
        - 这个函数用于找到下周周日的 ``23:59:59``。
        - 支持传入日期，也支持传入完整时间。
        - 不传参数时，默认按今天所在周往后推一周。
        - 输入格式不正确时，函数返回 ``None``。

    Args:
        时间字符串: ``YYYY-MM-DD`` 或 ``YYYY-MM-DD HH:MM:SS`` 格式的日期时间字符串。

    Returns:
        成功时返回下周结束时间；失败时返回 ``None``。
    """
    try:
        if 时间字符串 is None:
            日期对象 = datetime.now().date()
        else:
            时间字符串 = 时间字符串.strip()
            if len(时间字符串) == 10:
                日期对象 = datetime.strptime(时间字符串, "%Y-%m-%d").date()
            else:
                日期对象 = datetime.strptime(时间字符串, "%Y-%m-%d %H:%M:%S").date()

        本周开始 = 日期对象 - timedelta(days=日期对象.weekday())
        下周结束 = 本周开始 + timedelta(days=13)
        return f"{下周结束} 23:59:59"
    except Exception:
        return None


if __name__ == "__main__":
    print("===== 示例1：传入日期，取下周结束时间 =====")
    print("结果:", 取下周结束时间("2026-03-20"))

    print("\n===== 示例2：传入完整时间，取下周结束时间 =====")
    print("结果:", 取下周结束时间("2026-03-20 18:30:45"))

    print("\n===== 示例3：不传参数，取当前的下周结束时间 =====")
    print("结果:", 取下周结束时间())

    print("\n===== 示例4：输入错误格式 =====")
    print("结果:", 取下周结束时间("2026/03/20"))

    print("\n===== 示例5：输入不存在的日期 =====")
    print("结果:", 取下周结束时间("2026-02-30"))

    print("\n===== 示例6：应用场景-生成下周排班截止时间 =====")
    print("排班结束:", 取下周结束时间("2026-03-20 09:00:00"))
