class TelaPrincipal(QtGui.QMainWindow, telaPrincipal.Ui_MainWindow):
    def __init__(self):
        super(TelaPrincipal, self).__init__()
        self.setupUi(self)

        self.w = None
        self.cc = None
        self.cf = None
        self.control = Controller()
        self.control.gerar_banco()
        self.control.cadastro_locadora()

        self.actionFazer_login.triggered.connect(self.fazerLogin)
        self.actionCliente.triggered.connect(self.cadastroCliente)
        self.actionFilme.triggered.connect(self.cadastroFilme)

        self.menuCadastro.setEnabled(False)
        self.actionFilme.setEnabled(False)
        self.actionCliente.setEnabled(False)

        self.move(100, 200)
        self.resize(1000, 1000)

    def fazerLogin(self):
        self.w = Login(self)
        self.w.control = self.control
        self.w.show()

    def cadastroCliente(self):
        print("cliente")
        self.cc = CadastroCliente()
        self.cc.control = self.control
        self.cc.show()

    def cadastroFilme(self):
        self.cf = CadastroFilme(self, self.control)
        self.cf.show()
__author__ = 'andersonmarques'

from controller import Controller

#////////  Loop principal ///////////////
sair = False
control = Controller()
control.gerar_banco()
control.cadastro_locadora()
while not sair:
    control.menu()
    r = int(input("Digite a opcao: "))
    if r in [1, 2, 3]:
        control.switch(r)
    else:
        sair = True
print("Saindo\n bye")