Создать упрощенную CRM - системы. База продажника

from PyQt5.QtWidgets import QApplication, QWidget, QGridLayout, \
QTableWidget,QTableWidgetItem, QTableView, QTextEdit, QLineEdit, \
QPushButton, QDialog, QInputDialog, QMessageBox, QMenu
from PyQt5.QtGui import QFont
from PyQt5.QtSql import QSqlDatabase, QSqlQueryModel, QSqlQuery
import os
 
def initdb(dbname):
    db = QSqlDatabase.addDatabase('QSQLITE','default')
    db.setDatabaseName(dbname)
    if not db.open():
        print(db.lastError().text())
        return False
    query = QSqlQuery(db)
    query.exec_("drop table if exists pays")
    query.exec_(
    "create table if not exists pays"
    "(date datetime,"
    "name text,"
    "pay numeric default 0)")
    db.close()
    db.removeDatabase('default')
    return True
    
class EDialog(QDialog):
    def __init__(self,parent=None):
        super().__init__(parent)
        self.edit = QTextEdit()
        self.btnOk = QPushButton('Ok')
        self.btnCancel = QPushButton('Cancel')
        grid = QGridLayout(self)
        grid.setContentsMargins(0,0,0,0)
        grid.addWidget(self.edit,0,0,4,4)
        grid.addWidget(self.btnOk,4,0,1,1)
        grid.addWidget(self.btnCancel,4,1,1,1)
        self.btnOk.clicked.connect(self.accept)
        self.btnCancel.clicked.connect(self.reject)
    
    def get(self):
        txt = self.edit.toPlainText().split("\n")
        out = [i for i in txt if i.strip()]
        return out
    
    def setText(self,s=None):
        self.edit.insertPlainText(s)
    
    def setHtml(self,s=None):
        self.edit.setReadOnly(True)
        self.edit.setHtml(s)
    
        
class TDialog(QDialog):
    def __init__(self,parent=None,v_headers=None,h_headers=None,data=None):
        super().__init__(parent)
        self.v_headers = v_headers
        self.h_headers = h_headers
        self.data = data
        self.table = QTableWidget()
        self.table.setRowCount(len(v_headers))
        self.table.setColumnCount(len(h_headers))
        self.table.setVerticalHeaderLabels(v_headers)
        self.table.setHorizontalHeaderLabels(h_headers)
        self.table.horizontalHeader().setSectionResizeMode(1)
        self.table.verticalHeader().setSectionResizeMode(1)
        if self.data:
            for i in range(self.table.rowCount()):
                item = QTableWidgetItem()
                item.setText(self.data[i])
                self.table.setItem(i,0,item)
        self.btnOk = QPushButton('Ok')
        self.btnCancel = QPushButton('Cancel')
        grid = QGridLayout(self)
        grid.setContentsMargins(0,0,0,0)
        grid.addWidget(self.table,0,0,4,4)
        grid.addWidget(self.btnOk,4,0,1,1)
        grid.addWidget(self.btnCancel,4,1,1,1)
        self.btnOk.clicked.connect(self.accept)
        self.btnCancel.clicked.connect(self.reject)
    def get(self):
        out = []
        for i in range(self.table.rowCount()):
            try:
                txt = self.table.item(i,0).text()
                if txt.strip(): out.append([self.v_headers[i],txt])
            except: pass
        return out
        
class View(QWidget):
    def __init__(self):
        super().__init__()
        self.db = None
        self.model = QSqlQueryModel()
        self.view = QTableView()
        self.view.setModel(self.model)
        self.view.setSelectionMode(1)
        self.view.horizontalHeader().setSectionResizeMode(1)
        self.btnAdd = QPushButton('Добавить запись')
        self.btnClean = QPushButton('Очистить базу данных')
        self.btnAll = QPushButton('Итог')
        grid = QGridLayout(self)
        grid.setContentsMargins(0,0,0,0)
        grid.addWidget(self.view,0,0,8,8)
        grid.addWidget(self.btnClean,10,0,1,1)   
        grid.addWidget(self.btnAdd,10,1,1,1)
        grid.addWidget(self.btnAll,10,2,1,1)
        self.setFont(QFont('Arial',16))   
        self.btnAdd.clicked.connect(self.add_record)
        self.btnClean.clicked.connect(self.clean_db)
        self.btnAll.clicked.connect(self.all_pays)
        self.open_db()
        self.on_query()
        
    def open_db(self):
        os.chdir(os.getcwd())
        dbname = "payments.db"
        try: f = open(dbname)
        except:
            QMessageBox.critical(None, "","File not found",Qt.QMessageBox.Cancel)
            exit(1)
        else: f.close()
        self.db = QSqlDatabase.addDatabase('QSQLITE')
        self.db.setDatabaseName(dbname)
        if not self.db.open():
            QMessageBox.critical(None, "","Cannot open database",QMessageBox.Cancel)
            exit(1)
 
    def on_query(self):
        sql = '''select rowid as "ID",
        date as "Дата",
        name as "Имя",
        pay as "Платеж" from pays where pay'''
        self.model.setQuery(sql,self.db)
        self.view.hideColumn(0)
        
    def all_pays(self):
        q = QSqlQuery(self.db)
        if not q.exec_('select name,sum(pay) from pays group by name'):
            QMessageBox.critical(None,"Ошибка БД",q.lastError().text(),QMessageBox.Cancel)
            return
        pays = []
        while(q.next()): pays.append([q.value(0),q.value(1)])
        allpays = sum([i[1] for i in pays])
        midpay = round(allpays/len(pays),2)
        for i in pays: i.append(round(i[1] - midpay,2))
        s = '<table width=100% border=0.5>'
        s += "<tr><th></th><th><i>Всего: " + str(allpays) +"</i></th><th><i>Среднее: "+ str(midpay) +"</i></th></tr>"
        for i in pays:
            s += "<tr><td>" + i[0] + "</td><td>"+ str(i[1]) + "</td>"
            v = i[2]
            if v >=0: s += "<td><font color=green>+" + str(v)+"</font></td>"
            else:s += "<td><font color=red>" + str(v)+"</font></td>"
        s += '</tr></table>'
        d = EDialog(self)
        d.resize(600,200)
        d.setFont(self.font())
        d.setWindowTitle("Итог")
        d.setHtml(s)
        d.exec_()
    
    def add_record(self):
        q = QSqlQuery(self.db)
        q.exec_("select name from pays group by name")
        out = []
        while q.next(): out.append(q.value(0))
        d = TDialog(self,out,("Платеж",))
        d.setFont(self.font())
        d.resize(480,240)
        d.setWindowTitle("Добавить запись")
        if not d.exec_(): return
        for i in d.get():
            q = QSqlQuery('insert into pays (date,name,pay) values(date("now"),?,?)',self.db)
            q.bindValue(0,i[0])
            q.bindValue(1,i[1])
            if not q.exec_():
                print(q.lastError().text())
        self.on_query()
             
        
    def clean_db(self):
        val,ok = QInputDialog.getText(self,"","Введите пароль администратора",QLineEdit.Password)
        if not ok: return
        if val != 'sql': 
            QMessageBox.critical(self, "","Неверный пароль",QMessageBox.Cancel)
            return
        q = QSqlQuery()
        q.exec_('select distinct name from pays')
        out = ''
        while q.next():
            r  = q.record()
            out += q.value(0) + '\n'        
        d = EDialog(self)
        d.setWindowTitle("Очистить базу данных")
        d.setFont(self.font())
        d.setText(out)
        ret = d.exec_()
        if not ret: return
        names = d.get()
        self.model.setQuery('delete from pays',self.db)
        for i in names:
            query = QSqlQuery(self.db)
            query.prepare('insert into pays (date,name) values(date("now"),?)')
            query.bindValue(0,i)
            if not query.exec_():
                print(query.lastError().text())
        self.db.commit()
        self.on_query()
    
    def contextMenuEvent(self,e):
        pos =  self.view.mapToGlobal(e.pos())
        row  = self.view.currentIndex().row()
        rowid = self.view.model().index(row,0).data()
        #print(row,rowid)
        mnu = QMenu()
        mnu.addSeparator()
        mnu.addAction('Удалить запись').setObjectName('del')
        mnu.addSeparator()
        mnu.addAction('Редактировать запись').setObjectName('edit')
        mnu.addSeparator()
        ret = mnu.exec_(pos)
        if not ret:
            self.view.selectionModel().clear()
            return
        obj = ret.objectName()
        if obj == 'del':
            self.model.setQuery('delete from pays where rowid=='+ str(rowid))
        elif obj == 'edit':
            q = QSqlQuery(self.db)
            if not q.exec_('select name,pay from pays where rowid==' + str(rowid)):
                print(q.lastError().text())
                return
            while q.next(): k,v = q.value(0),str(q.value(1))
            t = TDialog(self,(k,),('Платеж',),(v,))
            t.setFont(self.font())
            t.setWindowTitle("Редактировать запись")
            t.resize(300,80)
            if not t.exec_(): 
                self.view.selectionModel().clear()
                return
            pay  = t.get()[0][1]
            q = QSqlQuery(self.db)
            if not q.exec_('update pays set pay=' + repr(pay) +' where rowid==' + str(rowid)):
                print(q.lastError().text())
        self.db.commit()
        self.on_query()
        
if __name__=="__main__":
    app = QApplication([])
    w = View()
    w.resize(800,600)
    w.move(0,0)
    w.show()
    app.exec_()