Python记账本#

1
import json
2
import datetime
3
import PySimpleGUI as sg
4
import pandas as pd
5
import os
6
import xlwt
7
# import openpyxl
8

9
#读取数据
10
def read_data():
11
    try:
12
    with open(r"data.txt", "r") as f:
13
    json_data=f.read()
14
    datalist=json.loads(json_data)
15

16
        return datalist
17
    except:
18
        return []#当文本里面没有数据时返回空列表
19

20
#写入数据
21
def write_data(datalist):
22
    with open(r"data.txt", "w") as f:
23
    f.write(json.dumps(datalist, ensure_ascii=False))#防止改变中文编码格式
24

25
#显示数据
26
def show_data():
27
    try:
28
        data = read_data()
29
        datalist = []
30
        for i in data:
31
            if i["分类"] == "支出":
32
                data_list = [i["时间"], i["项目"], i["金额"] * -1, i["分类"]]
33
                datalist.append(data_list)
34
            else:
35
                data_list = [i["时间"], i["项目"], i["金额"], i["分类"]]
36
                datalist.append(data_list)
37
                return datalist
38
            except:
39
                return []
40

41
#总收入支出  1为收入0为支出
42
def amount_sum(cla):
43
    try:
44
        data = read_data()
45
        sum = 0
46
        if cla == 1:
47
            for i in data:
48
                if i["分类"] == "收入":
49
                    sum += i["金额"]
50
        if cla == 0:
51
            for i in data:
52
                if i["分类"] == "支出":
53
                    sum += i["金额"]
54
        return sum
55
    except:
56
        return 0#当文本里面没有数据时返回0
57

58
#添加数据
59
def add_data(project, amount, cla):
60
    datalist=read_data()
61
    time=datetime.datetime.now().strftime("%Y/%m/%d")
62
    data={"时间": time, "项目": project, "金额": amount, "分类": cla}
63
    datalist.append(data)
64
    write_data(datalist)
65
    sg.popup_auto_close('项目录入成功')  # 弹窗
66

67
#删除数据
68
def delete_data(index):
69
    datalist=read_data()
70
    del datalist[index]
71
    write_data(datalist)
72
    sg.popup_auto_close('删除成功')
73

74
#导出Excel文件
75
def export(path):
76
    with open(r"data.txt", "r") as f:
77
        data=f.read()
78
        f.close()
79
    with open(r"data.json", "w", encoding='utf-8') as f:
80
        f.write(data)
81
        f.close()
82
    json_data=pd.read_json(r"data.json")
83
    json_data.to_excel(fr'{path}/{datetime.datetime.now().strftime("%Y%m%d")}.xls')
84
    os.remove(r"data.json")
85
    sg.popup("导出成功")
86

87
if __name__ == '__main__':
88
    List=show_data()
89
    #图形界面布局列表
90
    sg.theme('LightBrown')
91
    layout=[
92
        [sg.T("账目列表：")]+[sg.B("导出Excel文件", key="-export-", pad=((390, 0), (0, 0)))],
93
        [sg.Table(List,
94
                headings=["时间", "项目", "金额", "分类"],
95
                key="-show_table-",
96
                justification="c",
97
                auto_size_columns=False,
98
                def_col_width=15,)],
99
        [sg.T(f"总收入{amount_sum(1)}元  总支出{amount_sum(0)}元  结余:{amount_sum(1)-amount_sum(0)}元", key="-show_data-")],
100
        [sg.T("账单项目："), sg.In(key="-project-")],
101
        [sg.T("账单金额："), sg.In(key="-amount-")],
102
        [sg.T("项目分类：")] + [sg.Radio(i, group_id=1, key=i) for i in ["收入", "支出"]],#单选框
103
        [sg.B("提交项目", key="-submit-")]+[sg.B("删除项目", key="-remove-", pad=((430, 0), (0, 0)))]
104

105
        ]
106

107
    window=sg.Window("记账本", layout, font='宋体 13')
108

109
    while True:
110
        event, values =window.read()#获取用户事件
111
        if event=="-submit-":#用户触发提交数据
112

113
            #判断用户是否输入完整数据
114
            if values["-project-"]=='' or values["-amount-"]=='' or (values["收入"]==False and values["支出"]==False):
115
                sg.popup("请填写完整的数据")
116
                continue
117
            #判断用户输入金额是否为数字
118
            if not str.isdigit(values["-amount-"]):
119
                sg.popup("请输入正确的金额")
120
                continue
121
            # 获取输入框中的值
122
            project=values["-project-"]
123
            amount=float(values["-amount-"])
124

125
            for k, v in values.items():
126
                if v == True:
127
                    cla=k
128
                    add_data(project, amount, cla)
129
                    List=show_data()
130
                    text=f"总收入{amount_sum(1)}元  总支出{amount_sum(0)}元  结余:{amount_sum(1)-amount_sum(0)}元"
131
                    window["-show_table-"].update(values=List)
132
                    window["-show_data-"].update(value=text)
133
                    #清空输入栏
134
                    window["-project-"].update(value="")
135
                    window["-amount-"].update(value="")
136
                    window["收入"].ResetGroup()
137

138
        if event=="-remove-":#用户触发删除数据
139
            #判断用户是否选中数据
140
            if values["-show_table-"]:
141
                #确认删除弹窗
142
                confirm = sg.popup_ok_cancel("确认删除选中的项目")
143
                if confirm == "OK":
144
                    delete_data(values["-show_table-"][0])#删除数据
145
                    #更新表格
146
                    List = show_data()
147
                    text = f"总收入{amount_sum(1)}元  总支出{amount_sum(0)}元  结余:{amount_sum(1) - amount_sum(0)}元"
148
                    window["-show_table-"].update(values=List)
149
                    window["-show_data-"].update(value=text)
150
            else:
151
                sg.popup("请选中一个项目")
152

153
        if event=="-export-":#导出Excel文件
154
            path=sg.popup_get_folder("保存路径:")
155

156
            if path =='':#判断路径是否为空
157
                sg.popup("请选择正确的路径")
158
                continue
159
            elif path is None:#防止用户因取消路径选择而关闭程序
160
                continue
161
            else:
162
                export(path)
163

164
        if event in (None, 'Exit'):
165
            break
166
    window.close()