关于【每日一坑】记账本

jodie

crossin先生 发表于 2016-8-19 11:06
可以的

crossin先生，请问这个记账本哪里需要改进吗？

1. import os
   
2. import datetime
   
3. import re
   
4. 
   
5. class Account():
   
6.     def __init__(self):
   
7.         # 数据存储的文件路径
   
8.         self.file_name = "balance.txt"
   
9.         self.file_path = os.path.join(os.getcwd(), self.file_name)
   
10. 
    
11.         # 判断是否是首次记账，若否，则读出文件中的余额balance;若是，余额为0
    
12.         try:
    
13.             with open(self.file_path,"r", encoding="utf-8") as obj:
    
14.                 balance_string = obj.readline().strip("\n")
    
15.             self.balance = eval(balance_string)["balance"]
    
16.             self.is_first = 0
    
17.         except FileNotFoundError as e:
    
18.             self.balance = 0
    
19.             self.is_first = 1
    
20. 
    
21.     def keep_accounts(self,item,money):
    
22.         '''记账'''
    
23.         with open(self.file_path, "a", encoding = "utf-8") as obj:
    
24.             time = str(datetime.date.today())
    
25.             line = time + " " + money + " " + item + "\n"
    
26.             obj.write(line)
    
27. 
    
28.         self.balance += float(money)
    
29. 
    
30. 
    
31.     def check_the_balance(self):
    
32.         '''查询余额'''
    
33.         print(self.balance)
    
34. 
    
35. 
    
36.     def check_the_details(self):
    
37.         '''查询账单详情'''
    
38.         try:
    
39.             with open(self.file_path, "r", encoding = "utf-8") as obj:
    
40.                 details_list = obj.readlines()
    
41.                 for one in details_list:
    
42.                     if one.startswith("{"):
    
43.                         continue
    
44.                     print(one)
    
45.         except FileNotFoundError as e:
    
46.             print("尚未有收支记录！")
    
47. 
    
48. 
    
49. if __name__ == "__main__":
    
50.     account = Account()
    
51.     while True:
    
52.         choice = input("选择操作：\n\t1、记账\n\t2、查余额\n\t3、收支明细\n").strip()
    
53.         if choice == "1":
    
54.             money = input("金额： ").strip()
    
55.             item = input("名目： ").strip()
    
56.             account.keep_accounts(item,money)
    
57.         elif choice == "2":
    
58.             account.check_the_balance()
    
59.         elif choice == "3":
    
60.             account.check_the_details()
    
61.         elif choice == "leave":
    
62.             break
    
63. 
    
64. 
    
65.     with open(account.file_path,"r+", encoding ="utf-8") as obj:
    
66.         data = obj.read()
    
67. 
    
68.         # 若非首次记账，将新的余额替换旧的余额
    
69.         if account.is_first == 0:
    
70.             new_balance_string = '{{"balance":{}}}'.format(str(account.balance))
    
71.             data = re.sub(r'{"balance":.*?}',new_balance_string,data)
    
72.         # 若是首次记账，在文件头部记录每次记账结束后的余额
    
73.         else:
    
74.             new_balance_string = '{{"balance":{}}}\n'.format(str(account.balance))
    
75.             data = new_balance_string + data
    
76. 
    
77.         obj.seek(0)
    
78.         obj.write(data)

复制代码