_modify_sub_elements reworked for Element

Author: VladimirPodolian

mops/base/element.py

@@ -131,7 +131,7 @@ def __full_init__(self, driver_wrapper: Any = None):
             self.driver_wrapper = get_driver_wrapper_from_object(driver_wrapper)
 
         self._modify_object()
-        self._modify_children()
+        self._modify_sub_elements()
 
         if not self._initialized:
             self.__init_base_class__()
@@ -763,13 +763,16 @@ def _get_all_elements(self, sources: Union[tuple, list]) -> List[Any]:
 
         return wrapped_elements
 
-    def _modify_children(self):
+    def _modify_sub_elements(self):
         """
         Initializing of attributes with  type == Element.
         Required for classes with base == Element.
         """
-        self.sub_elements = get_child_elements_with_names(self, Element)
-        initialize_objects(self, self.sub_elements, Element)
+        self.sub_elements = {}
+
+        if type(self) is not Element:
+            self.sub_elements = get_child_elements_with_names(self, Element)
+            initialize_objects(self, self.sub_elements, Element)
 
     def _modify_object(self):
         """

mops/base/group.py

@@ -72,7 +72,7 @@ def __init__(
             driver_wrapper=driver_wrapper,
         )
 
-    def _modify_children(self) -> None:
+    def _modify_sub_elements(self) -> None:
         """
         Initializing of attributes with type == Group/Element.
         Required for classes with base == Group.

mops/base/page.py

@@ -84,7 +84,7 @@ def __init__(
 
         self._init_locals = locals()
         self._modify_page_driver_wrapper(driver_wrapper)
-        self._modify_children()
+        self._modify_sub_elements()
         self._safe_setter('__base_obj_id', id(self))
 
         self.page_elements: List[Element] = get_child_elements(self, Element)
@@ -192,7 +192,7 @@ def is_page_opened(self, with_elements: bool = False, with_url: bool = False) ->
 
         return result
 
-    def _modify_children(self):
+    def _modify_sub_elements(self):
         """
         Initializing of attributes with type == Element.
         Required for classes with base == Page.