Listagens feitas

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)

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']
 

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):