Codigo Principal

# -*- coding: utf-8 -*-
import sys
from PyQt4 import QtCore, QtGui
from gestor import Ui_bj2
import csv
import pylab
import blocks
import classgame


class Principal(QtGui.QWidget):
    def __init__(self):
        QtGui.QWidget.__init__(self)
        self.ventana=Ui_bj2()
        self.ventana.setupUi(self)
        ##

        #opcion salir
        self.connect(self.ventana.bsalir, QtCore.SIGNAL('clicked()'),QtCore.SLOT('close()'))
        #juegos de classes
        self.connect(self.ventana.bj1, QtCore.SIGNAL('clicked()'),self.gclass)
        #juego en pygame
        self.connect(self.ventana.pushButton_2, QtCore.SIGNAL('clicked()'),self.pgame)
        #grafica class
        self.connect(self.ventana.bgj1,QtCore.SIGNAL('clicked()'),self.grapclass)
        #grafica pygame
        self.connect(self.ventana.bgj2,QtCore.SIGNAL('clicked()'),self.grappg)
        #juego de clases

    def gclass(self):
        dmg=classgame.main()
        dmg1=dmg[0]
        dmg2=dmg[1]
        direc = "D:\Desktop\data\classgame.csv"
        lclass=[]
        try:
            archivo=open(direc,"r")
            archivo.close()
        except:
            archivo=open(direc,"w")
            archivo_csv=csv.writer(archivo)
            lclass.append([dmg1,dmg2])
            archivo_csv.writerows(lclass)
            archivo.close()
        archivo=open(direc,"r")
        archivo_csv=csv.reader(archivo)
        for dano1,dano2 in archivo_csv:
            lclass.append([dano1,dano2])
        archivo.close()
        lclass.insert(0,[dmg1,dmg2])
        archivo=open(direc,"w")
        archivo_csv=csv.writer(archivo)
        archivo_csv.writerows(lclass[:9])
        archivo.close()
        #llamar al juego de clases
    def pgame(self):
        score=blocks.main()
        #print score
        direc="D:\Desktop\data\pggame.csv"
        lscore=[]
        try:
            archivo=open(direc,"r")
            archivo.close()
        except:
            archivo=open(direc,"w")
            archivo_csv=csv.writer(archivo)
            lscore.append([score,score])
            archivo_csv.writerows(lscore)
            archivo.close()
        archivo=open(direc,"r")
        archivo_csv=csv.reader(archivo)
        for data1 in archivo_csv:
            lscore.append(data1)
        archivo.close()
        lscore.insert(0,[score,score])
        archivo=open(direc,"w")
        archivo_csv=csv.writer(archivo)
        archivo_csv.writerows(lscore[:9])
        archivo.close()

    def grapclass(self):
        direc = "D:\Desktop\data\classgame.csv"
        lclass=[]
        archivo=open(direc,"r")
        archivo_csv=csv.reader(archivo)
        for dano1,dano2 in archivo_csv:
            lclass.append([dano1,dano2])
        archivo.close()
        sc1=[]
        sc2=[]
        #print lclass
        for s1,s2 in lclass:
            sc1.append(s1)
            sc2.append(s2)
        numerate=range(1,len(sc1)+1)
        pylab.plot(numerate,sc1,numerate,sc2)
        pylab.title("Grafica de desempeño")
        pylab.grid('on')
        pylab.legend(['Player1', 'PLayer2'])
        pylab.show()
    def grappg(self):
        direc="D:\Desktop\data\pggame.csv"
        lscore=[]
        archivo=open(direc,"r")
        archivo_csv=csv.reader(archivo)
        for data1 in archivo_csv:
            lscore.append(data1)
        archivo.close()
        #print lscore
        ps=[]
        ps2=[]
        for pg1,pg2 in lscore:
            ps.append(pg1)
            ps2.append(pg2)
        nume=range(1,len(ps)+1)
        pylab.plot(nume,ps)
        pylab.title("Puntajes")
        pylab.grid('on')
        pylab.show()


def main():
    app=QtGui.QApplication(sys.argv)
    ventana=Principal()
    ventana.show()
    sys.exit(app.exec_())
if __name__=='__main__':
    main()