如何从每个帐户具有多个日期范围的表中计算帐户有效的累计天数？

import datetime

# 假设从数据库中获取的数据为以下格式的列表
data = [
    {'account_id': 1, 'start_date': '2022-01-01', 'end_date': '2022-01-10'},
    {'account_id': 1, 'start_date': '2022-01-05', 'end_date': '2022-01-15'},
    {'account_id': 2, 'start_date': '2022-02-01', 'end_date': '2022-02-10'},
    {'account_id': 2, 'start_date': '2022-02-05', 'end_date': '2022-02-15'}
]

# 合并日期范围并计算累计天数的函数
def calculate_total_days(data):
    # 根据账户ID对数据进行分组
    grouped_data = {}
    for item in data:
        account_id = item['account_id']
        if account_id not in grouped_data:
            grouped_data[account_id] = []
        grouped_data[account_id].append((item['start_date'], item['end_date']))

    # 合并日期范围并计算累计天数
    total_days = {}
    for account_id, date_ranges in grouped_data.items():
        merged_ranges = merge_date_ranges(date_ranges)
        account_total_days = calculate_days(merged_ranges)
        total_days[account_id] = account_total_days

    return total_days

# 合并日期范围的函数
def merge_date_ranges(date_ranges):
    sorted_ranges = sorted(date_ranges, key=lambda x: x[0])
    merged_ranges = [sorted_ranges[0]]
    for start_date, end_date in sorted_ranges[1:]:
        last_start_date, last_end_date = merged_ranges[-1]
        if start_date <= last_end_date:
            merged_ranges[-1] = (last_start_date, max(last_end_date, end_date))
        else:
            merged_ranges.append((start_date, end_date))
    return merged_ranges

# 计算日期范围的天数
def calculate_days(date_ranges):
    total_days = 0
    for start_date, end_date in date_ranges:
        start = datetime.datetime.strptime(start_date, '%Y-%m-%d')
        end = datetime.datetime.strptime(end_date, '%Y-%m-%d')
        total_days += (end - start).days + 1
    return total_days

# 调用函数计算帐户有效的累计天数
result = calculate_total_days(data)
print(result)