PyQt5: почему я могу использовать только два цвета для ячеек QTableWidget? - PullRequest
Новая
       13

PyQt5: почему я могу использовать только два цвета для ячеек QTableWidget?

0 голосов
/ 05 сентября 2018

Поэтому я пытаюсь раскрасить свой QTableWidget QTableWidgetItems в зависимости от item.text (). Это мой код: 

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

import sys

class Table(QTableWidget):

    def __init__(self, parent):
        super(Table, self).__init__(parent)
        self.setColumnCount(9)
        self.setRowCount(1)
        self.setItem(0, 0, QTableWidgetItem('A'))
        self.setItem(0, 1, QTableWidgetItem('B'))
        self.setItem(0, 2, QTableWidgetItem('C'))
        self.setItem(0, 3, QTableWidgetItem('D'))
        self.setItem(0, 4, QTableWidgetItem('E'))
        self.setItem(0, 5, QTableWidgetItem('F'))
        self.setItem(0, 6, QTableWidgetItem('A'))
        self.setItem(0, 7, QTableWidgetItem('C'))
        self.setItem(0, 8, QTableWidgetItem('D'))

        for r in range(self.rowCount()):
            for c in range(self.columnCount()):
                item = QTableWidgetItem()
                self.item(r, c).setBackground(self.colour(self.item(r, c).text()))

    def colour(self, letter):

        if letter == 'A':
            colour = QColor(233, 12, 24)
        if letter == 'B':
            colour = QColor(12, 45, 67)
        if letter in ['C', 'E']:
            colour = QColor(23, 57, 188)
        if letter == 'F':
            colour = QColor(45, 116, 75)
        else:
            colour = QColor(233, 244, 12)
        return colour

def main():
    app = QApplication(sys.argv)
    window = QWidget()
    window.setGeometry(200,200,1200,400)
    tw = Table(window)
    twLayout = QVBoxLayout()
    twLayout.addWidget(tw)
    window.setLayout(twLayout)

    window.show()
    sys.exit(app.exec_())

if __name__ == '__main__':
    main()

Когда я запускаю его, в моем столе остаются только два цвета, а остальные игнорируются. Как это исправить и как убедиться, что каждая ячейка получает свой индивидуальный цвет в зависимости от ее содержимого?

1 Ответ

0 голосов
/ 05 сентября 2018

В определении Table.colour у вас есть несколько операторов if, но предложение else будет соответствовать только последнему из них. Вы хотите ... 

def colour(self, letter):
    if letter == 'A':
        colour = QColor(233, 12, 24)
    elif letter == 'B':
        colour = QColor(12, 45, 67)
    elif letter in ['C', 'E']:
        colour = QColor(23, 57, 188)
    elif letter == 'F':
        colour = QColor(45, 116, 75)
    else:
        colour = QColor(233, 244, 12)
    return colour
...