|
|
def import_module(callback): |
|
|
""" |
|
|
Handle "magic" Flask extension imports: |
|
|
``flask.ext.foo`` is really ``flask_foo`` or ``flaskext.foo``. |
|
|
""" |
|
|
def wrapper(inference_state, import_names, module_context, *args, **kwargs): |
|
|
if len(import_names) == 3 and import_names[:2] == ('flask', 'ext'): |
|
|
|
|
|
ipath = ('flask_' + import_names[2]), |
|
|
value_set = callback(inference_state, ipath, None, *args, **kwargs) |
|
|
if value_set: |
|
|
return value_set |
|
|
value_set = callback(inference_state, ('flaskext',), None, *args, **kwargs) |
|
|
return callback( |
|
|
inference_state, |
|
|
('flaskext', import_names[2]), |
|
|
next(iter(value_set)), |
|
|
*args, **kwargs |
|
|
) |
|
|
return callback(inference_state, import_names, module_context, *args, **kwargs) |
|
|
return wrapper |
|
|
|
