关于【每日一坑】记账本

灰机

我想知道那个记账本的小程序怎么写的，不知道有没有可以告诉我？

crossin先生

jodie 发表于 2020-10-29 23:24
crossin先生，请问这个记账本哪里需要改进吗？

挺好的。
对输入内容可以做一下校验，用户有可能手误或者故意输入一些乱七八糟的数据

然后eval这个东西，虽然用起来爽，但很危险，它会把文字当做代码来执行。如果有人看到你这个代码，可以通过在数据文件里加一些恶意代码来做各种操作。如果你想存储数据，可以用json

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)

复制代码

feifei

新手打卡，crossin先生，请指点
import csv
import datetime

def keepAccounts():
    f=open('1.csv','a',encoding='utf-8',newline=''"")
    keeptime=datetime.date.today()
    mingmu=input("名目:")
    money=int(input("金额:"))
    csv_writer=csv.writer(f)
    csv_writer.writerow([keeptime,mingmu,money])
    f.close()
def checkAccount():
    f = open('1.csv', 'r',encoding='utf-8')
    csv_reader=csv.reader(f)
    sum1=0
    for i in csv_reader:
        #print (i[2])
        #print (type(i[2]))
        sum1=sum1+int(i[2])
    print (sum1)
def printAccount():
    f = open('1.csv', 'r', encoding='utf-8')
    csv_reader = csv.reader(f)
    for i in csv_reader:
        print (','.join(i))
while True:
    a = input("选择操作：\n1.记账\n2.查余额\n3.收支明细\n")
    if a=='1':
        keepAccounts()
    elif a=='2':
        checkAccount()
    elif a=='3':
        printAccount()
    else:
        print ('退出记账本')
        break

crossin先生

灰机 发表于 2016-8-18 20:25
还行吧？

可以的

灰机

crossin先生 发表于 2016-8-17 11:18
可以参考 写文件 和 用文件保存游戏 这几课的内容，在此基础上改进下程序流程。
先从最简单的，一个记录功 ...

还行吧？

灰机

谢谢crossin先生，我已经写了出来了原创的，新鲜出炉了

1. #coding:gbk
   
2. 
   
3. one=1
   
4. two=2
   
5. three=3
   
6. running=1
   
7. while running:
   
8.     print "      ============"
   
9.     print "      |||记账本|||"
   
10.     print '======================='
    
11.     print "请回复数字可执行："
    
12.     print "1.查询账单"
    
13.     print "2.写账单"
    
14.     print "3.退出"
    
15.     number=int(raw_input())
    
16.     print '======================='
    
17.     if number==two:
    
18.         date=str(raw_input("请输入日期,年月日: "))  
    
19.         m=str(raw_input("支出/存入"))
    
20.         b=str(raw_input("备注:"))
    
21.         o=open("jizhangben.txt","a")
    
22.         o.write("\n"+date+"   "+m+"   "+b)
    
23.         o.close()
    
24. 
    
25.     elif number==one:
    
26.         op=open("jizhangben.txt")
    
27.         for i in op:
    
28.             print i
    
29.         op.close()
    
30.     elif number==3:
    
31.         running=0
    
32.         print "已退出"
    
33.     else:
    
34.         print "没有这个功能"
    
35. 
    
36. 
    
37.     if number==False:
    
38.         print "You can not do this."
    
39.     print '======================='
    
40.     print
    
41.     print
    

复制代码

crossin先生

可以参考 写文件 和 用文件保存游戏 这几课的内容，在此基础上改进下程序流程。
先从最简单的，一个记录功能开始写，然后不断完善功能