59 lines
1.5 KiB
Go
59 lines
1.5 KiB
Go
// Command dumptab writes one sheet tab to a local CSV file.
|
|
package main
|
|
|
|
import (
|
|
"context"
|
|
"encoding/csv"
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"strings"
|
|
|
|
"heliosian/internal/data"
|
|
"google.golang.org/api/option"
|
|
"google.golang.org/api/sheets/v4"
|
|
)
|
|
|
|
func main() {
|
|
sheet := flag.String("sheet", "", "spreadsheet id")
|
|
tab := flag.String("tab", "", "tab title")
|
|
out := flag.String("out", "", "output csv path")
|
|
flag.Parse()
|
|
if *sheet == "" || *tab == "" || *out == "" {
|
|
log.Fatal("[ERROR] -sheet, -tab, and -out are required")
|
|
}
|
|
svc, err := sheets.NewService(context.Background(),
|
|
option.WithCredentialsFile(data.KeyFile),
|
|
option.WithScopes(sheets.SpreadsheetsReadonlyScope))
|
|
if err != nil {
|
|
log.Fatalf("[ERROR] create sheets client: %v", err)
|
|
}
|
|
resp, err := svc.Spreadsheets.Values.Get(*sheet, "'"+strings.ReplaceAll(*tab, "'", "''")+"'").Do()
|
|
if err != nil {
|
|
log.Fatalf("[ERROR] read tab %s: %v", *tab, err)
|
|
}
|
|
f, err := os.Create(*out)
|
|
if err != nil {
|
|
log.Fatalf("[ERROR] create %s: %v", *out, err)
|
|
}
|
|
w := csv.NewWriter(f)
|
|
for _, row := range resp.Values {
|
|
record := make([]string, len(row))
|
|
for i, cell := range row {
|
|
record[i] = fmt.Sprint(cell)
|
|
}
|
|
if err := w.Write(record); err != nil {
|
|
log.Fatalf("[ERROR] write row: %v", err)
|
|
}
|
|
}
|
|
w.Flush()
|
|
if err := w.Error(); err != nil {
|
|
log.Fatalf("[ERROR] flush csv: %v", err)
|
|
}
|
|
if err := f.Close(); err != nil {
|
|
log.Fatalf("[ERROR] close %s: %v", *out, err)
|
|
}
|
|
log.Printf("wrote %d rows to %s", len(resp.Values), *out)
|
|
}
|