《云计算全栈》-python篇：记账程序

案例7：记账程序
11.1 问题

创建account.py脚本，要求如下：

假设在记账时，有一万元钱
无论是开销还是收入都要进行记账
记账内容包括时间、金额和说明等
记账数据要求永久存储

11.2 方案

创建4个函数，分别实现记录开销、记录收入、查询收支、判断函数调用的四个方法，导入时间模块获取时间，导入os模块判断文件是否存在，导入pickle模块用来python特有类型与数据类型转换：

1.调用show_menu()函数后，先判断记录余额文件是否存在，如果不存在创建文件并写入余额，如果存在，利用while循环在交互端输出提示，请用户input0/1/2/3任意数值，如果输入的值不是0/1/2/3，打印输入值无效请重新输入并重新开始循环，如果输入的值是3，停止整个循环，如果输入的值是0/1/2通过字典键值对关联关系，调用相对应函数

2.如果输入的值是0，字典cmds中0键对应的值是spend_money，调用spend_money ()记录开销函数，让此函数实现获取当前系统日期、输入开销金额、输入开销备注信息、以二进制读方式打开记录余额文件计算本次开销后余额，以写方式打开记录余额文件将计算后开销余额写入文件，以追加方式打开记账文件，将日期、开销、备注、余额写入追加入记账文件最后

3.如果输入的值是1，字典cmds中0键对应的值是save_money，调用save_money ()记录收入函数，让此函数实现获取当前系统日期、输入收入金额、输入收入备注信息、以二进制读方式打开记录余额文件计算本次收入后余额，以写方式打开记录余额文件将计算后收入余额写入文件，以追加方式打开记账文件，将日期、开销、备注、余额写入追加入记账文件最后

4.如果输入的值是2，调用查询收支函数query ()，以二进制读方式打开记账文件，利用for循环遍历文件中数据，打印出来，打开记录余额文件读取余额并打印。

需要注意的是：为确保代码可以正常执行，while循环利用try except语句处理异常，优先匹配特殊异常，让用户按下Ctrl+C或Ctrl+D可以退出程序，遇到索引错误可以结束当次循环，重新开始选择选项。

将记录余额文件以及记账文件作为参数传入函数中
11.3 步骤

实现此案例需要按照如下步骤进行。

扫描二维码关注公众号，回复： 9686107 查看本文章

步骤一：编写脚本

[root@localhost day05]# vim account.py
#!/usr/bin/env python3
# 日期　　开销　　收入　　余额　　备注
import time
import os
import pickle as p
def spend_money(record, wallet):
    date = time.strftime('%Y-%m-%d')
    amount = int(input('金额: '))
    comment = input('备注: ')
    with open(wallet, 'rb') as fobj:
#load从数据文件中读取数据，并转换为Python的数据结构
        balance = p.load(fobj) – amount        
    with open(wallet, 'wb') as fobj:
           #dump将数据通过特殊形式转换为只有python语言认识的字符串，并写入文件
        p.dump(balance, fobj)
    with open(record, 'a') as fobj:
        fobj.write(
            "%-15s%-8s%-8s%-10s%-20s\n" %
            (date, amount, 'n/a', balance, comment)
        )
def save_money(record, wallet):
    date = time.strftime('%Y-%m-%d')
    amount = int(input('金额: '))
    comment = input('备注: ')
    with open(wallet, 'rb') as fobj:
        balance = p.load(fobj) + amount
    with open(wallet, 'wb') as fobj:
        p.dump(balance, fobj)
    with open(record, 'a') as fobj:
        fobj.write(
            "%-15s%-8s%-8s%-10s%-20s\n" %
            (date, 'n/a', amount, balance, comment)
        )
def query(record, wallet):
    with open(record) as fobj:
        for line in fobj:
            print(line, end='')
    with open(wallet, 'rb') as fobj:
          #load从数据文件中读取数据，并转换为Python的数据结构
        balance = p.load(fobj)
    print('当前余额: %s' % balance)
def show_menu():
    prompt = """(0) 记录开销
(1) 记录收入
(2) 查询收支记录
(3) 退出
请选择(0/1/2/3): """
    cmds = {'0': spend_money, '1': save_money, '2': query}
    record = 'record.txt'   # 记帐
    wallet = 'wallet.data'  # 记录余额
    if not os.path.exists(wallet):        #判断文件是否存在
        with open(wallet, 'wb') as fobj:
            p.dump(10000, fobj)
    while True:
        try:
            choice = input(prompt).strip()[0]
        except IndexError:
            continue
        except (KeyboardInterrupt, EOFError):
            print('\nBye-bye')
            choice = '3'
        if choice not in '0123':
            print('无效输入，请重试')
            continue
        if choice == '3':
            break
        cmds[choice](record, wallet)
if __name__ == '__main__':
    show_menu()

步骤二：测试脚本执行

[root@localhost day05]# python3 account.py
(0) 记录开销
(1) 记录收入
(2) 查询收支记录
(3) 退出
请选择(0/1/2/3): 0
金额: 2000
备注: huafei
(0) 记录开销
(1) 记录收入
(2) 查询收支记录
(3) 退出
请选择(0/1/2/3): 1
金额: 1000
备注: shouru
(0) 记录开销
(1) 记录收入
(2) 查询收支记录
(3) 退出
请选择(0/1/2/3): 2
2018-04-25     2000    n/a     28890     huafei              
2018-04-25     n/a     1000    29890     shouru              
当前余额: 29890
(0) 记录开销
(1) 记录收入
(2) 查询收支记录
(3) 退出
请选择(0/1/2/3): 3
[root@localhost day05]# python3 account.py
(0) 记录开销
(1) 记录收入
(2) 查询收支记录
(3) 退出
请选择(0/1/2/3): ^C
Bye-bye
[root@localhost day05]# python3 account.py
(0) 记录开销
(1) 记录收入
(2) 查询收支记录
(3) 退出
请选择(0/1/2/3): 
Bye-bye