py: dict.__setitem__ and dict.__getitem__

ncw · ncw · commit 7ac88d93e50e · 2015-05-15T18:27:14.000+01:00
diff --git a/py/dict.go b/py/dict.go
@@ -37,3 +37,23 @@ func (d StringDict) Copy() StringDict {
 	}
 	return e
 }
+
+func (d StringDict) M__getitem__(key Object) Object {
+	str, ok := key.(String)
+	if ok {
+		res, ok := d[string(str)]
+		if ok {
+			return res
+		}
+	}
+	panic(ExceptionNewf(KeyError, "%v", key))
+}
+
+func (d StringDict) M__setitem__(key, value Object) Object {
+	str, ok := key.(String)
+	if !ok {
+		panic("FIXME can only have string keys!")
+	}
+	d[string(str)] = value
+	return None
+}
