案例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 步骤
实现此案例需要按照如下步骤进行。
步骤一:编写脚本
[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