import csv
from django.contrib.auth.models import User, Group
from secrets import token_urlsafe
from enapet.settings import BASE_DIR
from enapet2018.views import _enviar_email_generico
from enapet_trabalhos.models import Avaliador, Atribuicao, AvaliacaoIndividual, Trabalho

i = 0
trab_not = []
avaliadores = []
with open(BASE_DIR + '/enapet_trabalhos/scripts/dadosCO.csv', newline='') as File:
    reader = csv.reader(File)
    header = next(reader, None)
    for row in reader:
        a = Avaliador(nome=row[0],
              pet="CO",
              filiacao="CO",
              telefone="CO",
              email=row[1],
              eixo_preferencia="")
        avaliadores.append(a)
        a.save()

for t in Trabalho.objects.all():
    try:
        AvaliacaoIndividual.objects.get(avaliador_trabalho__trabalho=t)
    except:
        trab_not.append(t)


size = len(avaliadores)
i = 0
for trab in trab_not:
    if i >= size:
        i = 0
    a = Atribuicao(avaliador=avaliadores[i],trabalho=trab)
    print("Atribuicao:"+avaliadores[i].nome + " + "+ str(trab.id))
    a.save()
    i += 1


group = Group.objects.get(name='co')
file = open(BASE_DIR+"/enapet_trabalhos/scripts/senhas_co", "w")

for a in avaliadores:
    try:
        token = token_urlsafe(16)
        email = a.email
        user = User.objects.create_user(username=email,
                                         email="",
                                         password=token)
        user.first_name = a.nome
        print(user)
        user.save()
        _enviar_email_generico("XXIII ENAPET - Avaliação de Trabalhos",
                                  "Sistema de correção de trabalhos",
                                  "Você se candidatou a ser avaliador(a) de resumos expandidos para o XXIII ENAPET. "
                                  "Para acessar a plataforma de correção de trabalhos basta acessar "
                                  "<a href='https://enapet2018.com.br/trabalhos/avaliadores/'>"
                                  "este link </a>"
                                  "e inserir seu usuário e senha que são: <br><b>Usuário:</b>"+email+""
                                  "<br><b>Senha:</b>"+token+"<br><br><br><h3>Não responda esse email!</h3>",{"nome":a.nome,"email":email})
        group.user_set.add(user)
        file.write("User:"+email+ " ,First name:"+a.nome + " ,Senha:" + token +"\n")
    except:
        print("DUPLICATED:"+a.email)
        pass

file.close()