60 lines
993 B
Go
60 lines
993 B
Go
package main
|
|
|
|
import (
|
|
"encoding/csv"
|
|
"encoding/json"
|
|
"flag"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"strconv"
|
|
)
|
|
|
|
var flagIn = flag.String("in", "", "input file")
|
|
var flagOut = flag.String("out", "country.csv", "output file")
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
in := *flagIn
|
|
out := *flagOut
|
|
|
|
data := make(map[string][]int, 0)
|
|
|
|
log.Println("Reading file " + in)
|
|
raw, err := ioutil.ReadFile(in)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
log.Println("Parsing file " + in)
|
|
err = json.Unmarshal(raw, &data)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
log.Println("File " + in + " read successfully")
|
|
|
|
log.Println("Converting to csv")
|
|
odata := make([][]string, 0)
|
|
for cc, ttls := range data {
|
|
for _, ttl := range ttls {
|
|
odata = append(odata, []string{
|
|
cc,
|
|
strconv.Itoa(ttl),
|
|
})
|
|
}
|
|
}
|
|
|
|
log.Println("Creating file " + out)
|
|
outFile, err := os.Create(out)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
log.Println("Encoding csv")
|
|
w := csv.NewWriter(outFile)
|
|
err = w.WriteAll(odata)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|