Improvement on the CSV reading and writing function

Author: fillol

CSV/createCSV.go

@@ -0,0 +1,68 @@
+package CSV
+
+import (
+	"github.com/Darklabel91/Summary_Classifier"
+	"github.com/Darklabel91/Summary_Classifier/Struct"
+)
+
+func CreateCSVs(raw []Struct.Raw_decision, nameResultFolder string) {
+	var totalInfered []Struct.Infered_decision
+	var exOfficioReview []Struct.Infered_decision
+	var diligence []Struct.Infered_decision
+	var affected []Struct.Infered_decision
+	var partial []Struct.Infered_decision
+	var groundless []Struct.Infered_decision
+	var hasGround []Struct.Infered_decision
+	var noInfo []Struct.Infered_decision
+	var notMap []Struct.Infered_decision
+
+	for i := 0; i < len(raw); i++ {
+		emp := Summary_Classifier.SummaryClassifier(raw[i].Summary, raw[i].Identifier, raw[i].Court)
+		totalInfered = append(totalInfered, emp)
+		if emp.Class == "Reexame Necessário" {
+			exOfficioReview = append(exOfficioReview, emp)
+		} else if emp.Class == "Convertido em Diligência" {
+			diligence = append(diligence, emp)
+		} else if emp.Class == "Prejudicado" {
+			affected = append(affected, emp)
+		} else if emp.Class == "Parcial Provimento" {
+			partial = append(partial, emp)
+		} else if emp.Class == "Improvimento" {
+			groundless = append(groundless, emp)
+		} else if emp.Class == "Provimento" {
+			hasGround = append(hasGround, emp)
+		} else if emp.Class == "Sem Informação" {
+			noInfo = append(noInfo, emp)
+		} else if emp.Class == "Não Mapeado" {
+			notMap = append(notMap, emp)
+		}
+
+	}
+
+	exportCSV("totalInfered", nameResultFolder, totalInfered)
+
+	if len(exOfficioReview) != 0 {
+		exportCSV("exOfficioReview", nameResultFolder, exOfficioReview)
+	}
+	if len(diligence) != 0 {
+		exportCSV("diligence", nameResultFolder, diligence)
+	}
+	if len(affected) != 0 {
+		exportCSV("affected", nameResultFolder, affected)
+	}
+	if len(partial) != 0 {
+		exportCSV("partial", nameResultFolder, partial)
+	}
+	if len(groundless) != 0 {
+		exportCSV("groundless", nameResultFolder, groundless)
+	}
+	if len(hasGround) != 0 {
+		exportCSV("hasGround", nameResultFolder, hasGround)
+	}
+	if len(noInfo) != 0 {
+		exportCSV("noInfo", nameResultFolder, noInfo)
+	}
+	if len(notMap) != 0 {
+		exportCSV("notMap", nameResultFolder, notMap)
+	}
+}

CSV/write.go

@@ -2,7 +2,6 @@ package CSV
 
 import (
 	"encoding/csv"
-	"fmt"
 	"github.com/Darklabel91/Summary_Classifier/Error"
 	"github.com/Darklabel91/Summary_Classifier/Struct"
 	"os"
@@ -13,11 +12,10 @@ func create(p string) (*os.File, error) {
 	if err := os.MkdirAll(filepath.Dir(p), 0770); err != nil {
 		return nil, err
 	}
-	fmt.Println("Criou .csv")
 	return os.Create(p)
 }
 
-func ExportCSV(nameFile string, nameFolder string, result []Struct.Infered_decision) {
+func exportCSV(nameFile string, nameFolder string, result []Struct.Infered_decision) {
 	empData := [][]string{}
 
 	for i := 0; i < len(result); i++ {

Summary_Classifier.go

@@ -1,6 +1,7 @@
 package Summary_Classifier
 
 import (
+	"fmt"
 	"github.com/Darklabel91/Summary_Classifier/CSV"
 	"github.com/Darklabel91/Summary_Classifier/Classifier"
 	"github.com/Darklabel91/Summary_Classifier/Struct"
@@ -24,13 +25,7 @@ func SummaryClassifier(summary string, identifier string, court string) Struct.I
 
 }
 func SummaryClassifierCSV(rawDecisionPath string, separator rune, nameResultFolder string) {
-	var infered []Struct.Infered_decision
-
 	raw := CSV.ReadCsvFile(rawDecisionPath, separator)
-
-	for i := 0; i < len(raw); i++ {
-		infered = append(infered, SummaryClassifier(raw[i].Summary, raw[i].Identifier, raw[i].Court))
-	}
-
-	CSV.ExportCSV("Infered Decision", nameResultFolder, infered)
+	CSV.CreateCSVs(raw, nameResultFolder)
+	fmt.Println("Files created")
 }