# coding: utf-8
from datetime import datetime
import mysql.connector

print "SISTEMA DE BIBLIOTECA"
print '----------------------------------------'

clear = "\n" * 100

# Função para cadastrar o livro.
def cadastrarLivros():
    print "Cadastrando Livros\n"

    # Recebendo os dados do livro.
    titulo = raw_input("Insira o título do livro: ");
    nome_autor = raw_input("Insira o primeiro nome do autor: ");
    sob_autor = raw_input("Insira o sobrenome do autor: ");
    ano = raw_input("Insira o ano do livro: ");
    editora = raw_input("Insira o nome da editora: ")
    categoria = raw_input("Insira a categoria do livro: ");
    quantidade = input("Insira a quantidade de estoque: ");

    # Fazendo a conexão com o mySQL.
    cnx = mysql.connector.connect(user='root', password='', host='localhost', database='bibliotecapython')
    cursor = cnx.cursor()

    # Linha de inserção do SQL.
    queryLivros = ("INSERT INTO livros (titulo, quantidade, nome_autor, sob_autor, ano, editora, categoria) "
                   "VALUES (%s, %s, %s, %s, %s, %s, %s)")

    # Variáveis.
    dadosLivros = (titulo, quantidade, nome_autor, sob_autor, ano, editora, categoria);

    # Executando a query com as variáveis.
    cursor.execute(queryLivros, dadosLivros);

    # Enviando para o banco de dados.
    cnx.commit();

    # Finalizando o processo e avisando ao usuário.
    cursor.close();
    cnx.close()

    print "= Livro cadastrado com sucesso! ="
# -----------------------------------------

# Função para cadastrar usuários.
def cadastrarAlunos():
    print "Cadastrando Alunos\n"

    # Recebendo os dados do usuário.
    nomeCompleto = raw_input("Insira o nome completo do aluno: ")
    idade = input("Insira a idade do aluno: ")
    CPF = input("Informe o CPF do aluno: ")
    matricula = raw_input("Informe o número de matrícula do aluno: ")
    senha = raw_input("Informe uma senha para o aluno: ")
    dataDeCadastro = datetime.now()

    # Fazendo a conexão com o mySQL.
    cnx = mysql.connector.connect(user='root', password='', host='localhost', database='bibliotecapython')
    cursor = cnx.cursor()

    # Query da conta do aluno.
    queryConta = ("INSERT INTO conta (login, senha, tipo) VALUES (%s, %s, %s)")
    dadosConta = (matricula, senha, 2);

    # Executando a query com as variáveis.
    cursor.execute(queryConta, dadosConta);

    # Enviando para o banco de dados.
    cnx.commit();

    # Query pra pegar o ID da conta que acabou de ser cadastrada.
    cursor.execute('SELECT * from conta where login="' + matricula + '"');
    row = cursor.fetchone()

    # Pegando o ID da conta.
    contaAlunoID = row[0];

    # Linha de inserção do SQL.
    queryAluno = ("INSERT INTO aluno (nomecompleto, idade, cpf, datadecadastro, contaid) VALUES (%s, %s, %s, %s, %s)")

    # Variáveis.
    dadosAluno = (nomeCompleto, idade, CPF, dataDeCadastro, contaAlunoID);

    # Executando a query com as variáveis.
    cursor.execute(queryAluno, dadosAluno);

    # Enviando para o banco de dados.
    cnx.commit();

    # Finalizando o processo e avisando ao usuário.
    cursor.close();
    cnx.close()

    print clear

    print "= Aluno cadastrado com sucesso! ="
# -----------------------------------------

def listarAlunos():
    # Criando a conexão com o MySQL.
    cnx = mysql.connector.connect(user='root', password='', host='localhost', database='bibliotecapython')

    # Criando um cursor (buffered pois não sei o porque)
    cursor = cnx.cursor(buffered=True);

    query = 'SELECT * FROM aluno';

    cursor.execute(query);

    for aluno in cursor:
        print "================="
        print "Nome Completo: " + aluno[1];
        print "Idade: " + aluno[2];
        print "CPF: " + aluno[3];
        print "Data de Cadastro: " + str(aluno[4]);

    cursor.close()
    cnx.close()
    print "================="
    escolha = input("Digite qualquer coisa para voltar: ");

    print clear

def emprestarLivro():
    print "- Livros disponíveis: -"

    # Criando a conexão com o MySQL.
    cnx = mysql.connector.connect(user='root', password='', host='localhost', database='bibliotecapython')

    # Criando um cursor (buffered pois não sei o porque)
    cursor = cnx.cursor(buffered=True);

    cursor.execute("SELECT * FROM livros")

    for livro in cursor:
        print "=================";
        print "ID: " + str(livro[0]);
        print "Titulo: " + livro[1];
        print "Nome do Autor: " + livro[3] + " " + livro[4];
        print "Ano: " + livro[5];
        print "Editora: " + livro[6];
        print "Categoria: " + livro[7];
        print "Quantidade Disponível: " + str(livro[2]);

    print "=================";
    idLivro = raw_input("Insira o ID do livro que será emprestado: ");
    print clear

    cursor.execute("SELECT * FROM aluno");

    for aluno in cursor:
        print "================="
        print "ID: " + str(aluno[0])
        print "Nome Completo: " + aluno[1];
        print "Idade: " + aluno[2];
        print "CPF: " + aluno[3];
        print "Data de Cadastro: " + str(aluno[4]);

    print "=================";
    idAluno = raw_input("Insira o ID do aluno que terá o livro emprestado: ")


# Recebendo os dados do usuário.
print clear
loginUser = raw_input("Insira o seu e-mail: ")
senhaUser = raw_input("Insira sua senha: ")
tipo = 0;

loginUser = loginUser;

# Criando a conexão com o MySQL.
cnx = mysql.connector.connect(user='root', password='', host='localhost', database='bibliotecapython')

# Criando um cursor (buffered pois não sei o porque)
cursor = cnx.cursor(buffered=True);

query = 'SELECT * FROM conta where login="' + loginUser + '"';

cursor.execute(query);

contaID = 0;

# Caso exista alguma conta com esse login.
if (cursor.rowcount > 0):
    row = cursor.fetchone()
    # Caso a senha esteja correta.
    if (senhaUser == row[2]):
        # Caso o tipo da conta seja administrador.
        if (row[3] == 1):
            # Pegando o ID da conta para pegar os dados do admin.
            contaID = row[0];

            query = 'SELECT * FROM admin where contaid=' + str(contaID) + '';

            cursor.execute(query);
            row = cursor.fetchone();

            # "Limpando" o console.
            print clear

            print "\nVocê está logado com sucesso! -", row[1];
            print "---------------------------------------"

            # Variável de escolha.
            escolha = 9999
            # Lista de usuários cadastrados.
            usuarios = []
            # Lista de livros cadastrados.
            livros = []

            while (escolha != 0):
                # Sistema de escolhas.
                print "Escolha uma opção:"
                print "1 - Cadastrar Livros"
                print "2 - Cadastrar Alunos"
                print "3 - Listar Alunos"
                print "4 - Empréstimo de Livros"
                print "0 - Cancelar"

                escolha = input("Escolha: ")

                # Caso o usuário escolha cadastrar livros.
                if (escolha == 1):
                    print "--------------------------------------"
                    print clear
                    cadastrarLivros()
                    print "--------------------------------------"

                # Caso o usuário escolha cadastrar usuários.
                elif (escolha == 2):
                    print "--------------------------------------"
                    print clear
                    cadastrarAlunos()
                    print "---------------------------------------"
                # Caso o usuário escolha listar os usuários cadastrados.
                elif (escolha == 3):
                    print clear
                    listarAlunos()
                    print "--------------------------------------"
                # Caso o usuário escolha alocar um livro.
                elif (escolha == 4):
                    print clear
                    emprestarLivro()
                    print "--------------------------------------"
                # Caso o usuário selecione sair do app.
                elif (escolha == 0):
                    cursor.close();
                    cnx.close()
        # else if tipo = usuario:
    else:
        print "\nSua senha está incorreta."
        print "----------------------------------"
else:
    print "\nSeu e-mail não está cadastrado."
    print "--------------------------------------"