关于【每日一坑】记账本 - Crossin的编程教室

#coding:gbk
one=1
two=2
three=3
running=1
while running:
print " ============"
print " |||记账本|||"
print '======================='
print "请回复数字可执行："
print "1.查询账单"
print "2.写账单"
print "3.退出"
number=int(raw_input())
print '======================='
if number==two:
date=str(raw_input("请输入日期,年月日: "))
m=str(raw_input("支出/存入"))
b=str(raw_input("备注:"))
o=open("jizhangben.txt","a")
o.write("\n"+date+" "+m+" "+b)
o.close()
elif number==one:
op=open("jizhangben.txt")
for i in op:
print i
op.close()
elif number==3:
running=0
print "已退出"
else:
print "没有这个功能"
if number==False:
print "You can not do this."
print '======================='
print
print

复制代码

import os
import datetime
import re
class Account():
def __init__(self):
# 数据存储的文件路径
self.file_name = "balance.txt"
self.file_path = os.path.join(os.getcwd(), self.file_name)
# 判断是否是首次记账，若否，则读出文件中的余额balance;若是，余额为0
try:
with open(self.file_path,"r", encoding="utf-8") as obj:
balance_string = obj.readline().strip("\n")
self.balance = eval(balance_string)["balance"]
self.is_first = 0
except FileNotFoundError as e:
self.balance = 0
self.is_first = 1
def keep_accounts(self,item,money):
'''记账'''
with open(self.file_path, "a", encoding = "utf-8") as obj:
time = str(datetime.date.today())
line = time + " " + money + " " + item + "\n"
obj.write(line)
self.balance += float(money)
def check_the_balance(self):
'''查询余额'''
print(self.balance)
def check_the_details(self):
'''查询账单详情'''
try:
with open(self.file_path, "r", encoding = "utf-8") as obj:
details_list = obj.readlines()
for one in details_list:
if one.startswith("{"):
continue
print(one)
except FileNotFoundError as e:
print("尚未有收支记录！")
if __name__ == "__main__":
account = Account()
while True:
choice = input("选择操作：\n\t1、记账\n\t2、查余额\n\t3、收支明细\n").strip()
if choice == "1":
money = input("金额： ").strip()
item = input("名目： ").strip()
account.keep_accounts(item,money)
elif choice == "2":
account.check_the_balance()
elif choice == "3":
account.check_the_details()
elif choice == "leave":
break
with open(account.file_path,"r+", encoding ="utf-8") as obj:
data = obj.read()
# 若非首次记账，将新的余额替换旧的余额
if account.is_first == 0:
new_balance_string = '{{"balance":{}}}'.format(str(account.balance))
data = re.sub(r'{"balance":.*?}',new_balance_string,data)
# 若是首次记账，在文件头部记录每次记账结束后的余额
else:
new_balance_string = '{{"balance":{}}}\n'.format(str(account.balance))
data = new_balance_string + data
obj.seek(0)
obj.write(data)

复制代码