Add __repr__, __str__ of range

HyeockJinKim · HyeockJinKim · commit 0ff757f76206 · 2019-09-20T22:58:20.000+09:00
__repr__ print start, stop of range if step is not one, step is also printed Fixes #86
diff --git a/py/range.go b/py/range.go
@@ -6,6 +6,10 @@
 
 package py
 
+import (
+	"strings"
+)
+
 // A python Range object
 // FIXME one day support BigInts too!
 type Range struct {
@@ -104,6 +108,42 @@ func (r *Range) M__iter__() (Object, error) {
 	}, nil
 }
 
+func (r *Range) M__str__() (Object, error) {
+	return r.M__repr__()
+}
+
+func (r *Range) repr() (Object, error) {
+	var b strings.Builder
+	b.WriteString("range(")
+	start, err := ReprAsString(r.Start)
+	if err != nil {
+		return nil, err
+	}
+	stop, err := ReprAsString(r.Stop)
+	if err != nil {
+		return nil, err
+	}
+	b.WriteString(start)
+	b.WriteString(", ")
+	b.WriteString(stop)
+
+	if r.Step != 1 {
+		step, err := ReprAsString(r.Step)
+		if err != nil {
+			return nil, err
+		}
+		b.WriteString(", ")
+		b.WriteString(step)
+	}
+	b.WriteString(")")
+
+	return String(b.String()), nil
+}
+
+func (r *Range) M__repr__() (Object, error) {
+	return r.repr()
+}
+
 func (r *Range) M__len__() (Object, error) {
 	return r.Length, nil
 }
