1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
"""
Implements flow graphs for Python callables
"""
from rpython.flowspace.model import FunctionGraph, Constant, Variable
from rpython.flowspace.framestate import FrameState
class PyGraph(FunctionGraph):
"""
Flow graph for a Python function
"""
def __init__(self, func, code):
from rpython.flowspace.flowcontext import SpamBlock
locals = [None] * code.co_nlocals
for i in range(code.formalargcount):
locals[i] = Variable(code.co_varnames[i])
state = FrameState(locals, [], None, [], 0)
initialblock = SpamBlock(state)
super(PyGraph, self).__init__(self._sanitize_funcname(func), initialblock)
self.func = func
self.signature = code.signature
self.defaults = func.func_defaults or ()
@staticmethod
def _sanitize_funcname(func):
# CallableFactory.pycall may add class_ to functions that are methods
name = func.func_name
class_ = getattr(func, 'class_', None)
if class_ is not None:
name = '%s.%s' % (class_.__name__, name)
for c in "<>&!":
name = name.replace(c, '_')
return name
|
