#importando módulo do Banco de Dados
import pymysql as mariadb

class BD_Cliente():


    def __init__(self):
        self.conexao = mariadb.connect(user='root', password='', host='127.0.0.1', database='cliente')
        self.createTable()


    def createTable(self):
        ConectaDB = self.conexao.cursor()

        ConectaDB.execute("""create table if not exists cliente (
                     NumeroConta integer primary key autoincrement ,
                     CPF text,
                     Nome text,
                     Senha text,
                     TipoConta1 boolean,
                     TipoConta2 boolean)""")

        self.conexao.commit()
        ConectaDB.close()

from BD_Cliente import BD_Cliente

class Cliente(object):


    def __init__(self, NumeroConta=0, CPF="", Nome="", Senha="", TipoConta1= "", TipoConta2= ""):
        self.info = {}
        self.NumeroConta = NumeroConta
        self.CPF = CPF
        self.Nome = Nome
        self.Senha = Senha
        self.TipoConta1 = TipoConta1
        self.TipoConta2 = TipoConta2


    def insertCliente(self):

        banco = BD_Cliente()
        try:

            ConectaDB = banco.conexao.cursor()

            ConectaDB.execute("insert into NumeroConta (CPF, Nome, Senha, TipoConta1, TipoConta2) values ('" + self.CPF + "', '" + self.Nome + "', '" + self.Senha + "', '" + self.TipoConta1 + "', '" + self.TipoConta2 + "')")

            banco.conexao.commit()
            ConectaDB.close()

            return "Cliente cadastrado com sucesso!"
        except:
            return "Ocorreu um erro na inserção do Cliente"


    def updateCliente(self):

        banco = BD_Cliente()
        try:

            ConectaDB = banco.conexao.cursor()

            ConectaDB.execute("update NumeroConta set CPF = '" + self.CPF + "', Nome = '" + self.Nome + "', Senha = '" + self.Senha + "', TipoConta1 = '" + self.TipoConta1 + "', TipoConta2 = '" + self.TipoConta2 + "' where NumeroConta = " + self.NumeroConta + " ")

            banco.conexao.commit()
            ConectaDB.close()

            return "Cliente atualizado com sucesso!"
        except:
            return "Ocorreu um erro na alteração do Cliente"


    def deleteCliente(self):

        banco = BD_Cliente()
        try:

            remove = banco.conexao.cursor()

            remove.execute("delete from Cliente where NumeroConta = " + self.NumeroConta + " ")

            banco.conexao.commit()
            remove.close()

            return "Cliente excluído com sucesso!"
        except:
            return "Ocorreu um erro na exclusão do Cliente"


    def selectCliente(self, NumeroConta):
        banco = BD_Cliente()
        try:

            ConectaDB = banco.conexao.cursor()

            ConectaDB.execute("select * from Cliente where NumeroConta = " + NumeroConta + "  ")

            for linha in ConectaDB:
                self.NumeroConta = linha[0]
                self.CPF = linha[1]
                self.Nome = linha[2]
                self.Senha = linha[3]
                self.TipoConta1 = linha[4]
                self.TipoConta2 = linha[5]

            ConectaDB.close()

            return "Busca feita com sucesso!"
        except:
            return "Ocorreu um erro na busca do Cliente"