| 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 |
|
|
