from getpass import getpass

from bs4 import BeautifulSoup

import controle, sys


class Disciplina:

    def __init__(self, cod, nome, turma):
        self.cod = cod.strip()
        self.nome = nome.strip()
        self.turma = turma.strip()

    def __str__(self):
        return self.cod + " - " + self.turma + " - " + self.nome

    def __repr__(self):
        return self.__str__()


class Aluno:

    def __init__(self, matr, curso, nome, estado):
        self.matr = matr.strip()
        self.curso = curso.strip()
        self.nome = nome.strip()
        self.estado = estado.strip()

    def __str__(self):
        return self.curso + ";" + self.matr + ";" + self.nome + ";" + self.estado

    def __repr__(self):
        return self.__str__()


def data_resumo(codigo, turma):
    data = {"command": "CoordenacaoTurmaResumo",
            "codigo": codigo,
            "turma": turma
            }
    return data


def main():
    login_ = input("LOGIN: ")
    senha_ = getpass("SENHA: ")
    jsessionid = controle.login(login_, senha_, "Coordenacao")
    print(jsessionid)
    resultado_listagem = controle.chamada({"command": "CoordenacaoDisciplinasOfertadasListar"}, jsessionid).read()
    disciplinas = processa(resultado_listagem)
    open('disciplinas.html', 'wb').write(resultado_listagem)
    print(disciplinas)

    for disciplina in disciplinas:
        try:
            alunos = resumo(disciplina, jsessionid)
            open(disciplina.nome + " - " + disciplina.turma + ".html", 'wb').write(alunos)
            arquivo = open(disciplina.nome + " - " + disciplina.turma + ".csv", 'wb')
            for x in processa_resumo(alunos):
                arquivo.write((str(x) + '\n').encode())
            arquivo.close()
        except IOError:
            print("I/O error")
        except ValueError:
            print("Could not convert data to an integer.")
        except:
            print("Unexpected error:", sys.exc_info()[0])

def processa(html):
    soup = BeautifulSoup(html, 'html.parser')
    disciplinas = []
    for tb in soup.find_all('tbody'):
        for row in tb.find_all('tr'):
            content = [c.getText().strip().replace('\n','').replace('\r','') for c in row.find_all('td')]
            disciplina = Disciplina(*content[0:3])
            disciplinas.append(disciplina)
    return disciplinas


def processa_resumo(html):
    soup = BeautifulSoup(html, 'html.parser')
    alunos = []
    for tb in soup.find_all('tbody'):
        for row in tb.find_all('tr'):
            content = [c.getText().strip().replace('\n','').replace('\r','') for c in row.find_all('td')]
            _, matr, curso, nome, estado = content[0:5]
            aluno = Aluno(matr, curso, nome, estado)
            alunos.append(aluno)
    return alunos


def resumo(disciplina, jsessionid):
    return controle.chamada(data_resumo(disciplina.cod, disciplina.turma), jsessionid).read()


if __name__ == "__main__":
    main()