data-manipulation/gui.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-

import sys
from PyQt5.QtWidgets import (QWidget, QHBoxLayout, QFrame,
    QSplitter, QStyleFactory, QApplication)
from PyQt5.QtCore import Qt
#from PyQt5.QtWidgets import QMainWindow, QTextEdit, QAction, QApplication, QListView, QListWidget, QTableWidget , QTableView, QTableWidgetItem, QHeaderView, QPushButton, QMenu
from PyQt5.QtGui import QIcon
from PyQt5.QtWidgets import *
#import PyQt5.QtGui
from PyQt5.QtSql import *

#import pymysql
import dm

# conn = pymysql.connect(
# db='ats',
# user='dba',
# passwd='AlsprofilinE',
# host='kis',
# charset='utf8')
# c = conn.cursor()

class MainWin(QMainWindow):

    def __init__(self):
        super().__init__()

        self.initUI()
    #заглушка
    def qqqq(self):
        print("Нажата клавиша DEL")
        #tablesDataWidget.setCellWidget(0, 0, edit)
        #callEdit = tablesDataWidget.cellWidget(0, 0)
        #callEdit.text()

    def initUI(self):

        textEdit = QTextEdit()
        splitter = WorkArea()
        self.setCentralWidget(splitter)
        #self.setCentralWidget(textEdit)

        newAction = QAction(QIcon('img/new.gif'), 'Добавить', self)
        newAction.setShortcut('Ins')
        newAction.setStatusTip('Добавить')
        newAction.triggered.connect(self.qqqq)

        deleteAction = QAction(QIcon('img/delete.gif'), 'Удалить', self)
        deleteAction.setShortcut('Del')
        deleteAction.setStatusTip('Удалить')
        deleteAction.triggered.connect(self.qqqq)

        exitAction = QAction(QIcon('img/exit.gif'), 'Выход', self)
        exitAction.setShortcut('Ctrl+Q')
        exitAction.setStatusTip('Выход')
        exitAction.triggered.connect(self.close)

        cutAction = QAction(QIcon('img/cut.gif'), 'Вырезать', self)
        cutAction.setShortcut('Ctrl+X')
        cutAction.setStatusTip('Вырезать')
        #cutAction.triggered.connect(self.close)

        copyAction = QAction(QIcon('img/copy.gif'), 'Копировать', self)
        copyAction.setShortcut('Ctrl+С')
        copyAction.setStatusTip('Копировать')
        #copyAction.triggered.connect(self.close)

        pasteAction = QAction(QIcon('img/paste.gif'), 'Вставить', self)
        pasteAction.setShortcut('Ctrl+V')
        pasteAction.setStatusTip('Вставить')
        #pasteAction.triggered.connect(self.close)

        findAction = QAction(QIcon('img/find.gif'), 'Копировать', self)
        findAction.setShortcut('Ctrl+F')
        findAction.setStatusTip('Искать')
        # findAction.triggered.connect(self.close)

        printAction = QAction(QIcon('img/print.gif'), 'Печатать', self)
        printAction.setShortcut('Ctrl+P')
        printAction.setStatusTip('Печатать')
        # printAction.triggered.connect(self.close)


        self.statusBar()

        menubar = self.menuBar()
        fileMenu = menubar.addMenu('&Файл')
        fileMenu.addAction(newAction)
        fileMenu.addAction(deleteAction)
        fileMenu.addAction(printAction)
        fileMenu.addAction(exitAction)

        editMenu = menubar.addMenu('&Редактирование')
        editMenu.addAction(copyAction)
        editMenu.addAction(cutAction)
        editMenu.addAction(pasteAction)
        editMenu.addAction(printAction)

        helpMenu = menubar.addMenu('&Помощь')

        #toolbar = self.addToolBar('Редактирование')

        toolbar = self.addToolBar('Панель инструментов')
        toolbar.addAction(newAction)
        toolbar.addAction(deleteAction)
        toolbar.addAction(copyAction)
        toolbar.addAction(cutAction)
        toolbar.addAction(pasteAction)
        toolbar.addAction(findAction)
        toolbar.addAction(printAction)
        #toolbar.addAction(exitAction)

        self.setGeometry(300, 300, 850, 650)
        self.setWindowTitle('Ацкый быдлокод')
        self.show()

class MyTable(QTableWidget):
    def __init__(self, tbl, thestruct, *args):
        QTableWidget.__init__(self, *args)
        self.data = thestruct
        self.setmydata()
        self.tbl = tbl
        self.resizeColumnsToContents()
        self.horizontalHeader().setSortIndicatorShown(True)
        self.horizontalHeader().setStretchLastSection(True)
        #self.horizontalHeader().stretchLastSection()
        self.horizontalHeader().setCascadingSectionResizes(True)
        # равномерное изменение ширины столбцов
        #self.horizontalHeader().setSectionResizeMode(1)
        # изменение ширины столбцов по размеру текста
        self.horizontalHeader().setSectionResizeMode(3)
        self.horizontalHeader().setStyleSheet("color: blue;")

        #print(self.tbl)
        n = 0
        for item in dm.getTablesStructure(self.tbl):
            # установка заголовков столбцов таблицы
            self.setHorizontalHeaderItem(n, QTableWidgetItem(item))
            n = n+1

        # обработка нажатия мышой на таблице
        self.doubleClicked.connect(self.on_click)
       # self.doubleClicked.connect(self.contextMenuEvent(self.cursor()))

    # добавлем кнопку в ячейку
        #self.clicked.connect(self.addButtonToCell())
    def addButtonToCell(self):
        modelIndex = self.currentIndex()
        col = modelIndex.column()
        row = modelIndex.row()
        button = QPushButton('...', self)
        self.setCellWidget(row, col + 1, button)
    # показ контекстного меню
    def contextMenuEvent(self, event):
        Rmenu = QMenu(self)
        delRecord = Rmenu.addAction("Удалить запись")
        relations = Rmenu.addAction("Показать связанные документы")
        action = Rmenu.exec_(self.mapToGlobal(event.pos()))
        if action == delRecord:
            row = self.rowAt(event.pos().y())
            col = self.columnAt(event.pos().x())
            print("row:%d, col:%d" % (row, col))
            item = self.item(self.currentIndex().row(), self.currentIndex().column())
            print(item.text())

    def on_click(self):
        print(self.cursor().pos().x())
        #self.contextMenuEvent(self.cursor())
        for currentQTableWidgetItem in self.selectedItems():
            print(currentQTableWidgetItem.row(), currentQTableWidgetItem.column(), currentQTableWidgetItem.text())

    def setmydata(self):
        n = 0
        for key in self.data:
            m = 0
            for item in key:
                # преобразуем все типы в строку
                newitem = QTableWidgetItem(str(item))
                self.setItem(m, n, newitem)
                n += 1
            m += 1


class WorkArea(QWidget):

    def __init__(self):
        super().__init__()
        self.initUI()

    def initUI(self):
        hbox = QHBoxLayout(self)
        # список таблиц
        listTablesWidget = QListWidget()
        listTablesWidget.setFrameShape(QFrame.StyledPanel)
        listTablesWidget.setFixedWidth(200)
        # таблица для вывода данных
        tablesDataWidget = QTableView()
        tablesDataWidget.setFrameShape(QFrame.StyledPanel)

        splitter1 = QSplitter(Qt.Horizontal)
        splitter1.addWidget(listTablesWidget)
        splitter1.addWidget(tablesDataWidget)

        splitter2 = QSplitter(Qt.Vertical)
        splitter2.addWidget(splitter1)
        #splitter1.setSizes([100, 300])
        hbox.addWidget(splitter2)
        #splitter1.setStretchFactor(1, 0)
        #splitter1.sizeHint()

        self.setLayout(hbox)

        # выводим список таблиц в левом поле
        tblList = dm.initDBstructure()
        for i in tblList:
            listTablesWidget.addItem(i[1])

        # заполняем правый список данными из выбранной таблицы
        def addData(tbl_descr):
            # выбираем название таблицы по описанию
            for i in tblList:
                if i[1] == tbl_descr:
                    tbl_name = i[0]

           # удаляем предыдущий виджет таблицы
            item = splitter1.widget(1)
            if item is not None:
                item.deleteLater()

            data = dm.selectData(tbl_name)
            # проверка на наличие записей в таблице
            if data:
                # количество строк
                rows = len(data)
                # КОЛИЧЕСТВО КОЛОНОК
                cols = len(data[0])
                tablesDataWidget = MyTable(tbl_name, data, rows, cols)
                tablesDataWidget.setFrameShape(QFrame.StyledPanel)
                splitter1.addWidget(tablesDataWidget)
            else:
                # если данных в БД нет выводим пустую табличку
                tablesDataWidget = QTableView()
                tablesDataWidget.setFrameShape(QFrame.StyledPanel)
                splitter1.addWidget(tablesDataWidget)
        # обработка нажатия мышой на списке
        #listTablesWidget.itemClicked.connect(lambda: dm.selectData(listTablesWidget.currentItem))
        listTablesWidget.itemClicked.connect(lambda: addData(listTablesWidget.model().data(listTablesWidget.currentIndex())))

    def onChanged(self, text):
        self.lbl.setText(text)
        self.lbl.adjustSize()

if __name__ == '__main__':

    app = QApplication(sys.argv)
    ex = MainWin()
    sys.exit(app.exec_())