laipeng/reflex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

cleanup code and add typing

Browse Source
This commit is contained in:
Lendemor 2024-12-18 16:50:17 +01:00
parent 24ac8a519c
commit 565a70b361
3 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
reflex/.templates/jinja/web/pages/custom_component.js.jinja2
View File

@ -8,7 +8,7 @@
{% endfor %}
export const {{component.name}} = memo(({ {{-component.props|join(", ")-}} }) => {
{{ renderHooks(component.hooks.items()) }}
{{ renderHooks(component.hooks) }}
return(
{{utils.render(component.render)}}

2
reflex/.templates/jinja/web/pages/stateful_component.js.jinja2
View File

@ -1,6 +1,6 @@
{% import 'web/pages/utils.js.jinja2' as utils %}
{% from 'web/pages/macros.js.jinja2' import renderHooksWithMemo %}
{% set all_hooks = component._get_all_hooks().items() %}
{% set all_hooks = component._get_all_hooks() %}
export function {{tag_name}} () {
{{ renderHooksWithMemo(all_hooks, memo_trigger_hooks) }}

5
reflex/compiler/templates.py
View File

@ -5,9 +5,10 @@ from jinja2 import Environment, FileSystemLoader, Template
from reflex import constants
from reflex.constants import Hooks
from reflex.utils.format import format_state_name, json_dumps
from reflex.vars.base import VarData
def _sort_hooks(hooks):
def _sort_hooks(hooks: dict[str, VarData | None]):
"""Sort the hooks by their position.
Args:
@ -22,7 +23,7 @@ def _sort_hooks(hooks):
Hooks.HookPosition.POST_TRIGGER: [],
}
for hook, data in hooks:
for hook, data in hooks.items():
if data and data.position and data.position == Hooks.HookPosition.INTERNAL:
sorted_hooks[Hooks.HookPosition.INTERNAL].append((hook, data))
elif not data or (