Skip to content

we should always deeply unmount #620

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Jan 27, 2022
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
29 changes: 14 additions & 15 deletions src/idom/core/layout.py
Original file line number Diff line number Diff line change
Expand Up @@ -100,7 +100,7 @@ def __enter__(self: _Self) -> _Self:
def __exit__(self, *exc: Any) -> None:
root_csid = self._root_life_cycle_state_id
root_model_state = self._model_states_by_life_cycle_state_id[root_csid]
self._deep_unmount_model_states([root_model_state])
self._unmount_model_states([root_model_state])

# delete attributes here to avoid access after exiting context manager
del self._event_handlers
Expand Down Expand Up @@ -320,7 +320,7 @@ def _render_model_children(
self._render_model_children_without_old_state(new_state, raw_children)
return None
elif not raw_children:
self._deep_unmount_model_states(list(old_state.children_by_key.values()))
self._unmount_model_states(list(old_state.children_by_key.values()))
return None

child_type_key_tuples = list(_process_child_type_and_key(raw_children))
Expand All @@ -335,7 +335,7 @@ def _render_model_children(

old_keys = set(old_state.children_by_key).difference(new_keys)
if old_keys:
self._deep_unmount_model_states(
self._unmount_model_states(
[old_state.children_by_key[key] for key in old_keys]
)

Expand All @@ -352,7 +352,7 @@ def _render_model_children(
)
else:
if old_child_state.is_component_state:
self._shallow_unmount_model_state(old_child_state)
self._unmount_model_states([old_child_state])
new_child_state = _update_element_model_state(
old_child_state,
new_state,
Expand Down Expand Up @@ -383,7 +383,7 @@ def _render_model_children(
else:
old_child_state = old_state.children_by_key.get(key)
if old_child_state is not None:
self._deep_unmount_model_states([old_child_state])
self._unmount_model_states([old_child_state])
new_children.append(child)

def _render_model_children_without_old_state(
Expand All @@ -406,21 +406,20 @@ def _render_model_children_without_old_state(
else:
new_children.append(child)

def _deep_unmount_model_states(self, old_states: List[_ModelState]) -> None:
def _unmount_model_states(self, old_states: List[_ModelState]) -> None:
to_unmount = old_states[::-1] # unmount in reversed order of rendering
while to_unmount:
model_state = to_unmount.pop()
self._shallow_unmount_model_state(model_state)
to_unmount.extend(model_state.children_by_key.values())

def _shallow_unmount_model_state(self, old_state: _ModelState) -> None:
for target in old_state.targets_by_event.values():
del self._event_handlers[target]
for target in model_state.targets_by_event.values():
del self._event_handlers[target]

if model_state.is_component_state:
life_cycle_state = model_state.life_cycle_state
del self._model_states_by_life_cycle_state_id[life_cycle_state.id]
life_cycle_state.hook.component_will_unmount()

if old_state.is_component_state:
life_cycle_state = old_state.life_cycle_state
del self._model_states_by_life_cycle_state_id[life_cycle_state.id]
life_cycle_state.hook.component_will_unmount()
to_unmount.extend(model_state.children_by_key.values())

def __repr__(self) -> str:
return f"{type(self).__name__}({self.root})"
Expand Down