From 71ec31970c9e6116183f2ca66da138ee9fec801a Mon Sep 17 00:00:00 2001
From: odair mario <badwolves123@gmail.com>
Date: Thu, 2 Nov 2017 16:25:24 -0200
Subject: [PATCH] valores default para campos null definidos de forma decente

---
 script/analysis/course_analysis.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/script/analysis/course_analysis.py b/script/analysis/course_analysis.py
index b862844..378ae7b 100644
--- a/script/analysis/course_analysis.py
+++ b/script/analysis/course_analysis.py
@@ -45,10 +45,10 @@ def df_to_json(disciplina,qnt_matr):
         disc = disciplina.loc[disciplina['COD_ATIV_CURRIC']==dis].drop('COD_ATIV_CURRIC',1) # separa o dataframe em disciplina e elimina a coluna codigo
         # seta a coluna SITUACAO como index
         disc = disc.set_index('SITUACAO').to_dict()
-        for i in range(16):
-            if not(i in disc['Quantidade'].keys()):
-                disc['Quantidade'][i] = 0
-                disc['Taxas gerais'][i] = 0
+        for i in Situation.SITUATIONS:
+            if not(i[0] in disc['Quantidade'].keys()):
+                disc['Quantidade'][i[0]] = 0
+                disc['Taxas gerais'][i[0]] = 0
         # cria o json
         with open(dis+'.json','w') as f:
             json.dump(disc,f,indent=4)
@@ -76,6 +76,7 @@ def analysis_semestre(df):
 
     return discipline_semestre
 def Main(df):
+
     Analysis = analysis(df)
     Analysis_semestre = analysis_semestre(df)
     matr = counts_matr(df)
-- 
GitLab