Add dumptab and writetab tools for sheet population

2026-08-16 10:13:04 -07:00
parent bbdeb3e91e
commit 516ef2dea0
2 changed files with 144 additions and 0 deletions
@@ -0,0 +1,58 @@
+// 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)
+}