defimport_module(callback):""" Handle "magic" Flask extension imports: ``flask.ext.foo`` is really ``flask_foo`` or ``flaskext.foo``. """defwrapper(inference_state,import_names,module_context,*args,**kwargs):iflen(import_names)==3andimport_names[:2]==('flask','ext'):# New style.ipath=(u'flask_'+import_names[2]),value_set=callback(inference_state,ipath,None,*args,**kwargs)ifvalue_set:returnvalue_setvalue_set=callback(inference_state,(u'flaskext',),None,*args,**kwargs)returncallback(inference_state,(u'flaskext',import_names[2]),next(iter(value_set)),*args,**kwargs)returncallback(inference_state,import_names,module_context,*args,**kwargs)returnwrapper