L3-Bataille-Navale/source/gui/scene/abc/Scene.py

57 lines
1.5 KiB
Python

from abc import ABC
from typing import TYPE_CHECKING, Type
from source.gui.event import EventPropagationMixin
if TYPE_CHECKING:
from source.gui.window import Window
from source.gui.widget.abc import Widget
class Scene(ABC, EventPropagationMixin):
"""
A scene that can be attached to a window.
It allows to switch the whole comportment of the window in a simpler way.
It can react to any "on_" event from the window.
"""
def __init__(self, window: "Window", *args, **kwargs):
self.window = window
self._widgets: list["Widget"] = list()
# Event propagation
@property
def childs(self):
return self._widgets
# Widget Managing
def add_widget(self, widget_class: Type["Widget"], **widget_kwargs):
"""
Add a widget to the scene.
:param widget_class: the class of the widget to add.
:param widget_args: args for the creation of the widget object.
:param widget_kwargs: kwargs for the creation of the widget object.
:return: the new created widget.
"""
widget: "Widget" = widget_class(self, **widget_kwargs)
self._widgets.append(widget)
return widget
def remove_widget(self, widget: "Widget") -> None:
"""
Remove a widget from the scene.
:param widget: the widget to remove.
"""
self._widgets.remove(widget)
def clear_widget(self) -> None:
"""
Clear the scene from all the widgets.
"""
self._widgets.clear()