import os
import pickle
from time import strftime
# 入账
def save(fname):
amount = int(input('金额:'))
comment = input('备注:')
date = strftime('%Y-%m-%d')
# 从文件中取出全部记录
with open(fname,'rb') as fobj:
recods = pickle.load(fobj)
# 计算最新余额
balance = recods[-1][-2] + amount
recod = [date,amount,0,balance,comment]
recods.append(recod)
# 将更新后的数据重新写入文件
with open(fname,'wb')as fobj:
pickle.dump(recods,fobj)
# 开销
def cost(fname):
amount = int(input('金额:'))
comment = input('备注:')
date = strftime('%Y-%m-%d')
# 从文件中取出全部记录
with open(fname, 'rb') as fobj:
recods = pickle.load(fobj)
# 计算最新余额
balance = recods[-1][-2] - amount
recod = [date, 0, amount, balance, comment]
recods.append(recod)
# 将更新后的数据重新写入文件
with open(fname, 'wb')as fobj:
pickle.dump(recods, fobj)
#查询
def query(fname):
with open(fname,'rb') as fobj:
recods = pickle.load(fobj)
print("请选择查询条件:1今日明细/2每周明细/3每月明细/4年度明细:")
m=int(input())
print('%-11s%-8s%-8s%-12s%-20s' % ('时间','入账','开销','余额','备注'))
a=0
b=0
if m==1:
for i in recods:
if str(i[0])==strftime('%Y-%m-%d'):
print('%-12s%-10s%-10s%-12s%-20s'%tuple(i))
a=a+int(i[1])
b=b+int(i[2])
print(strftime('%Y-%m-%d'),"总入账:",str(a),"总出账:",str(b))
x=0
y=0
if m==3:
print("请选择月份(01/02/03....10/11/12):")
n=str(input())
for i in recods:
if str(i[0][5:7])==n:
print('%-12s%-10s%-10s%-12s%-20s'%tuple(i))
x=x+int(i[1])
y=y+int(i[2])
print(n,"月份总入账:",str(x),"总出账:",str(y))
if m==4:
print("请选择年份(2021/2022/2023....2099):")
n=str(input())
for i in tuple(recods):
if str(i[0][0:4])==n:
print('%-12s%-10s%-10s%-12s%-20s'%tuple(i))
x=x+int(i[1])
y=y+int(i[2])
print(n,"年度总入账:",str(x),"总出账:",str(y))
if m==2:
print("请选择周数(01/02/03....53/54/55):","本周周数:",strftime("%W"))
n=str(input())
for i in recods:
if strftime("%W")==n:
print('%-12s%-10s%-10s%-12s%-20s'%tuple(i))
x=x+int(i[1])
y=y+int(i[2])
print(n,"周总入账:",str(x),"总出账:",str(y))
# 显示
def show_menu():
prompt = ''' 0 入账
1 开销
2 查询
3 退出
请选择(0/1/2/3):'''
cmds = {'0': save,'1': cost,'2': query}
fname = 'account.data'
init_data = [
[strftime('%Y-%m-%d') , 0, 0, 100000, 'init data']
]
# 如果文件不存在,把初始化数据写进去
if not os.path.exists(fname):
with open(fname,'wb') as fobj:
pickle.dump(init_data,fobj)
while True:
choice = input(prompt).strip()
if choice not in ['0','1','2','3']:
print('无效输入')
continue
if choice == '3':
print('\nBye~~')
break
cmds[choice](fname)
if __name__ == '__main__':
show_menu()
python记账,改编自yuan156941828的文章《基于Python的模拟记账功能》,增加了每日、周、月年明细查询
最新推荐文章于 2024-03-01 11:28:29 发布