From a97552ad71454b43abd93b1a985cd0ec6edc5fb4 Mon Sep 17 00:00:00 2001 From: Antonia Karaisl Date: Tue, 28 Jul 2020 14:15:08 +0100 Subject: Autodetect queue for booktopipeline based on file extension --- cmd/booktopipeline/main.go | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) (limited to 'cmd') diff --git a/cmd/booktopipeline/main.go b/cmd/booktopipeline/main.go index 609c3b3..b0605c6 100644 --- a/cmd/booktopipeline/main.go +++ b/cmd/booktopipeline/main.go @@ -78,7 +78,7 @@ func main() { } else { bookname = filepath.Base(bookdir) } - + if *verbose { verboselog = log.New(os.Stdout, "", log.LstdFlags) } else { @@ -106,6 +106,19 @@ func main() { } else { qid = conn.PreQueueId() } + + pngdirs, _ := filepath.Glob(bookdir + "/*.png") + jpgdirs, _ := filepath.Glob(bookdir + "/*.jpg") + pngcount := len(pngdirs) + jpgcount := len(jpgdirs) + if pngcount > jpgcount { + qid = conn.WipeQueueId() + fmt.Println("Uploading book to wipe-only queue") + } else { + qid = conn.PreQueueId() + fmt.Println("Uploading book to preprocess queue") + } + verboselog.Println("Walking", bookdir) walker := make(fileWalk) -- cgit v1.2.1-24-ge1ad