feat(progress): added template update option

snakeice · Jul 18, 2020 · 5d29b69 · 5d29b69
1 parent ffa9e74
commit 5d29b69
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/progress.go b/progress.go
@@ -292,3 +292,11 @@ func (pb *Progress) NewProxyWriter(r io.Writer) *Writer {
 	pb.Units = format.U_BYTES
 	return &Writer{r, pb}
 }
+
+func (pb *Progress) SetTemplate(template string) {
+	pb.frameParser.UpdateTemplate(template)
+}
+
+func (pb *Progress) ResetTemplate() {
+	pb.frameParser.UpdateTemplate(DefaultTemplate)
+}