built in packages know their own names

glycerine · Dec 27, 2018 · d12a0a1 · d12a0a1
1 parent 4fe3294
commit d12a0a1
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/repl/repl2.go b/repl/repl2.go
@@ -152,8 +152,13 @@ func (env *MontyEnv) Init() {
 	}
 
 	// freeze them all, to avoid user overwriting them.
-	for _, pkg := range dict.Map {
+	for nm, pkg := range dict.Map {
 		pkg.Freeze()
+		// name the built in packages
+		sd, ok := pkg.(*starlark.StringDict)
+		if ok {
+			sd.PackageName = nm
+		}
 	}
 
 	// add the struct constructor.