summaryrefslogtreecommitdiff
path: root/cmd/getpipelinebook
diff options
context:
space:
mode:
authorNick White <git@njw.name>2019-12-17 12:47:59 +0000
committerNick White <git@njw.name>2019-12-17 12:47:59 +0000
commit6c1947940fa6a8299d7ac0c880781ed40d3affe3 (patch)
treedb2a0a06015c91b33fb687395d734efc09ef3371 /cmd/getpipelinebook
parent353f5c31efca61e60b5ea5a30b77c8a649cce62b (diff)
Add pdf flag to getpipelinebook
Diffstat (limited to 'cmd/getpipelinebook')
-rw-r--r--cmd/getpipelinebook/main.go35
1 files changed, 20 insertions, 15 deletions
diff --git a/cmd/getpipelinebook/main.go b/cmd/getpipelinebook/main.go
index d0f2e3b..5bb2e28 100644
--- a/cmd/getpipelinebook/main.go
+++ b/cmd/getpipelinebook/main.go
@@ -11,7 +11,7 @@ import (
"rescribe.xyz/bookpipeline"
)
-const usage = `Usage: getpipelinebook [-a] [-v] bookname
+const usage = `Usage: getpipelinebook [-a] [-pdf] [-v] bookname
Downloads the pipeline results for a book.
@@ -38,8 +38,20 @@ type Pipeliner interface {
WIPStorageId() string
}
+func getpdfs(conn Pipeliner, l *log.Logger, bookname string) {
+ for _, suffix := range []string { ".colour.pdf", ".binarised.pdf" } {
+ fn := filepath.Join(bookname, bookname + suffix)
+ l.Println("Downloading PDF", fn)
+ err := conn.Download(conn.WIPStorageId(), fn, fn)
+ if err != nil {
+ log.Printf("Failed to download %s: %s\n", fn, err)
+ }
+ }
+}
+
func main() {
all := flag.Bool("a", false, "Get all files for book")
+ pdf := flag.Bool("pdf", false, "Only download PDFs")
verbose := flag.Bool("v", false, "Verbose")
flag.Usage = func() {
fmt.Fprintf(flag.CommandLine.Output(), usage)
@@ -93,6 +105,11 @@ func main() {
return
}
+ if *pdf {
+ getpdfs(conn, verboselog, bookname)
+ return
+ }
+
verboselog.Println("Downloading best file")
fn := filepath.Join(bookname, "best")
err = conn.Download(conn.WIPStorageId(), fn, fn)
@@ -117,22 +134,10 @@ func main() {
}
verboselog.Println("Downloading PDF files")
- fn = filepath.Join(bookname, bookname + ".colour.pdf")
- verboselog.Println("Downloading file", fn)
- err = conn.Download(conn.WIPStorageId(), fn, fn)
- if err != nil {
- log.Printf("Failed to download %s: %s\n", fn, err)
- }
- fn = filepath.Join(bookname, bookname + ".binarised.pdf")
- verboselog.Println("Downloading file", fn)
- err = conn.Download(conn.WIPStorageId(), fn, fn)
- if err != nil {
- log.Printf("Failed to download %s: %s\n", fn, err)
- }
+ getpdfs(conn, verboselog, bookname)
- analyses := []string{"conf", "graph.png"}
verboselog.Println("Downloading analysis files")
- for _, a := range analyses {
+ for _, a := range []string{"conf", "graph.png"} {
fn = filepath.Join(bookname, a)
verboselog.Println("Downloading file", fn)
err = conn.Download(conn.WIPStorageId(), fn, fn)