mirror/ascode
1
1
Fork 0
mirror of https://github.com/mcuadros/ascode synced 2024-11-26 06:01:08 +01:00
Code Issues

starlark/types: Attribute.Get, support for map attributes

Browse Source
This commit is contained in:
Máximo Cuadros 2020-04-13 10:34:59 +02:00
parent 8b962d26cc
commit 6fc80cda96
No known key found for this signature in database
GPG Key ID: 17A5DFEDC735AE4B
2 changed files with 33 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
starlark/types/attribute.go
View File

@ -155,9 +155,35 @@ func (c *Attribute) Index(i int) starlark.Value {
return NewAttributeWithPath(c.r, *c.t.ListElementType(), c.name, path)
}
if c.t.IsMapType() {
return NewAttributeWithPath(c.r, c.t, c.name, path)
}
return starlark.None
}
// Get honors the starlark.Mapping interface.
func (c *Attribute) Get(key starlark.Value) (v starlark.Value, found bool, err error) {
switch vKey := key.(type) {
case starlark.Int:
if !c.t.IsSetType() && !c.t.IsListType() && !c.t.IsMapType() {
return nil, false, fmt.Errorf("%s does not support index", c.name)
}
index, _ := vKey.Int64()
return c.Index(int(index)), true, nil
case starlark.String:
if !c.t.IsMapType() {
return nil, false, fmt.Errorf("%s it's not a dict", c.name)
}
path := fmt.Sprintf("%s.%s", c.path, vKey.GoString())
return NewAttributeWithPath(c.r, c.t, c.name, path), true, nil
default:
return nil, false, fmt.Errorf("%s: unexpected key type %s", c.name, key.Type())
}
}
// Len honors the starlark.Indexable interface.
func (c *Attribute) Len() int {
if !c.t.IsSetType() && !c.t.IsListType() {

8
starlark/types/testdata/attribute.star vendored
View File

@ -47,6 +47,12 @@ assert.eq(str(cluster.master_auth.client_certificate), "${google_container_clust
# attr non-object
assert.fails(lambda: web.ami.foo, "Attribute<string> it's not a object")
# fn wrapping
assert.eq(str(fn("base64encode", web.ami)), "${base64encode(data.aws_ami.id_2.id)}")
# attribute of dict
k8s = tf.provider("kubernetes")
secret = k8s.data.secret("foo")
assert.eq(str(secret.data["qux"]), "${data.kubernetes_secret.foo.data.qux}")
assert.eq(str(secret.data["qux"][0]), "${data.kubernetes_secret.foo.data.qux.0}")