From 7806f1134d6f80c5598531bcc3cc88f7ae8c2aec Mon Sep 17 00:00:00 2001
From: bhm15 <bruno.meyer@ufpr.br>
Date: Tue, 6 Feb 2018 14:13:09 -0200
Subject: [PATCH] Listagens feitas

---
 script/analysis/student_analysis.py | 18 +++++-------------
 script/base/dataframe_base.py       |  1 +
 script/build_cache.py               | 15 +++++++++++++++
 3 files changed, 21 insertions(+), 13 deletions(-)

diff --git a/script/analysis/student_analysis.py b/script/analysis/student_analysis.py
index 9a180dd..5203b53 100644
--- a/script/analysis/student_analysis.py
+++ b/script/analysis/student_analysis.py
@@ -15,21 +15,13 @@ def listagem_alunos(df):
 	#~ ativos = df[["MATR_ALUNO", "NOME_PESSOA",]][df["FORMA_EVASAO"] == EvasionForm.EF_ATIVO].drop_duplicates()
 	situacoes = df.groupby(["MATR_ALUNO", "NOME_PESSOA", "FORMA_EVASAO"])
 	situacoes = list(pd.DataFrame({'count' : situacoes.size()}).reset_index().groupby(["FORMA_EVASAO"]))
-	
 	#~ Cria lista de nome de listagens
-	
-	#~ Percorre o situacoes, verifica se o grupo esta em algumas das listagens e insere na correta (ou se não achar insere na listagem de outros)
-	
-	#~ Retorna dicionario com as listagens
-	
-	#~ print(list(situacoes.indices))
-	#~ print(list(situacoes))
+	retorno = {}
+	for s in situacoes:
+		#Busca a lista de alunos relacionados a um codigo
+		retorno[s[0]] = list(s[1]["MATR_ALUNO"])
 
-	#~ iras = ira_alunos(df)
-	
-				
-	
-	return list()
+	return retorno
 
 def ira_alunos(df):
 	iras = ira_por_quantidade_disciplinas(df)
diff --git a/script/base/dataframe_base.py b/script/base/dataframe_base.py
index 586c9e4..820fd33 100644
--- a/script/base/dataframe_base.py
+++ b/script/base/dataframe_base.py
@@ -49,6 +49,7 @@ def fix_dataframes(dataframes):
 	for df in dataframes:
 		if df['name'] == 'historico.xls' or df['name'] == 'historico.csv':
 			history = df['dataframe']
+			history.rename(columns={'DESCR_SITUACAO': 'SITUACAO'}, inplace=True)
 		if df['name'] == 'matricula.xls'  or df['name'] == 'matricula.csv':
 			register = df['dataframe']
 
diff --git a/script/build_cache.py b/script/build_cache.py
index f1fed32..3c4d2e5 100644
--- a/script/build_cache.py
+++ b/script/build_cache.py
@@ -118,6 +118,21 @@ def generate_student_data(path, dataframe):
 			
 		save_json(path+x+".json", student_data[x])
 	
+	listagens_arquivos = [
+		EvasionForm.EF_ABANDONO,
+		EvasionForm.EF_DESISTENCIA,
+		EvasionForm.EF_FORMATURA,
+		EvasionForm.EF_ATIVO
+	]
+	
+	listagens = listagem_alunos(dataframe)
+	for l in listagens:
+		if(l in listagens_arquivos):
+			save_json(path+"listagem/"+str(l)+".json", listagens[l])
+	
+	
+	
+	
 	#Falta verificar se alguem nao recebeu algumas analises
 	
 def generate_student_list(path):
-- 
GitLab