如何编写从任意格式化的字符串中检索日期的函数

import datetime
import re

def retrieve_date_from_string(string):
    # 定义日期格式化特征的正则表达式模式
    pattern = r'\d{4}-\d{2}-\d{2}'  # 假设日期格式为YYYY-MM-DD

    # 使用正则表达式提取日期字符串
    match = re.search(pattern, string)
    if match:
        date_str = match.group(0)
        # 将日期字符串转换为日期对象
        date_obj = datetime.datetime.strptime(date_str, '%Y-%m-%d')
        return date_obj
    else:
        return None