Add SetCtx and SetCtxer

ncw · ncw · commit 817f25d8c826 · 2014-11-23T20:59:33.000Z
diff --git a/ast/ast.go b/ast/ast.go
@@ -273,7 +273,7 @@ func (o *StmtBase) stmtNode() {}
 type FunctionDef struct {
 	StmtBase
 	Name          Identifier
-	Args          Arguments
+	Args          *Arguments
 	Body          []Stmt
 	DecoratorList []Expr
 	Returns       Expr
@@ -283,7 +283,7 @@ type ClassDef struct {
 	StmtBase
 	Name          Identifier
 	Bases         []Expr
-	Keywords      []Keyword
+	Keywords      []*Keyword
 	Starargs      Expr
 	Kwargs        Expr
 	Body          []Stmt
@@ -498,7 +498,7 @@ type Call struct {
 	ExprBase
 	Function Expr
 	Args     []Expr
-	Keywords []Keyword
+	Keywords []*Keyword
 	Starargs Expr
 	Kwargs   Expr
 }
@@ -534,37 +534,76 @@ type Attribute struct {
 	Ctx   ExprContext
 }
 
+// ExprNodes which have a settable context implement this
+type SetCtxer interface {
+	SetCtx(ExprContext)
+}
+
+func (o *Attribute) SetCtx(Ctx ExprContext) { o.Ctx = Ctx }
+
+var _ = SetCtxer((*Attribute)(nil))
+
 type Subscript struct {
 	ExprBase
 	Value Expr
 	Slice Slicer
 	Ctx   ExprContext
 }
 
+func (o *Subscript) SetCtx(Ctx ExprContext) { o.Ctx = Ctx }
+
+var _ = SetCtxer((*Subscript)(nil))
+
 type Starred struct {
 	ExprBase
 	Value Expr
 	Ctx   ExprContext
 }
 
+func (o *Starred) SetCtx(Ctx ExprContext) { o.Ctx = Ctx }
+
+var _ = SetCtxer((*Starred)(nil))
+
 type Name struct {
 	ExprBase
 	Id  Identifier
 	Ctx ExprContext
 }
 
+func (o *Name) SetCtx(Ctx ExprContext) { o.Ctx = Ctx }
+
+var _ = SetCtxer((*Name)(nil))
+
 type List struct {
 	ExprBase
 	Elts []Expr
 	Ctx  ExprContext
 }
 
+func (o *List) SetCtx(Ctx ExprContext) {
+	o.Ctx = Ctx
+	for i := range o.Elts {
+		o.Elts[i].(SetCtxer).SetCtx(Ctx)
+	}
+}
+
+var _ = SetCtxer((*List)(nil))
+
 type Tuple struct {
 	ExprBase
 	Elts []Expr
 	Ctx  ExprContext
 }
 
+func (o *Tuple) SetCtx(Ctx ExprContext) {
+	o.Ctx = Ctx
+	for i := range o.Elts {
+		o.Elts[i].(SetCtxer).SetCtx(Ctx)
+	}
+}
+
+var _ = SetCtxer((*Tuple)(nil))
+
 // ------------------------------------------------------------
 // Slicer nodes
 // ------------------------------------------------------------
