def test_build():
code = Code()
with code.block('if True:'):
code.w('x = 42')
#
src = code.build()
assert src == ("if True:\n" " x = 42")
def test_def_():
code = Code()
with code.def_('foo', ['x', 'y']):
code.w('return x+y')
#
code.compile()
foo = code['foo']
assert foo(1, 2) == 3
def test_formatting():
code = Code()
code.w('{hello}')
assert code.build() == '{hello}'
#
code = Code()
code.w('{hello}', hello='world')
assert code.build() == 'world'
def test_def__default_args():
code = Code()
with code.def_('foo', ['x', ('y', 3)]):
code.w('return x+y')
#
code.compile()
foo = code['foo']
assert foo(1) == 4
assert foo(1, 10) == 11
def test_cpdef():
code = Code()
with code.cpdef_('foo', ['x', 'y']):
code.w('return x+y')
src = code.build()
assert src == ("def foo(x, y):\n" " return x+y")
#
code = Code(pyx=True)
with code.cpdef_('foo', ['x', 'y']):
code.w('return x+y')
src = code.build()
assert src == ("cpdef foo(object x, object y):\n" " return x+y")
def test_cdef_var():
code = Code()
with code.def_('foo', ['x', 'y']):
code.cdef_var('long', 'foo')
code.cdef_var('int', 'bar', 42)
code.w('return')
src = code.build()
assert src == ("def foo(x, y):\n" " return")
#
code = Code(pyx=True)
with code.def_('foo', ['x', 'y']):
code.cdef_var('long', 'foo')
code.cdef_var('int', 'bar', 42)
code.w('return')
src = code.build()
assert src == ("def foo(object x, object y):\n"
" cdef long foo\n"
" cdef int bar = 42\n"
" return")
class ModuleGenerator(object):
def __init__(self, request, convert_case, pyx, version_check, standalone):
self.code = Code(pyx=pyx)
self.request = request
self.convert_case = convert_case
self.pyx = pyx
self.version_check = version_check
self.standalone = standalone
self.allnodes = {} # id -> node
self.children = defaultdict(list) # nodeId -> nested nodes
self.importnames = {} # filename -> import name
self.extra_annotations = defaultdict(list) # obj -> [ann]
self.field_override = {} # obj -> obj
def register_extra_annotation(self, obj, ann):
self.extra_annotations[obj].append(ann.annotation)
def register_field_override(self, origin, target):
self.field_override[origin] = target
def has_annotation(self, obj, anncls):
annotations = self.extra_annotations.get(obj, [])
if obj.annotations is not None:
annotations += obj.annotations
for ann in annotations:
if ann.id == anncls.__id__:
# XXX: probably "annotation" should be taken by the
# constructor
res = anncls()
res.annotation = ann
res.target = obj
return res
return None
def w(self, *args, **kwargs):
self.code.w(*args, **kwargs)
def block(self, *args, **kwargs):
return self.code.block(*args, **kwargs)
def register_import(self, fname):
name = py.path.local(fname).purebasename
name = name.replace('+', 'PLUS')
name = '_%s_capnp' % name
if name in self.importnames.values():
# avoid name clashes
name = '%s_%s' % (name, len(self.filenames))
self.importnames[fname] = name
return name
def generate(self):
self.request.emit(self)
return self.code.build()
def _dump_node(self, node):
def visit(node, deep=0):
print('%s%s: %s' % (' ' * deep, node.which(), node.displayName))
for child in self.children[node.id]:
visit(child, deep + 2)
visit(node)
def _convert_name(self, name):
name = ensure_unicode(name) if PY3 else name
if self.convert_case:
return from_camel_case(name)
else:
return name
def _field_name(self, field):
name = self._convert_name(field.name)
name = self._mangle_name(name)
return name
def _mangle_name(self, name):
if name in keyword.kwlist:
return name + '_'
return name
def declare_enum(self, compile_name, name, items):
# this method cannot go on Node__Enum because it's also called by
# Node__Struct (for __tag__)
items = list(map(repr, items))
ns = self.code.new_scope()
ns.name = compile_name
ns.members = "(%s,)" % (', '.join(items))
ns.prebuilt = [
ns.format('{name}({i})', i=i) for i in range(len(items))
]
ns.prebuilt = ', '.join(ns.prebuilt)
ns.prebuilt = ns.format('({prebuilt},)')
with ns.block("{cdef class} {name}(_BaseEnum):"):
ns.w("__members__ = {members}")
#
# define the _new staticmethod, to create new instances.
if self.pyx:
# on CPython, make a tuple of prebuilt instances for the
# statically known values
ns.w("@staticmethod")
with ns.block(
'cdef _new(long x, __prebuilt={prebuilt}):') as ns:
ns.ww("""
try:
return __prebuilt[x]
except IndexError:
return {name}(x)
""")
else:
# on PyPy, always create a new object, so that the JIT will be
# able to make it virtual
ns.w("@staticmethod")
with ns.block('def _new(x):') as ns:
ns.w('return {name}(x)')
ns.w("_fill_enum({name})")
def def_property(self, ns, name, src):
if self.pyx:
with ns.block('property {name}:', name=name):
with ns.block('def __get__(self):'):
ns.ww(src)
else:
ns.w('@property')
with ns.block('def {name}(self):', name=name):
ns.ww(src)
ns.w()
class ModuleGenerator(object):
def __init__(self, request, convert_case, pyx, standalone):
self.code = Code(pyx=pyx)
self.request = request
self.convert_case = convert_case
self.pyx = pyx
self.standalone = standalone
self.allnodes = {} # id -> node
self.children = defaultdict(list) # nodeId -> nested nodes
self.importnames = {} # filename -> import name
self.extra_annotations = defaultdict(list) # obj -> [ann]
def register_extra_annotation(self, obj, ann):
self.extra_annotations[obj].append(ann.annotation)
def has_annotation(self, obj, anncls):
annotations = self.extra_annotations.get(obj, [])
if obj.annotations is not None:
annotations += obj.annotations
for ann in annotations:
if ann.id == anncls.__id__:
# XXX: probably "annotation" should be taken by the
# constructor
res = anncls()
res.annotation = ann
res.target = obj
return res
return None
def w(self, *args, **kwargs):
self.code.w(*args, **kwargs)
def block(self, *args, **kwargs):
return self.code.block(*args, **kwargs)
def register_import(self, fname):
name = py.path.local(fname).purebasename
name = name.replace('+', 'PLUS')
name = '_%s_capnp' % name
if name in self.importnames.values():
# avoid name clashes
name = '%s_%s' % (name, len(self.filenames))
self.importnames[fname] = name
return name
def generate(self):
self.request.emit(self)
return self.code.build()
def _dump_node(self, node):
def visit(node, deep=0):
print '%s%s: %s' % (' ' * deep, node.which(), node.displayName)
for child in self.children[node.id]:
visit(child, deep+2)
visit(node)
def _convert_name(self, name):
if self.convert_case:
return from_camel_case(name)
else:
return name
def _field_name(self, field):
name = self._convert_name(field.name)
name = self._mangle_name(name)
return name
def _mangle_name(self, name):
if name in keyword.kwlist:
return name + '_'
return name
def declare_enum(self, var_name, enum_name, items):
# this method cannot go on Node__Enum because it's also called by
# Node__Struct (for __tag__)
items = map(repr, items)
decl = "%s = _enum(%r, [%s])" % (var_name, enum_name, ', '.join(items))
self.w(decl)
def def_property(self, ns, name, src):
if self.pyx:
with ns.block('property {name}:', name=name):
with ns.block('def __get__(self):'):
ns.ww(src)
else:
ns.w('@property')
with ns.block('def {name}(self):', name=name):
ns.ww(src)
ns.w()
def test_global_kwargs_override():
code = Code()
code.global_scope.x = 42
code.w('return {x}', x=52)
assert code.build() == 'return 52'
def test_global_kwargs():
code = Code()
code.global_scope.x = 42
code.w('return {x}')
assert code.build() == 'return 42'
def test_compile():
code = Code()
code.w('x = 42')
code.compile()
assert code['x'] == 42