7f1aee6dc8
* enable N rules for naming conventions * fix pyi * address comments * remove unneeded code
92 lines
2.2 KiB
Python
92 lines
2.2 KiB
Python
"""Add standard Hooks wrapper for React."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from reflex.utils.imports import ImportVar
|
|
from reflex.vars import VarData
|
|
from reflex.vars.base import Var
|
|
|
|
|
|
def _compose_react_imports(tags: list[str]) -> dict[str, list[ImportVar]]:
|
|
return {"react": [ImportVar(tag=tag) for tag in tags]}
|
|
|
|
|
|
def const(name, value) -> Var:
|
|
"""Create a constant Var.
|
|
|
|
Args:
|
|
name: The name of the constant.
|
|
value: The value of the constant.
|
|
|
|
Returns:
|
|
The constant Var.
|
|
"""
|
|
if isinstance(name, list):
|
|
return Var(_js_expr=f"const [{', '.join(name)}] = {value}")
|
|
return Var(_js_expr=f"const {name} = {value}")
|
|
|
|
|
|
def useCallback(func, deps) -> Var: # noqa: N802
|
|
"""Create a useCallback hook with a function and dependencies.
|
|
|
|
Args:
|
|
func: The function to wrap.
|
|
deps: The dependencies of the function.
|
|
|
|
Returns:
|
|
The useCallback hook.
|
|
"""
|
|
return Var(
|
|
_js_expr=f"useCallback({func}, {deps})" if deps else f"useCallback({func})",
|
|
_var_data=VarData(imports=_compose_react_imports(["useCallback"])),
|
|
)
|
|
|
|
|
|
def useContext(context) -> Var: # noqa: N802
|
|
"""Create a useContext hook with a context.
|
|
|
|
Args:
|
|
context: The context to use.
|
|
|
|
Returns:
|
|
The useContext hook.
|
|
"""
|
|
return Var(
|
|
_js_expr=f"useContext({context})",
|
|
_var_data=VarData(imports=_compose_react_imports(["useContext"])),
|
|
)
|
|
|
|
|
|
def useRef(default) -> Var: # noqa: N802
|
|
"""Create a useRef hook with a default value.
|
|
|
|
Args:
|
|
default: The default value of the ref.
|
|
|
|
Returns:
|
|
The useRef hook.
|
|
"""
|
|
return Var(
|
|
_js_expr=f"useRef({default})",
|
|
_var_data=VarData(imports=_compose_react_imports(["useRef"])),
|
|
)
|
|
|
|
|
|
def useState(var_name, default=None) -> Var: # noqa: N802
|
|
"""Create a useState hook with a variable name and setter name.
|
|
|
|
Args:
|
|
var_name: The name of the variable.
|
|
default: The default value of the variable.
|
|
|
|
Returns:
|
|
A useState hook.
|
|
"""
|
|
return const(
|
|
[var_name, f"set{var_name.capitalize()}"],
|
|
Var(
|
|
_js_expr=f"useState({default})",
|
|
_var_data=VarData(imports=_compose_react_imports(["useState"])),
|
|
),
|
|
)
|