def sqlcontentclass_engine_created_handler(self, source, target):
"""create and register the handler for the IEngineCreatedEvent.
"""
if source.stereotype('pyegg:stub'):
return
targetpack=read_target_node(source,target.target)
targetpack['saconfig.py']=Module()
module = targetpack['saconfig.py']
imps = Imports(module)
# check if one of the parent packages has the z3c_saconfig stereotype
z3c_saconfig=get_z3c_saconfig(source)
# add engine-created handler
if z3c_saconfig:
eggtgv=TaggedValues(z3c_saconfig)
engine_name = eggtgv.direct(
'engine_name', 'sql:z3c_saconfig', 'default')
globalfuncs = [f for f in module.filtereditems(IFunction)]
globalfuncnames = [f.functionname for f in globalfuncs]
if 'engineCreatedHandler' not in globalfuncnames:
imps.set('z3c.saconfig.interfaces',
[['IEngineCreatedEvent', None]])
imps.set('zope', [['component', None]])
imps.set('z3c.saconfig.interfaces',[['IEngineFactory',None]])
imps.set('sqlalchemy.ext.declarative','declarative_base')
#att = [att for att in module.filtereditems(IAttribute) \
# if att.targets == ['Base']][0]
bases=module.attributes('Base')
if not bases:
base=Attribute('Base','declarative_base()')
module.insertafterimports(base)
else:
base=bases[0]
ff = Function('engineCreatedHandler')
# 'engineCreatedHandler'
ff.__name__ = ff.uuid
ff.args = ('event',)
dec = Decorator('component.adapter')
dec.__name__ = dec.uuid
dec.args = ('IEngineCreatedEvent',)
ff.insertfirst(dec)
block_lines = [
"fact = component.queryUtility(IEngineFactory, '%s')" \
% engine_name,
"if fact and fact._args == event.engine_args:",
" Base.metadata.create_all(event.engine)",
]
block = Block('\n'.join(block_lines))
block.__name__ = block.uuid
ff.insertlast(block)
module.insertafter(ff, base)
prov = Block('component.provideHandler(engineCreatedHandler)')
prov.__name__ = prov.uuid
module.insertafter(prov, ff)
def create_site_root(self, source, target):
targetclass = read_target_node(source, target.target)
module = targetclass.parent
if not module.functions('bootstrap'):
func = Function('bootstrap')
func.insertlast(Block(bootstrap_templ % source.name))
module.insertafter(func, targetclass)
else:
func = module.functions('bootstrap')[0]
# mark the class because it has to be referenced by main
token('site_root', True, site_root=source, klass=read_target_node(source, target.target), bootstrap=func)
def generate_view_class(self, source, target):
targetklass = read_target_node(source, target.target)
try:
module = targetklass.parent
except AttributeError:
# if there is no parent, this handler is not responsible
return
tdir = module.parent
try:
tok = token(str(source.uuid), False)
has_view_methods = tok.has_view_methods
except:
has_view_methods = False
view_class = source.stereotype('pyramid:view_class')
# create init method
if view_class or has_view_methods:
inits = targetklass.functions('__init__')
if inits:
init = inits[0]
else:
init = Function('__init__')
targetklass.insertfirst(init)
# import pdb;pdb.set_trace()
init.args = ['context', 'request']
if not init.blocks('self.request'):
init.insertfirst(Block('self.request = request'))
if not init.blocks('self.context'):
init.insertfirst(Block('self.context = context'))
# if its a view_class create the global_template
if view_class:
gtemplate = view_class.taggedvalue('global_template').value
from_gtemplate = view_class.taggedvalue('from_global_template').value
create_template_file(tdir, gtemplate, from_gtemplate)
imps = Imports(module)
imps.set('pyramid.renderers', 'get_renderer')
if not init.blocks('global_template'):
init.insertlast(Block('renderer = get_renderer("%s")' % gtemplate))
init.insertlast(Block('self.global_template = renderer.implementation()'))
init.insertlast(Block('self.macros = self.global_template.macros'))
