Add import csv command
This commit is contained in:
parent
69f15a4de0
commit
ebcd0375aa
2 changed files with 37 additions and 13 deletions
44
cmd/add.go
44
cmd/add.go
|
@ -2,6 +2,7 @@ package cmd
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bufio"
|
"bufio"
|
||||||
|
"encoding/csv"
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
"os"
|
||||||
"strconv"
|
"strconv"
|
||||||
|
@ -59,17 +60,13 @@ func GovetAddCmd() *cobra.Command {
|
||||||
}
|
}
|
||||||
|
|
||||||
func CsvAddCmd() *cobra.Command {
|
func CsvAddCmd() *cobra.Command {
|
||||||
cmd := &cobra.Command{
|
return &cobra.Command{
|
||||||
Use: "csv",
|
Use: "csv",
|
||||||
Short: "Generates the tickets reading a csv file",
|
Short: "Generates the tickets reading a csv file",
|
||||||
Example: ` campaigner add csv --file tickets.csv`,
|
Example: ` campaigner add csv tickets.csv`,
|
||||||
Args: cobra.NoArgs,
|
Args: cobra.ExactArgs(1),
|
||||||
RunE: csvAddCmdF,
|
Run: csvAddCmdF,
|
||||||
}
|
}
|
||||||
|
|
||||||
cmd.Flags().BoolP("file-only", "f", false, "Generates one ticket per file instead of per match")
|
|
||||||
|
|
||||||
return cmd
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func AddCmd() *cobra.Command {
|
func AddCmd() *cobra.Command {
|
||||||
|
@ -150,6 +147,33 @@ func govetAddCmdF(_ *cobra.Command, _ []string) error {
|
||||||
return fmt.Errorf("not implemented yet")
|
return fmt.Errorf("not implemented yet")
|
||||||
}
|
}
|
||||||
|
|
||||||
func csvAddCmdF(cmd *cobra.Command, _ []string) error {
|
func csvAddCmdF(cmd *cobra.Command, args []string) {
|
||||||
return fmt.Errorf("not implemented yet")
|
file, err := os.Open(args[0])
|
||||||
|
if err != nil {
|
||||||
|
ErrorAndExit(cmd, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
cmp, err := campaign.Read()
|
||||||
|
if err != nil {
|
||||||
|
ErrorAndExit(cmd, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
csvReader := csv.NewReader(bufio.NewReader(file))
|
||||||
|
records, err := csvReader.ReadAll()
|
||||||
|
if err != nil {
|
||||||
|
ErrorAndExit(cmd, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
headers := records[0]
|
||||||
|
for _, line := range records[1:] {
|
||||||
|
data := map[string]interface{}{}
|
||||||
|
for i, header := range headers {
|
||||||
|
data[header] = line[i]
|
||||||
|
}
|
||||||
|
cmp.Tickets = append(cmp.Tickets, &model.Ticket{Data: data})
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := campaign.Save(cmp); err != nil {
|
||||||
|
ErrorAndExit(cmd, err)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue