python记账,改编自yuan156941828的文章《基于Python的模拟记账功能》，增加了每日、周、月年明细查询

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()