这个 怎么写 用循环 还有function

输入这个月的电表数 上月电表数是1603 返回 本月电费金额：** 用电量：** 本月电表数：**
10度（含） 0.5元/度
10-20度（含） 0.85元/度
20以上 1元/度

用循环 加函数

def count_electricity():
    flag = True
    count = 0
    while flag != 'q':
        number = input("输入电表读读数(按q退出）:")

        try:
            number = float(number)
            if number <= 10:
                count = number * 0.5
            elif 10 < number <= 20:
                count = number * 0.85
            elif number > 20:
                count = number
            print("本月电费金额:", count)
            print("用电量:", number)
            print("本月电表数:", number)
        except ValueError:
            print("请重新输入:")


count_electricity()