RoundsEditorWidget.py

import sys
from PyQt4.QtGui import *
from PyQt4.QtCore import *
from gameEditor import mainWindow
from QuestionEditorWidget import QuestionEditor
class RoundsEditor(QWidget):

    buttonClicked = pyqtSignal(int)
    
    def __init__(self, parent = None):

        super(RoundsEditor, self).__init__(parent)

        self.type = "html"
        self.template = "template.html"
        self.size = {}
        self.rounds = []
        self.jsondict = {}
        self.jsondict["type"] = "html"
        self.jsondict["template"] = "template.html"
        
        self.isOpen = False
	
    def setupGui(self):
        self.frame = QGridLayout()
        self.setLayout(self.frame)

        label = QLabel("Rounds:")
        label.setAlignment(Qt.AlignCenter)
        label.setFont(QFont(QFont.defaultFamily(label.font()), 10))
        self.frame.addWidget(label, 0, 0)

        # adding size buttons
        sizes = QGridLayout()
        self.widthSpin = QDoubleSpinBox()
        self.widthSpin.setRange(0, 3000)
        self.widthSpin.setSingleStep(100)
        self.widthSpin.setDecimals(0)
        self.widthSpin.setFixedWidth(80)
        self.widthSpin.setValue(600)
        
        self.heightSpin = QDoubleSpinBox()
        self.heightSpin.setRange(0, 3000)
        self.heightSpin.setSingleStep(100)
        self.heightSpin.setDecimals(0)
        self.heightSpin.setFixedWidth(80)
        self.heightSpin.setValue(400)

        sizes.addWidget(QLabel("Width:"), 0, 0)
        sizes.addWidget(self.widthSpin, 0, 1)
        sizes.addWidget(QLabel("Height:"), 0, 2)
        sizes.addWidget(self.heightSpin, 0, 3)
        self.frame.addLayout(sizes, 1, 0)

        self.widthSpin.valueChanged.connect(self.values)
        self.heightSpin.valueChanged.connect(self.values)
        
        
        self.layout = QGridLayout()
        self.frame.addLayout(self.layout, 2, 0)

        # adding Save button
        buttonLayout = QGridLayout() # in case we need another button
        saveButton = QPushButton("Save")
        saveButton.setFixedWidth(100)
        buttonLayout.addWidget(saveButton, 0, 0)
        self.frame.addLayout(buttonLayout, 3, 0) # added last in the grid
        saveButton.clicked.connect(self.saveGame)

        self.count = -1 # number of rounds created
        
        self.addRound()
        self.getLineEdit(self.count).setFocus()

    def values(self):
        self.size["width"] = self.widthSpin.value()
        self.size["height"] = self.heightSpin.value()
        print self.size
        
    def addRound(self):
        self.count += 1
        
        lineedit = QLineEdit()
        lineedit.setFixedWidth(200)
        button = QPushButton("Edit")
        button.setEnabled(False)
        button.setFixedWidth(40)
        button.setFixedHeight(22)

        self.layout.addWidget(QLabel("%d." %(self.count+1)), self.count, 0)
        self.layout.addWidget(lineedit, self.count, 1)
        self.layout.addWidget(button, self.count, 2)

        self.rounds.append({"buttonFontSize":14, "labelFontSize":18, "title":""})
    
        button.clicked.connect(self.editRound)
        self.getLineEdit(self.count).textChanged.connect(self.enableButton)
        
    def removeRound(self):
        w = self.getLabelNumber(self.count)
        self.layout.removeWidget(w)
        w.setParent(None)
        
        w = self.getLineEdit(self.count)
        self.layout.removeWidget(w)
        w.setParent(None)
        
        w = self.getButton(self.count)
        self.layout.removeWidget(w)
        w.setParent(None)

        self.frame.setSizeConstraint(QLayout.SetFixedSize) #resizes window
        
        self.count -= 1
        self.rounds = self.rounds[:-1] #removing last entry from list
        self.getButton(self.count).setEnabled(False)
        
    def enableButton(self, string):

        button = self.getButton(self.count) # button -> last made button
        lineedit = self.getLineEdit(self.count) # lineedit -> last made lineedit

        if lineedit.hasFocus() == True:        
            if string == "":
                button.setEnabled(False)
            else:
                button.setEnabled(True)
                self.addRound()
        else:
            if self.getLineEdit(self.count-1).text() == "": # if second to last one is empty, remove last one
                self.removeRound()


    def editRound(self):
        row = (self.layout.indexOf(self.sender())-2) /3
        
        #before editing the round, the round list must be updated
        aux = str(self.getLineEdit(row).text()) # adding round to list

        self.rounds[row]["title"] = aux
        self.size["width"] = int(self.widthSpin.value())
        self.size["height"] = int(self.heightSpin.value())
        
        print "button is on row: ", row
        print self.rounds
        
        if self.isOpen == False:
            round_ =  mainWindow(self.rounds[row]['title'], self.size["width"], self.size["height"], self)
            self.isOpen = True
            round_.show()
        else:
            print "cannot open other editor, current window still running"
            #maybe add a warning

    def saveGame(self):

        # if any changes are made tot the rounds names after 'edit'
        # they will be updated in the self.rounds list
        for i in range(self.count):
            aux = str(self.getLineEdit(i).text())
            self.rounds[i]["title"] = aux
        
        self.rounds = self.rounds[:-1]
        self.jsondict["size"] = self.size
        self.jsondict["rounds"] = self.rounds
        print self.jsondict
        self.close()
        
    def getLabelNumber(self, i):
        return self.layout.itemAtPosition(i, 0).widget()

    def getLineEdit(self, i):
        return self.layout.itemAtPosition(i, 1).widget()
        
    def getButton(self, i):
        return self.layout.itemAtPosition(i, 2).widget()


def main():
    app = QApplication(sys.argv)
    gui = RoundsEditor()
    gui.setupGui()
    gui.show()
    app.exec_()

if __name__ == '__main__':
    main()