diff --git a/wrp.go b/wrp.go
index 3b94c4e..f8199b6 100644
--- a/wrp.go
+++ b/wrp.go
@@ -181,9 +181,16 @@ func capture(gourl string, w int64, h int64, s float64, co int, p int64, i bool,
chromedp.Nodes("a", &nodes, chromedp.ByQueryAll))
if err != nil {
- log.Printf("%s %s", c, err)
- fmt.Fprintf(out, "
%s
", err)
- return
+ if err.Error() == "context canceled" {
+ log.Printf("%s Contex cancelled, try again", c)
+ fmt.Fprintf(out, "
%s
-- restarting, try again", err)
+ ctx, cancel = chromedp.NewContext(context.Background())
+ return
+ } else {
+ log.Printf("%s %s", c, err)
+ fmt.Fprintf(out, "
%s
", err)
+ return
+ }
}
log.Printf("%s Landed on: %s, Nodes: %d\n", c, loc, len(nodes))