mirror of https://github.com/python/cpython
Fix _convert_NAME() so that it doesn't store locals for class bodies.
Fix list comp code generation -- emit GET_ITER instead of Const(0) after the list. Add CO_GENERATOR flag to generators. Get CO_xxx flags from the new module
This commit is contained in:
parent
017cb2c7d8
commit
71ebc3359b
|
@ -1,6 +1,5 @@
|
|||
# code flags
|
||||
CO_VARARGS = 1
|
||||
CO_VARKEYWORDS = 2
|
||||
from new import * # import all the CO_xxx flags
|
||||
del classobj, code, function, instance, instancemethod, module
|
||||
|
||||
# operation flags
|
||||
OP_ASSIGN = 'OP_ASSIGN'
|
||||
|
|
|
@ -7,6 +7,8 @@ import sys
|
|||
import types
|
||||
|
||||
from compiler import misc
|
||||
from compiler.consts import CO_OPTIMIZED, CO_NEWLOCALS, CO_VARARGS, \
|
||||
CO_VARKEYWORDS
|
||||
|
||||
def xxx_sort(l):
|
||||
l = l[:]
|
||||
|
@ -311,11 +313,6 @@ class Block:
|
|||
return contained
|
||||
|
||||
# flags for code objects
|
||||
CO_OPTIMIZED = 0x0001
|
||||
CO_NEWLOCALS = 0x0002
|
||||
CO_VARARGS = 0x0004
|
||||
CO_VARKEYWORDS = 0x0008
|
||||
CO_NESTED = 0x0010
|
||||
|
||||
# the FlowGraph is transformed in place; it exists in one of these states
|
||||
RAW = "RAW"
|
||||
|
@ -503,7 +500,8 @@ class PyFlowGraph(FlowGraph):
|
|||
return self._lookupName(arg, self.names)
|
||||
|
||||
def _convert_NAME(self, arg):
|
||||
self._lookupName(arg, self.varnames)
|
||||
if self.klass is None:
|
||||
self._lookupName(arg, self.varnames)
|
||||
return self._lookupName(arg, self.names)
|
||||
_convert_STORE_NAME = _convert_NAME
|
||||
_convert_DELETE_NAME = _convert_NAME
|
||||
|
@ -739,9 +737,8 @@ class StackDepthTracker:
|
|||
'DELETE_SUBSCR': -2,
|
||||
# PRINT_EXPR?
|
||||
'PRINT_ITEM': -1,
|
||||
'LOAD_LOCALS': 1,
|
||||
'RETURN_VALUE': -1,
|
||||
'EXEC_STMT': -2,
|
||||
'EXEC_STMT': -3,
|
||||
'BUILD_CLASS': -2,
|
||||
'STORE_NAME': -1,
|
||||
'STORE_ATTR': -2,
|
||||
|
@ -756,6 +753,7 @@ class StackDepthTracker:
|
|||
# close enough...
|
||||
'SETUP_EXCEPT': 3,
|
||||
'SETUP_FINALLY': 3,
|
||||
'FOR_ITER': 1,
|
||||
}
|
||||
# use pattern match
|
||||
patterns = [
|
||||
|
|
|
@ -11,8 +11,9 @@ from cStringIO import StringIO
|
|||
from compiler import ast, parse, walk
|
||||
from compiler import pyassem, misc, future, symbols
|
||||
from compiler.consts import SC_LOCAL, SC_GLOBAL, SC_FREE, SC_CELL
|
||||
from compiler.pyassem import CO_VARARGS, CO_VARKEYWORDS, CO_NEWLOCALS,\
|
||||
CO_NESTED, TupleArg
|
||||
from compiler.consts import CO_VARARGS, CO_VARKEYWORDS, CO_NEWLOCALS,\
|
||||
CO_NESTED, CO_GENERATOR
|
||||
from compiler.pyassem import TupleArg
|
||||
|
||||
# Do we have Python 1.x or Python 2.x?
|
||||
try:
|
||||
|
@ -495,10 +496,10 @@ class CodeGenerator:
|
|||
anchor = self.newBlock()
|
||||
|
||||
self.visit(node.list)
|
||||
self.visit(ast.Const(0))
|
||||
self.emit('GET_ITER')
|
||||
self.nextBlock(start)
|
||||
self.emit('SET_LINENO', node.lineno)
|
||||
self.emit('FOR_LOOP', anchor)
|
||||
self.emit('FOR_ITER', anchor)
|
||||
self.nextBlock()
|
||||
self.visit(node.assign)
|
||||
return start, anchor
|
||||
|
@ -1199,6 +1200,8 @@ class NestedFunctionCodeGenerator(AbstractFunctionCode,
|
|||
self.__super_init(func, filename, scopes, isLambda, class_name)
|
||||
self.graph.setFreeVars(self.scope.get_free_vars())
|
||||
self.graph.setCellVars(self.scope.get_cell_vars())
|
||||
if self.scope.generator is not None:
|
||||
self.graph.setFlag(CO_GENERATOR)
|
||||
## self.graph.setFlag(CO_NESTED)
|
||||
|
||||
class AbstractClassCode:
|
||||
|
|
|
@ -1,6 +1,5 @@
|
|||
# code flags
|
||||
CO_VARARGS = 1
|
||||
CO_VARKEYWORDS = 2
|
||||
from new import * # import all the CO_xxx flags
|
||||
del classobj, code, function, instance, instancemethod, module
|
||||
|
||||
# operation flags
|
||||
OP_ASSIGN = 'OP_ASSIGN'
|
||||
|
|
|
@ -7,6 +7,8 @@ import sys
|
|||
import types
|
||||
|
||||
from compiler import misc
|
||||
from compiler.consts import CO_OPTIMIZED, CO_NEWLOCALS, CO_VARARGS, \
|
||||
CO_VARKEYWORDS
|
||||
|
||||
def xxx_sort(l):
|
||||
l = l[:]
|
||||
|
@ -311,11 +313,6 @@ class Block:
|
|||
return contained
|
||||
|
||||
# flags for code objects
|
||||
CO_OPTIMIZED = 0x0001
|
||||
CO_NEWLOCALS = 0x0002
|
||||
CO_VARARGS = 0x0004
|
||||
CO_VARKEYWORDS = 0x0008
|
||||
CO_NESTED = 0x0010
|
||||
|
||||
# the FlowGraph is transformed in place; it exists in one of these states
|
||||
RAW = "RAW"
|
||||
|
@ -503,7 +500,8 @@ class PyFlowGraph(FlowGraph):
|
|||
return self._lookupName(arg, self.names)
|
||||
|
||||
def _convert_NAME(self, arg):
|
||||
self._lookupName(arg, self.varnames)
|
||||
if self.klass is None:
|
||||
self._lookupName(arg, self.varnames)
|
||||
return self._lookupName(arg, self.names)
|
||||
_convert_STORE_NAME = _convert_NAME
|
||||
_convert_DELETE_NAME = _convert_NAME
|
||||
|
@ -739,9 +737,8 @@ class StackDepthTracker:
|
|||
'DELETE_SUBSCR': -2,
|
||||
# PRINT_EXPR?
|
||||
'PRINT_ITEM': -1,
|
||||
'LOAD_LOCALS': 1,
|
||||
'RETURN_VALUE': -1,
|
||||
'EXEC_STMT': -2,
|
||||
'EXEC_STMT': -3,
|
||||
'BUILD_CLASS': -2,
|
||||
'STORE_NAME': -1,
|
||||
'STORE_ATTR': -2,
|
||||
|
@ -756,6 +753,7 @@ class StackDepthTracker:
|
|||
# close enough...
|
||||
'SETUP_EXCEPT': 3,
|
||||
'SETUP_FINALLY': 3,
|
||||
'FOR_ITER': 1,
|
||||
}
|
||||
# use pattern match
|
||||
patterns = [
|
||||
|
|
|
@ -11,8 +11,9 @@ from cStringIO import StringIO
|
|||
from compiler import ast, parse, walk
|
||||
from compiler import pyassem, misc, future, symbols
|
||||
from compiler.consts import SC_LOCAL, SC_GLOBAL, SC_FREE, SC_CELL
|
||||
from compiler.pyassem import CO_VARARGS, CO_VARKEYWORDS, CO_NEWLOCALS,\
|
||||
CO_NESTED, TupleArg
|
||||
from compiler.consts import CO_VARARGS, CO_VARKEYWORDS, CO_NEWLOCALS,\
|
||||
CO_NESTED, CO_GENERATOR
|
||||
from compiler.pyassem import TupleArg
|
||||
|
||||
# Do we have Python 1.x or Python 2.x?
|
||||
try:
|
||||
|
@ -495,10 +496,10 @@ class CodeGenerator:
|
|||
anchor = self.newBlock()
|
||||
|
||||
self.visit(node.list)
|
||||
self.visit(ast.Const(0))
|
||||
self.emit('GET_ITER')
|
||||
self.nextBlock(start)
|
||||
self.emit('SET_LINENO', node.lineno)
|
||||
self.emit('FOR_LOOP', anchor)
|
||||
self.emit('FOR_ITER', anchor)
|
||||
self.nextBlock()
|
||||
self.visit(node.assign)
|
||||
return start, anchor
|
||||
|
@ -1199,6 +1200,8 @@ class NestedFunctionCodeGenerator(AbstractFunctionCode,
|
|||
self.__super_init(func, filename, scopes, isLambda, class_name)
|
||||
self.graph.setFreeVars(self.scope.get_free_vars())
|
||||
self.graph.setCellVars(self.scope.get_cell_vars())
|
||||
if self.scope.generator is not None:
|
||||
self.graph.setFlag(CO_GENERATOR)
|
||||
## self.graph.setFlag(CO_NESTED)
|
||||
|
||||
class AbstractClassCode:
|
||||
|
|
Loading…
Reference in New Issue