记账本+数据库

尝试运行了以下别人发的记账本代码，自己找了数据库相关的
可运行

import sqlite3

# 连接到SQLite数据库
# 数据库文件是mrsoft.db 如果文件不存在，会自动在当前目录中创建

conn = sqlite3.connect('database.db')

# 创建一个cursor
cursor = conn.cursor()

# 使用游标对象执行SQL语句，创建user表
cursor.execute('create table accounting (name varchar(20) primary key,price int(10), type varchar(20))')

# 关闭游标
cursor.close()

# 关闭Connection
conn.close()

把这个提前运行就可以再继续运行下面的记账本了

from tkinter import ttk

from tkinter import *

import sqlite3

class Accounting:

  #数据库名

  db_name='database.db'

  #初始化

  

  def __init__(self, window):

    self.win=window

    self.win.title('简单记账本')



    #创建容器

    frame=LabelFrame(self.win,text='日常收支记录')

    frame.grid(row=0,column=0,columnspan=3,pady=20)



    #输入项目

    Label(frame,text='项目').grid(row=1,column=0)

    self.name=Entry(frame)

    self.name.focus()

    self.name.grid(row=1,column=1)

    Label(frame,text='费用').grid(row=2,column=0)

    self.price=Entry(frame)

    self.price.focus()

    self.price.grid(row=2,column=1)

    Label(frame,text='类别').grid(row=3,column=0)

    self.type=ttk.Combobox(frame,values=['收入','支出'],state='readonly')

    self.type.grid(row=3,column=1)

    self.type.current(0)

    Button(frame,text='保存',command=self.add_accounting).grid(row=4,columnspan=2,sticky=W+E)

    #数据操作结果提示信息

    self.message=Label(text='',fg='red')

    self.message.grid(row=3,column=0,columnspan=2,sticky=W+E)



    #treeview进行表格信息显示

    self.tree=ttk.Treeview(height=10,columns=("#0","#1","#2"))

    self.tree.grid(row=4,column=0,columnspan=3)

    self.tree.heading("#0",text='项目',anchor='center')

    self.tree.heading("#1",text='费用',anchor='center')

    self.tree.heading("#2",text='类型',anchor='center')



    ttk.Button(text='删除',command=self.delete_accounting).grid(row=5,column=0,sticky=W+E)

    ttk.Button(text='编辑',command=self.edit_accounting_win).grid(row=5,column=1,sticky=W+E)

    self.get_accounting()
#数据库操作方法

  def run_query(self,query,params=()):

    with sqlite3.connect(self.db_name) as conn:

      cursor=conn.cursor()

      result=cursor.execute(query,params)

      conn.commit()

      return result

  #数据完整性验证

  def validation(self):

    return len(self.name.get())!=0 and len(self.price.get())!=0
#数据添加处理

  def add_accounting(self):

    if self.validation():

      query="insert into accounting(name,price,type) values(?,?,?)"

      params=(self.name.get(),self.price.get(),self.type.get())

      self.run_query(query,params)

      self.message['text']="{}添加成功".format(self.name.get())

      self.get_accounting()



    else:

      self.message['text']='项目和费用不能为空'
#显示所有项目

  def get_accounting(self):

    #重绘表格内容

    records=self.tree.get_children()

    for element in records:

      self.tree.delete(element)

    query="select * from accounting order by type"

    db_rows=self.run_query(query).fetchall()

    print(db_rows)

    # #填充表格

    for row in db_rows:

      self.tree.insert("",0,text=row[0],values=(row[1],row[2]))
#删除项目

  def delete_accounting(self):

    self.message['text']=''

    try:

      self.tree.item(self.tree.selection())['text'][0]



    except IndexError as e:

      self.message['text']='请选择操作项目'

      return

    self.message['text']=''

    name=self.tree.item(self.tree.selection())['text']

    query='delete from accounting where name=?'

    self.run_query(query,(name,))

    self.message['text']='{}删除成功'.format(name)

    self.get_accounting()
#修改的窗体实现

  def edit_accounting_win(self):

    self.message['text']=''

    try:

      self.tree.item(self.tree.selection())['text'][0]



    except IndexError as e:

      self.message['text']='请选择操作项目'

      return

    old_name=self.tree.item(self.tree.selection())['text']

    old_price=self.tree.item(self.tree.selection())['values'][0]

    old_type=self.tree.item(self.tree.selection())['values'][1]

    self.edit_win=Toplevel()

    self.edit_win.title='编辑项目'

    #原来的项目

    Label(self.edit_win,text='原来的项目').grid(row=0,column=1)

    Entry(self.edit_win,textvariable=StringVar(self.edit_win,value=old_name),state='readonly').grid(row=0,column=2)



    Label(self.edit_win,text='新的项目').grid(row=1,column=1)

    new_name=Entry(self.edit_win)

    new_name.grid(row=1,column=2)



    Label(self.edit_win,text='原来费用').grid(row=2,column=1)

    Entry(self.edit_win,textvariable=StringVar(self.edit_win,value=old_price),state='readonly').grid(row=2,column=2)



    Label(self.edit_win,text='新的费用').grid(row=3,column=1)

    new_price=Entry(self.edit_win)

    new_price.grid(row=3,column=2)



    Label(self.edit_win,text='类别').grid(row=4,column=1)

    new_type=ttk.Combobox(self.edit_win,values=['收入','支出'],state='readonly')

    new_type.grid(row=4,column=2)

    new_type.set(old_type)



    Button(self.edit_win,text='更新',command=lambda:self.edit_record(new_name.get(),old_name,new_price.get(),old_price,new_type.get(),old_type)).grid(row=5,column=2,sticky=W+E)



    self.edit_win.mainloop()
#数据修改处理

  def edit_record(self,new_name,old_name,new_price,old_price,new_type,old_type):

    query="update accounting set name=?,price=?,type=? where name=? and price=? and type=?"

    params=(new_name,new_price,new_type,old_name,old_price,old_type)

    self.run_query(query,params)

    self.edit_win.destroy()

    self.message['text']="{}更新成功".format(old_name)

    self.get_accounting()
if __name__ == '__main__':

  window=Tk()

  application=Accounting(window)

  window.mainloop()