CharakterVorteileWrapper.py

# -*- coding: utf-8 -*-
"""
Created on Sat Mar 18 12:21:03 2017

@author: Aeolitus
"""
from Wolke import Wolke
import CharakterVorteile
import CharakterMinderpaktWrapper
from Charakter import VariableKosten
from PyQt5 import QtWidgets, QtCore
from Definitionen import VorteilTypen
import logging
from EventBus import EventBus

class CharakterVorteileWrapper(QtCore.QObject):
    modified = QtCore.pyqtSignal()
    
    def __init__(self):
        super().__init__()
        logging.debug("Initializing VorteileWrapper...")
        self.formVor = QtWidgets.QWidget()
        self.uiVor = CharakterVorteile.Ui_Form()
        self.uiVor.setupUi(self.formVor)
        
        self.uiVor.treeWidget.itemSelectionChanged.connect(self.vortClicked)
        self.uiVor.treeWidget.itemChanged.connect(self.itemChangeHandler)
        self.uiVor.treeWidget.header().setSectionResizeMode(0,1)
        
        if len(Wolke.Char.vorteile) > 0:
            self.currentVort = Wolke.Char.vorteile[0]
        else:
            self.currentVort = ""
            
        self.itemWidgets = {}
        
        self.initVorteile()

    def initVorteile(self):
        self.uiVor.treeWidget.blockSignals(True)
        vortList = []
        for vortTyp in VorteilTypen:
            vortList.append([])

        for el in Wolke.DB.vorteile:
            idx = Wolke.DB.vorteile[el].typ
            vortList[idx].append(el)
        
        for vorteile in vortList:
            vorteile.sort()

        for i in range(len(vortList)):
            parent = QtWidgets.QTreeWidgetItem(self.uiVor.treeWidget)
            parent.setText(0, VorteilTypen[i])
            parent.setText(1,"")
            parent.setExpanded(True)
            for el in vortList[i]:
                child = QtWidgets.QTreeWidgetItem(parent)
                child.setText(0, Wolke.DB.vorteile[el].name)
                if el in Wolke.Char.vorteile:    
                    child.setCheckState(0, QtCore.Qt.Checked)
                else:
                    child.setCheckState(0, QtCore.Qt.Unchecked)
                if Wolke.DB.vorteile[el].variable!=-1:
                    spin = QtWidgets.QSpinBox()
                    spin.setMinimum(-9999)
                    spin.setSuffix(" EP")
                    spin.setButtonSymbols(QtWidgets.QAbstractSpinBox.NoButtons)
                    spin.setMaximum(9999)
                    if el == Wolke.Char.minderpakt:
                        spin.setValue(20)
                        spin.setReadOnly(True)
                    else:
                        if el in Wolke.Char.vorteileVariable:
                            spin.setValue(Wolke.Char.vorteileVariable[el].kosten)
                        else:
                            spin.setValue(Wolke.DB.vorteile[el].kosten)
                    spin.setSingleStep(20)
                    self.itemWidgets[el] = spin
                    spin.valueChanged.connect(lambda state, name=el: self.spinnerChanged(name,state))
                    self.uiVor.treeWidget.setItemWidget(child,1,spin)
                    if child.checkState(0) == QtCore.Qt.Checked:
                        self.handleAddKommentarWidget(el, child)
                else:
                    child.setText(1, "20 EP" if el == Wolke.Char.minderpakt else str(Wolke.DB.vorteile[el].kosten) + " EP")
                if Wolke.Char.voraussetzungenPrüfen(Wolke.DB.vorteile[el].voraussetzungen):
                    child.setHidden(False)
                else:
                    child.setHidden(True)
        self.updateInfo()
        self.uiVor.treeWidget.blockSignals(False)
        
    def load(self):
        self.uiVor.treeWidget.blockSignals(True)
        vortList = []
        for vortTyp in VorteilTypen:
            vortList.append([])

        for el in Wolke.DB.vorteile:
            if Wolke.Char.voraussetzungenPrüfen(Wolke.DB.vorteile[el].voraussetzungen):
                idx = Wolke.DB.vorteile[el].typ
                vortList[idx].append(el)
        for i in range(len(vortList)):
            itm = self.uiVor.treeWidget.topLevelItem(i)
            if type(itm) != QtWidgets.QTreeWidgetItem:
                    continue
            if itm == 0: 
                continue
            for j in range(itm.childCount()):
                chi = itm.child(j)
                if type(chi) != QtWidgets.QTreeWidgetItem:
                    continue
                txt = chi.text(0)
                if txt in Wolke.Char.vorteile or txt == Wolke.Char.minderpakt:    
                    chi.setCheckState(0, QtCore.Qt.Checked)
                    if txt in Wolke.Char.vorteileVariable:
                        self.itemWidgets[txt].setValue(Wolke.Char.vorteileVariable[txt].kosten)
                        self.handleAddKommentarWidget(txt, chi)
                else:
                    chi.setCheckState(0, QtCore.Qt.Unchecked) 
                if txt not in vortList[i] and txt != Wolke.Char.minderpakt:
                    chi.setHidden(True)
                    if txt in Wolke.Char.vorteile:
                        chi.setCheckState(0,QtCore.Qt.Unchecked)
                        Wolke.Char.vorteile.remove(txt)
                else:
                    chi.setHidden(False)
                if Wolke.DB.vorteile[txt].variable!=-1 and not txt in Wolke.Char.vorteileVariable:
                    self.setVariableKosten(txt, self.itemWidgets[txt].value(), "")
        self.updateInfo()
        self.uiVor.treeWidget.blockSignals(False)
        
    def update(self):
        pass

    def setVariableKosten(self, name, kosten, kommentar):
        if not name in Wolke.Char.vorteileVariable:
            vk = VariableKosten()
            Wolke.Char.vorteileVariable[name] = vk

        if kosten != None:
            Wolke.Char.vorteileVariable[name].kosten = kosten
        if kommentar != None:
            Wolke.Char.vorteileVariable[name].kommentar = kommentar

    def spinnerChanged(self,name,state):
        self.setVariableKosten(name, state, None)
        self.currentVort = name
        self.modified.emit()
        self.updateInfo()

    def kommentarChanged(self, name, text):
        self.setVariableKosten(name, None, text)
        self.currentVort = name
        self.modified.emit()
        self.updateInfo()

    def handleAddKommentarWidget(self, name, parent):
        if Wolke.DB.vorteile[name].variable == -1 or parent.childCount() > 0:
            return
        kommentar = ""
        if name in Wolke.Char.vorteileVariable:
            kommentar = Wolke.Char.vorteileVariable[name].kommentar

        w = QtWidgets.QWidget()
        layout = QtWidgets.QHBoxLayout()
        label = QtWidgets.QLabel("Kommentar")
        text = QtWidgets.QLineEdit(kommentar)
        layout.addWidget(label)
        layout.addWidget(text)
        w.setLayout(layout)
        text.textChanged.connect(lambda text, name=name: self.kommentarChanged(name, text))

        child = QtWidgets.QTreeWidgetItem(parent)
        self.uiVor.treeWidget.setItemWidget(child,0,w)
        parent.setExpanded(True)

    def handleRemoveKommentarWidget(self, parent):
        parent.takeChild(0) #doesnt do anything if it doesnt exist

    def handleAddMinderpakt(self, name, item):
        if name != "Minderpakt":
            return None
        minderp = CharakterMinderpaktWrapper.CharakterMinderpaktWrapper()
        if minderp.minderpakt is None:
            Wolke.Char.vorteile.remove(name)
            return None
        if minderp.minderpakt in Wolke.DB.vorteile and minderp.minderpakt not in Wolke.Char.vorteile:
            Wolke.Char.minderpakt = minderp.minderpakt
            Wolke.Char.vorteile.append(minderp.minderpakt)
            EventBus.doAction("vorteil_gekauft", { "name" : minderp.minderpakt})
            minderpaktWidget = self.uiVor.treeWidget.findItems(Wolke.Char.minderpakt, QtCore.Qt.MatchRecursive)[0]

            if Wolke.Char.minderpakt in self.itemWidgets:
                self.itemWidgets[Wolke.Char.minderpakt].setReadOnly(True)
                self.itemWidgets[Wolke.Char.minderpakt].setValue(20)
            else:
                minderpaktWidget.setText(1, "20 EP")
            return minderpaktWidget
        return None

    def restoreMinderpaktWidgets(self, name, item):
            if name in self.itemWidgets:
                self.itemWidgets[name].setReadOnly(False)
                self.itemWidgets[name].setValue(Wolke.DB.vorteile[name].kosten)
            else:
                item.setText(1, str(Wolke.DB.vorteile[name].kosten) + " EP")

    def handleRemoveMinderpakt(self, name, item):
        if name == "Minderpakt":
            if Wolke.Char.minderpakt in Wolke.Char.vorteile:
                Wolke.Char.vorteile.remove(Wolke.Char.minderpakt)
                EventBus.doAction("vorteil_entfernt", { "name" : Wolke.Char.minderpakt})
            minderpaktWidget = self.uiVor.treeWidget.findItems(Wolke.Char.minderpakt, QtCore.Qt.MatchRecursive)[0]
            self.restoreMinderpaktWidgets(Wolke.Char.minderpakt, minderpaktWidget)
            Wolke.Char.minderpakt = None
            return minderpaktWidget

        if Wolke.Char.minderpakt is not None and name == Wolke.Char.minderpakt:
            if "Minderpakt" in Wolke.Char.vorteile:
                Wolke.Char.vorteile.remove("Minderpakt")
                EventBus.doAction("vorteil_entfernt", { "name" : "Minderpakt"})
            self.restoreMinderpaktWidgets(Wolke.Char.minderpakt, item)
            Wolke.Char.minderpakt = None
        return None
    
    def itemChangeHandler(self, item, column):
        # Block Signals to make sure we dont repeat infinitely
        self.uiVor.treeWidget.blockSignals(True)
        name = item.text(0)
        self.currentVort = name
        self.updateInfo()
        cs = item.checkState(0)
        manualUpdate = None

        if cs == QtCore.Qt.Checked and name not in Wolke.Char.vorteile and name != "":
            Wolke.Char.vorteile.append(name)
            self.handleAddKommentarWidget(name, item)
            manualUpdate = self.handleAddMinderpakt(name, item)
            EventBus.doAction("vorteil_gekauft", { "name" : name})
        elif cs != QtCore.Qt.Checked and name in Wolke.Char.vorteile:
            Wolke.Char.vorteile.remove(name)
            self.handleRemoveKommentarWidget(item)
            manualUpdate = self.handleRemoveMinderpakt(name, item)
            EventBus.doAction("vorteil_entfernt", { "name" : name})

        self.modified.emit()
        self.load() 
        self.uiVor.treeWidget.blockSignals(False)

        if manualUpdate:
            self.itemChangeHandler(manualUpdate, 0)
    
    def vortClicked(self):
        for el in self.uiVor.treeWidget.selectedItems():
            if el.text(0) in VorteilTypen:
                continue
            self.currentVort = el.text(0)
            break #First one should be all of them
        self.updateInfo()
        
    def updateInfo(self):
        if self.currentVort != "":
            self.uiVor.labelVorteil.setText(Wolke.DB.vorteile[self.currentVort].name)
            self.uiVor.labelTyp.setText(VorteilTypen[Wolke.DB.vorteile[self.currentVort].typ])
            self.uiVor.labelNachkauf.setText(Wolke.DB.vorteile[self.currentVort].nachkauf)
            self.uiVor.plainText.setPlainText(Wolke.DB.vorteile[self.currentVort].text)
            if Wolke.DB.vorteile[self.currentVort].variable!=-1 and self.currentVort in Wolke.Char.vorteileVariable:
                self.uiVor.spinKosten.setValue(Wolke.Char.vorteileVariable[self.currentVort].kosten)
            else:
                self.uiVor.spinKosten.setValue(Wolke.DB.vorteile[self.currentVort].kosten)