File contents

import MySQLdb

con = MySQLdb.connect(host="localhost", user="root", passwd="admin",db="mysql")



cursor = con.cursor()

#comando = ALTER TABLE mysql.TB_FUNCIONARIO MODIFY INCODFUNCIONARIO INT NOT NULL AUTO_INCREMENT;



#menu
op=""
while (op.upper()!="S"):
    print ("CT --> cadastrar novo tipo de quarto\n CQ--> cadastrar novo quarto \n S--> sair")
    op = raw_input("Digite a opacao: ")

    if (op=="CT"):

        vatipoquarto = raw_input("digite o tipo do quarto: ")
        
        vacusto = raw_input("digite o custo: ")

        comando = "INSERT INTO mysql.tb_tipoquarto (`var_tipoquarto`, `custo`) VALUES ('" + vatipoquarto + "', '" + vacusto + "')"

        #executando comando
        cursor.execute(comando)
        con.commit()

    elif (op=="CQ"):
        cursor.execute('select * from tb_tipoquarto')
        print (cursor.fetchall())

        
        incodigotipoquarto = (raw_input("digite codigo do tipo do quarto: "))
        vaquarto = raw_input("digite o nome do quarto: ")

        comando = "INSERT INTO mysql.tb_quarto (`cod_tipoquarto`, `var_nome`) VALUES ('" + incodigotipoquarto + "', '" + vaquarto + "')"

        #executando comando
        cursor.execute(comando)
        con.commit()



cursor.execute('select * from tb_tipoquarto')

#Recebendo resultados 
a = cursor.fetchall()
con.close()

print (a)