[config.git] / djavu-asus / elpy / rpc-venv / lib / python3.11 / site-packages / jedi / plugins / flask.py

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'):
            # New style.
            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
Commit	Line	Data
53e6db90 DC	1	def import_module(callback):
	2	"""
	3	Handle "magic" Flask extension imports:
	4	``flask.ext.foo`` is really ``flask_foo`` or ``flaskext.foo``.
	5	"""
	6	def wrapper(inference_state, import_names, module_context, args, *kwargs):
	7	if len(import_names) == 3 and import_names[:2] == ('flask', 'ext'):
	8	# New style.
	9	ipath = ('flask_' + import_names[2]),
	10	value_set = callback(inference_state, ipath, None, args, *kwargs)
	11	if value_set:
	12	return value_set
	13	value_set = callback(inference_state, ('flaskext',), None, args, *kwargs)
	14	return callback(
	15	inference_state,
	16	('flaskext', import_names[2]),
	17	next(iter(value_set)),
	18	args, *kwargs
	19	)
	20	return callback(inference_state, import_names, module_context, args, *kwargs)
	21	return wrapper