Support for the yield statement.

This commit is contained in:
Darryl Pogue
2011-09-18 22:31:09 -07:00
parent dfd8a620f2
commit 51ca98d766
2 changed files with 27 additions and 4 deletions

View File

@@ -160,13 +160,19 @@ private:
class ASTReturn : public ASTNode {
public:
ASTReturn(PycRef<ASTNode> value)
: ASTNode(NODE_RETURN), m_value(value) { }
enum RetType {
RETURN, YIELD
};
ASTReturn(PycRef<ASTNode> value, RetType rettype = RETURN)
: ASTNode(NODE_RETURN), m_value(value), m_rettype(rettype) { }
PycRef<ASTNode> value() const { return m_value; }
RetType rettype() const { return m_rettype; }
private:
PycRef<ASTNode> m_value;
RetType m_rettype;
};