使用python编写一个计算快递费的程序

某快递公司根据投送目的地距离公司的远近，将全国划分成5个区域：
快递费按邮件重量计算，由起重费用、续重费用两部分构成，具体计算方法如下：
（1）起重（首重）1公斤按起重资费计算（不足1公斤，按1公斤计算），超过首重的重量，按公斤（不足1公斤，按1公斤计算）收取续重费；
（2）同城起重资费12元，续重3元/公斤；
（3）寄往1区的邮件，起重资费12元，续重4元；
（4）寄往其他地区的邮件，起重资费统一为15元。而续重部分，不同区域价格不同：2区的续重5元/公斤，3区的续重6.5元/公斤，4区的续重10元/公斤。
编写程序，从键盘输入邮件的目的区域编码和重量，计算并输出运费，计算结果保留2位小数（输入区域为0，重量为-1时退出系统）。
【注意】本实验功能用函数实现。

望采纳，谢谢

def cal(code, weight):
    if code == '0':
        if weight <=1:
            return 12.0
        else:
            return 12.0 + (weight-1) * 3.0
    elif code == '1':
        if weight <=1:
            return 12.0
        else:
            return 12.0 + (weight-1) * 4.0
    else:
        if weight <=1:
            return 15.0
        else:
            if code =='2':
                return 12.0 + (weight-1) * 5.0
            elif code =='3':
                return 12.0 + (weight-1) * 6.5
            elif code =='4':
                return 12.0 + (weight-1) * 10.0

while True:
    code,weight = input('区域编码（0为同城），重量：(输入区域为0，重量为-1时退出系统)').split()
    if code == '0' and weight == '-1':
        break
    res = cal(code, float(weight))
    print('运费为{:.2f}'.format(res))

import math

def fee(area, quan):
    deliver = {1:[12,3],2:[15,5],3:[15,6.5],4:[15,10],5:[12,3]}
    return deliver[area][0]+(math.ceil(quan)-1)*deliver[area][1]

while True:
    area, quan = input().split()
    if area=='0' or quan == '-1':break
    if area =='5':
        print(f'同城投送{quan}公斤包裹的费用为：{fee(int(area),eval(quan))}元')
    else:
        print(f'投送{quan}公斤包裹前往{area}区的费用为：{fee(int(area),eval(quan))}元')