pomcorn 0.7.5__tar.gz → 0.8.1__tar.gz

{pomcorn-0.7.5 → pomcorn-0.8.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pomcorn
-Version: 0.7.5
+Version: 0.8.1
 Summary: Base implementation of Page Object Model
 Home-page: https://pypi.org/project/pomcorn/
 License: MIT
@@ -119,7 +119,6 @@ Below is the code that opens ``PyPI.org``, searches for packages by name and pri
 
   class PackageList(ListComponent[Package, PyPIPage]):
 
-      item_class = Package
       relative_item_locator = locators.ClassLocator("snippet__name")
 
       @property

{pomcorn-0.7.5 → pomcorn-0.8.1}/README.md

@@ -93,7 +93,6 @@ Below is the code that opens ``PyPI.org``, searches for packages by name and pri
 
   class PackageList(ListComponent[Package, PyPIPage]):
 
-      item_class = Package
       relative_item_locator = locators.ClassLocator("snippet__name")
 
       @property

{pomcorn-0.7.5 → pomcorn-0.8.1}/pomcorn/component.py

@@ -1,4 +1,13 @@
-from typing import Generic, TypeVar, overload
+from inspect import isclass
+from typing import (
+    Any,
+    Generic,
+    Literal,
+    TypeVar,
+    get_args,
+    get_origin,
+    overload,
+)
 
 from . import locators
 from .element import XPathElement
@@ -8,6 +17,17 @@ from .web_view import WebView
 TPage = TypeVar("TPage", bound=Page)
 
 
+class _EmptyValue:
+    """Singleton to use as default value for empty class attribute."""
+
+    def __bool__(self) -> Literal[False]:
+        """Allow `EmptyValue` to be used in bool expressions."""
+        return False
+
+
+EmptyValue: Any = _EmptyValue()
+
+
 class Component(Generic[TPage], WebView):
     """The class to represent a page component that depends on base locator.
 
@@ -177,21 +197,36 @@ class ListComponent(Generic[ListItemType, TPage], Component[TPage]):
     * all
     * get_item_by_text()
 
-    Waits for `base_item_locator` property to be implemented and value of
-    `item_class` to be set.
-
     Waits for `base_item_locator` property  to be overridden or one of the
     attributes (`item_locator` or `relative_item_locator`) to be specified.
 
-    The `item_class` attribute is also required.
-
     """
 
-    item_class: type[ListItemType]
+    _item_class: type[ListItemType] = EmptyValue
 
     item_locator: locators.XPathLocator | None = None
     relative_item_locator: locators.XPathLocator | None = None
 
+    def __init_subclass__(cls) -> None:
+        """Run logic for getting/overriding item_class attr for subclasses."""
+        super().__init_subclass__()
+
+        # If class has valid `_item_class` attribute from a parent class
+        if cls.is_valid_item_class(cls._item_class):
+            # We leave using of parent `item_class`
+            return
+
+        # Try to get `item_class` from first generic variable
+        list_item_class = cls.get_list_item_class()
+
+        if not list_item_class:
+            # If `item_class` is not specified in generic we leave it empty
+            # because it maybe not specified in base class but will be
+            # specified in child
+            return
+
+        cls._item_class = list_item_class
+
     @property
     def base_item_locator(self) -> locators.XPathLocator:
         """Get the base locator of list item.
@@ -236,21 +271,49 @@ class ListComponent(Generic[ListItemType, TPage], Component[TPage]):
 
         items: list[ListItemType] = []
         for locator in self.iter_locators(self.base_item_locator):
-            items.append(self.item_class(page=self.page, base_locator=locator))
+            items.append(
+                self._item_class(page=self.page, base_locator=locator),
+            )
         return items
 
+    @classmethod
+    def get_list_item_class(cls) -> type[ListItemType] | None:
+        """Return class passed in `Generic[ListItemType]`."""
+        base_class = next(
+            _class
+            for _class in cls.__orig_bases__  # type: ignore
+            if isclass(get_origin(_class))
+            and issubclass(get_origin(_class), ListComponent)
+        )
+
+        # Get first generic variable and return it if it is valid item class
+        item_class = get_args(base_class)[0]
+        if cls.is_valid_item_class(item_class):
+            return item_class
+
+        return None
+
+    @classmethod
+    def is_valid_item_class(cls, item_class: Any) -> bool:
+        """Check that specified ``item_class`` is valid.
+
+        Valid `item_class` should be a class and subclass of ``Component``.
+
+        """
+        return isclass(item_class) and issubclass(item_class, Component)
+
     def get_item_by_text(self, text: str) -> ListItemType:
         """Get list item by text."""
         locator = self.base_item_locator.extend_query(
             extra_query=f"[contains(.,'{text}')]",
         )
-        return self.item_class(page=self.page, base_locator=locator)
+        return self._item_class(page=self.page, base_locator=locator)
 
     def __repr__(self) -> str:
         return (
             "ListComponent("
             f"component={self.__class__}, "
-            f"item_class={self.item_class}, "
+            f"item_class={self._item_class}, "
             f"base_item_locator={self.base_item_locator}, "
             f"count={self.count}, "
             f"items={self.all}, "

{pomcorn-0.7.5 → pomcorn-0.8.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pomcorn"
-version = "0.7.5"
+version = "0.8.1"
 description = "Base implementation of Page Object Model"
 authors = [
   "Saritasa <pypi@saritasa.com>",
@@ -64,7 +64,7 @@ sphinx = ">= 7.2.6"
 sphinx-rtd-theme = ">= 1.3.0rc1"
 # Support mermaid diagrams
 # https://github.com/mgaitan/sphinxcontrib-mermaid/tree/master
-sphinxcontrib-mermaid = "^0.9.2"
+sphinxcontrib-mermaid = ">=0.9.2,<1.1.0"
 
 [tool.poetry.group.demo.dependencies]
 pytest = ">= 7.4.2"
@@ -171,6 +171,11 @@ pytest-parametrize-names-type = "list"
 pytest-parametrize-values-type = "list"
 pytest-parametrize-values-row-type = "list"
 
+per-file-ignores = [
+  # Ignore using lambda in `item_class` tests
+  "tests/list_component/test_item_class.py: E731",
+]
+
 [tool.mypy]
 # mypy configurations: https://mypy.readthedocs.io/en/latest/config_file.html
 # https://mypy.readthedocs.io/en/stable/running_mypy.html#missing-imports