from PySide6.QtGui import QColor, QBrush
from PySide6.QtWidgets import QTreeWidgetItem
from PySide6.QtCore import Qt

###############################################################################
#                 Kolory pastelowe (top-level + jaśniejsze dla dzieci)        #
###############################################################################
GREEN = QColor("#d2ffd2")   # ogólne
RED   = QColor("#ffd2d2")   # szczegółowe
BLUE  = QColor("#d2d2ff")   # przekrojowe

GREEN_LIGHT = QColor("#f0fff0")
RED_LIGHT   = QColor("#ffecec")
BLUE_LIGHT  = QColor("#f0f0ff")

def color_top_level_item(item: QTreeWidgetItem, color: QColor):
    """Ustawia tło w całym wierszu (kolumnach) na podany kolor."""
    brush = QBrush(color)
    for c in range(item.columnCount()):
        item.setBackground(c, brush)

def color_item_by_title(item: QTreeWidgetItem):
    """Koloruje w zależności od kategorii najwyższego rodzica (ogólne / szczegółowe / przekrojowe)."""
    top_parent = item
    while top_parent.parent() is not None:
        top_parent = top_parent.parent()
    
    title_lower = top_parent.text(3).lower()
    is_child = (item != top_parent)

    if "wymagania ogólne" in title_lower:
        color = GREEN_LIGHT if is_child else GREEN
    elif "wymagana przekrojowe" in title_lower:
        color = BLUE_LIGHT if is_child else BLUE
    elif "wymagania szczegółowe" in title_lower:
        color = RED_LIGHT if is_child else RED
    else:
        return  # Brak dopasowania

    color_top_level_item(item, color)