b8b3f8910e
* add more type annotations through the code * add typing in reflex/utils * misc typing * more typings * state typing * keep typing * typing init and utils * more typing for components * fix attempt for 3.9 * need more __future * more typings * type event plz * type model * type vars/base.py * enable 'ANN001' for reflex folder (ignore tests and benchmarks) * fix pyi * add missing annotations * use more precise error when ignoring
41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
"""Module to implement lazy loading in reflex."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import copy
|
|
|
|
import lazy_loader as lazy
|
|
|
|
|
|
def attach(
|
|
package_name: str,
|
|
submodules: set | None = None,
|
|
submod_attrs: dict | None = None,
|
|
):
|
|
"""Replaces a package's __getattr__, __dir__, and __all__ attributes using lazy.attach.
|
|
The lazy loader __getattr__ doesn't support tuples as list values. We needed to add
|
|
this functionality (tuples) in Reflex to support 'import as _' statements. This function
|
|
reformats the submod_attrs dictionary to flatten the module list before passing it to
|
|
lazy_loader.
|
|
|
|
Args:
|
|
package_name: name of the package.
|
|
submodules : List of submodules to attach.
|
|
submod_attrs : Dictionary of submodule -> list of attributes / functions.
|
|
These attributes are imported as they are used.
|
|
|
|
Returns:
|
|
__getattr__, __dir__, __all__
|
|
"""
|
|
_submod_attrs = copy.deepcopy(submod_attrs)
|
|
if _submod_attrs:
|
|
for k, v in _submod_attrs.items():
|
|
# when flattening the list, only keep the alias in the tuple(mod[1])
|
|
_submod_attrs[k] = [
|
|
mod if not isinstance(mod, tuple) else mod[1] for mod in v
|
|
]
|
|
|
|
return lazy.attach(
|
|
package_name=package_name, submodules=submodules, submod_attrs=_submod_attrs
|
|
)
|