Python制作记账程序

2023年12月9日发(作者：远大前程读后感)

-

Python制作记账程序

7.1 问题

创建脚本，要求如下：

假设在记账时，有一万元钱

无论是开销还是收入都要进行记账

记账内容包括时间、金额和说明等

记账数据要求永久存储

7.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可以

退出程序，遇到索引错误可以结束当次循环，重新开始选择选项。

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

7.3 步骤

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

步骤一：编写脚本

[root@localhost day05]# vim

#!/usr/bin/env python3

#日期　　开销　　收入　　余额　　备注

import time

import os

import pickle as p

def spend_money(record, wallet):

date = me('%Y-%m-%d')

amount = int(input('金额: '))

comment = input('备注: ')

with open(wallet, 'rb') as fobj:

#load从数据文件中读取数据，并转换为Python的数据结构

balance = (fobj) – amount

with open(wallet, 'wb') as fobj:

#dump将数据通过特殊形式转换为只有python语言认识的字符串，并写入文件

(balance, fobj)

with open(record, 'a') as fobj:

(

"%-15s%-8s%-8s%-10s%-20sn" %

(date, amount, 'n/a', balance, comment)

)

def save_money(record, wallet):

date = me('%Y-%m-%d')

amount = int(input('金额: '))

comment = input('备注: ')

comment = input('备注: ')

with open(wallet, 'rb') as fobj:

balance = (fobj) + amount

with open(wallet, 'wb') as fobj:

(balance, fobj)

with open(record, 'a') as fobj:

(

"%-15s%-8s%-8s%-10s%-20sn" %

(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 = (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 = '' # 记帐

wallet = '' # 记录余额

if not (wallet): #判断文件是否存在

with open(wallet, 'wb') as fobj:

(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

(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

(0) 记录开销

(1) 记录收入

(2) 查询收支记录

(3) 退出

请选择(0/1/2/3): ^C

Bye-bye

[root@localhost day05]# python3

(0) 记录开销

(1) 记录收入

(2) 查询收支记录

(3) 退出

请选择(0/1/2/3):

Bye-bye

-