基于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