设计一个计算电费的通用函数

设计一个计算电费的通用函数通常需要考虑几个关键因素：电价、用电量以及可能存在的阶梯电价或特殊费用。以下是一个简单的Python函数示例，它接受基础电价、阶梯电价规则（如分段价格）以及实际用电量作为参数：

def calculate_bill(basic_rate, tier_prices, total_usage, additional_charges={}):
    """
    计算电费的通用函数

    :param basic_rate: 基础电价（元/度）
    :param tier_prices: 阶梯电价分段及对应的单价，例如：[([0, 100], basic_rate), ([101, 500], rate_1), ...]
    :param total_usage: 总用电量（度）
    :param additional_charges: 可选的额外费用字典，如峰谷价差等 {'peak': peak_rate, 'valley': valley_rate}

    :return: 总电费
    """
    total_cost = 0
    for tier in tier_prices:
        if len(tier) == 2 and tier[0][1] < total_usage <= tier[1][1]:
            # 如果在当前阶梯范围内，则按阶梯内最高价收费
            cost_for_tier = tier[1][0]
        else:
            # 超出上一阶梯范围，直接加上基本价
            cost_for_tier = basic_rate * (total_usage - sum(range(*tier[0])))

        total_cost += cost_for_tier

    # 应用额外费用
    for charge_type, charge_rate in additional_charges.items():
        if charge_type in ('peak', 'valley') and charge_rate:
            # 检查是否有特定时间的电价
            peak_valley_hours = get_peak_valley_hours()  # 这里假设有个获取峰谷时段的函数
            if is_in_period(total_usage, peak_valley_hours, charge_type):
                total_cost += charge_rate * total_usage

    return total_cost

# 示例：阶梯电价每100度一个阶段，超出部分按基础价加收
tier_prices = [[0, 100], [101, 200], [201, float('inf')), basic_rate=0.6]

# 峰谷电附加费
additional_charges = {'peak': 0.8, 'valley': 0.4}

# 调用函数并传入电量
electricity_bill = calculate_bill(0.6, tier_prices, 150, additional_charges)