Обрабатывать со словарем в QtableWidget - PullRequest
0 голосов
/ 17 марта 2019

У меня в проекте есть модуль CSV, который импортирует данные из файла CSV. Метод «generateCsvData» преобразует его в словарь, я хочу вставить данные во многие QtableWidgets (зависит от некоторой обработки и расчета). Например, «updateData» может вставить элемент в ячейку 2-го QtableWidget (24,5), проблема здесь, когда я хочу выполнить программу и настройки чтения, в консоли отладки показывает мне: {} Никто. Я знаю, что ссылка Python на первое значение результата глобальной области видимости = {}

class MyWindow(QMainWindow):
    def __init__(self, aPath, parent=None):
        super(MyWindow, self).__init__(parent)
        self.delimit = '\t'
        self.fileName = ""
        self.result = {}

def generateCsvData(self, fileName):
        f = open(fileName, 'r')
        mystring = f.read()
        if mystring.count(",") > mystring.count('\t'):
            if mystring.count(",") > mystring.count(';'):
                self.delimit = ","
            elif mystring.count(";") > mystring.count(','):
                self.delimit = ";"
            else:
                self.delimit = "\t"
        elif mystring.count(";") > mystring.count('\t'):
            self.delimit = ';'
        else:
            self.delimit = "\t"
        f.close()
        f = open(fileName, 'r') 
        with f:
            f = open(fileName, 'r')
            reader = csv.reader(f, delimiter=self.delimit)
            header=[[head] for head in next(reader)]
            for row in reader:
                key = row[0]  #####
                print(row)
                if key in self.result:
                        pass
                self.result[key] = (row [1:])
            print (self.result)

def updateData(self, k, v, n, m):
    print (self.result)
    #print("Value : %s" % self.result.get(k, "Never"))
    item = self.result.get(k)[v]
    print(item)
    newItem = QTableWidgetItem(str(item))
    newItem.setForeground(QBrush(QColor(255, 0, 0)))
    self.ui.tableWidget_Bilan.setItem(n, m, newItem)

if __name__ == "__main__":
    import sys

    app = QApplication(sys.argv)
    app.setApplicationName('MyWindow')
    main = MyWindow('')
    main.updateData(k="11110000", v=0, n=24, m=4)  
    main.setMinimumSize(820, 300)
    main.show()

sys.exit(app.exec_())

когда я нажал кнопку, чтобы выполнить метод "updateData", я получил TypeError: updateData (), пропустив 3 обязательных позиционных аргумента: 'v', 'n' и 'm'. Так как же я могу использовать "updateData" или любой другой метод после того, как dict будет заполнен? Любая идея, пожалуйста, и спасибо!

1 Ответ

2 голосов
/ 17 марта 2019

Попробуйте:

main.py

import csv
from PyQt5.QtCore    import *
from PyQt5.QtGui     import *
from PyQt5.QtWidgets import *

class MyWindow(QMainWindow):
    def __init__(self, aPath, parent=None):
        super(MyWindow, self).__init__(parent)

        centralWidget = QWidget()
        self.setCentralWidget(centralWidget)

        self.tableWidget = QTableWidget(self)
        self.tableWidget.setRowCount(55)
        self.tableWidget.setColumnCount(14)

        conLayout = QHBoxLayout(centralWidget)
        conLayout.addWidget(self.tableWidget)

        self.fileName = "csv.csv"                              # +
        self.result   = {}

        self.generateCsvData(self.fileName)                    # +

    def generateCsvData(self, fileName):                       # +
        with open(fileName, "r") as fileInput:                    
            for i, row in enumerate(csv.reader(fileInput)):
                if i:
                    self.result[row[0]] = (row[1:])

    def updateData(self, k, v, n, m):
        [print(kk, vv) for kk, vv in self.result.items()]

        item = self.result.get(k)[v]
        print("\nitem =`{}`".format(item))

        newItem = QTableWidgetItem(str(item))
        newItem.setForeground(QBrush(QColor(255, 0, 0)))
#        self.ui.tableWidget_Bilan.setItem(n, m, newItem)    # -
        self.tableWidget.setItem(n, m, newItem)              # +

if __name__ == "__main__":
    import sys
    app = QApplication(sys.argv)
    app.setApplicationName('MyWindow')
    main = MyWindow('')
    main.updateData(k="11110000", v=0, n=24, m=4)  
    main.setMinimumSize(820, 300)
    main.show()
    sys.exit(app.exec_())

csv.csv

key,    item1,  item2, item3,  item4,  item5,  item6,  item7, item8, item9, item10, item11, item12, item13
1111, Title 1,  12521, 94565,       ,       ,       , 684651,      ,      ,  44651,       ,       ,  
1112, Title 2,  65115,      , 466149,  46645, 555641,       ,      ,      , 412045,  98416,       ,  
1113, Title 3, 243968,      ,       , 466149,  46645, 555641, 98656,      ,       , 412045,  98416, 
11110000,Title 4,243968,    ,       , 466149,  46645, 555641, 98656,      ,       , 412045,  98416, 

enter image description here


Обновление

import csv
from PyQt5.QtCore    import *
from PyQt5.QtGui     import *
from PyQt5.QtWidgets import *

class MyWindow(QMainWindow):
    def __init__(self, aPath, parent=None):
        super(MyWindow, self).__init__(parent)

        centralWidget = QWidget()
        self.setCentralWidget(centralWidget)

        self.tableWidget = QTableWidget(self)
        self.tableWidget.setRowCount(55)
        self.tableWidget.setColumnCount(14)

        button = QPushButton("Select csv file")               # +
        button.clicked.connect(self.selectCsvFile)            # +

        conLayout = QHBoxLayout(centralWidget)
        conLayout.addWidget(self.tableWidget)
        conLayout.addWidget(button)                           # +

        self.fileName = ""                                    # - "csv.csv"                           
        self.result   = {}

    def generateCsvData(self, fileName):                       
        with open(fileName, "r") as fileInput:                    
            for i, row in enumerate(csv.reader(fileInput)):
                if i:
                    self.result[row[0]] = (row[1:])

        if self.result:                                       # +
            self.updateData(k="11110000", v=0, n=24, m=4)     # +   !!!         

    def updateData(self, k, v, n, m):
        if not self.result: return                            # +

        [print(kk, vv) for kk, vv in self.result.items()]

        item = self.result.get(k)[v]
        print("\nitem =`{}`".format(item))

        newItem = QTableWidgetItem(str(item))
        newItem.setForeground(QBrush(QColor(255, 0, 0)))
        self.tableWidget.setItem(n, m, newItem)              

# +++ vvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvvv        
    def selectCsvFile(self):
        self.fileName, _ = QFileDialog.getOpenFileName(
                           self, 
                           'Open file', 
                           '',
                           "Csv Files (*.csv)")
        if self.fileName: 
            print("\n {} \n".format(self.fileName))
            self.generateCsvData(self.fileName)
# +++ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^    


if __name__ == "__main__":
    import sys
    app = QApplication(sys.argv)
    app.setApplicationName('MyWindow')
    main = MyWindow('')
#    main.updateData(k="11110000", v=0, n=24, m=4)            # --- !!!
    main.setMinimumSize(820, 300)
    main.show()
    sys.exit(app.exec_())

enter image description here

...