1 files changed, 104 insertions, 0 deletions
diff --git a/venv/lib/python3.11/site-packages/polyfactory/pytest_plugin.py b/venv/lib/python3.11/site-packages/polyfactory/pytest_plugin.py
new file mode 100644
index 0000000..0ca9196
--- /dev/null
+++ b/venv/lib/python3.11/site-packages/polyfactory/pytest_plugin.py
@@ -0,0 +1,104 @@
+from __future__ import annotations
+
+import re
+from typing import (
+    Any,
+    Callable,
+    ClassVar,
+    Literal,
+    Union,
+)
+
+from pytest import Config, fixture  # noqa: PT013
+
+from polyfactory.exceptions import ParameterException
+from polyfactory.factories.base import BaseFactory
+from polyfactory.utils.predicates import is_safe_subclass
+
+Scope = Union[
+    Literal["session", "package", "module", "class", "function"],
+    Callable[[str, Config], Literal["session", "package", "module", "class", "function"]],
+]
+
+
+split_pattern_1 = re.compile(r"([A-Z]+)([A-Z][a-z])")
+split_pattern_2 = re.compile(r"([a-z\d])([A-Z])")
+
+
+def _get_fixture_name(name: str) -> str:
+    """From inflection.underscore.
+
+    :param name: str: A name.
+
+    :returns: Normalized fixture name.
+
+    """
+    name = re.sub(split_pattern_1, r"\1_\2", name)
+    name = re.sub(split_pattern_2, r"\1_\2", name)
+    name = name.replace("-", "_")
+    return name.lower()
+
+
+class FactoryFixture:
+    """Decorator that creates a pytest fixture from a factory"""
+
+    __slots__ = ("scope", "autouse", "name")
+
+    factory_class_map: ClassVar[dict[Callable, type[BaseFactory[Any]]]] = {}
+
+    def __init__(
+        self,
+        scope: Scope = "function",
+        autouse: bool = False,
+        name: str | None = None,
+    ) -> None:
+        """Create a factory fixture decorator
+
+        :param scope: Fixture scope
+        :param autouse: Autouse the fixture
+        :param name: Fixture name
+        """
+        self.scope = scope
+        self.autouse = autouse
+        self.name = name
+
+    def __call__(self, factory: type[BaseFactory[Any]]) -> Any:
+        if not is_safe_subclass(factory, BaseFactory):
+            msg = f"{factory.__name__} is not a BaseFactory subclass."
+            raise ParameterException(msg)
+
+        fixture_name = self.name or _get_fixture_name(factory.__name__)
+        fixture_register = fixture(
+            scope=self.scope,  # pyright: ignore[reportGeneralTypeIssues]
+            name=fixture_name,
+            autouse=self.autouse,
+        )
+
+        def _factory_fixture() -> type[BaseFactory[Any]]:
+            """The wrapped factory"""
+            return factory
+
+        _factory_fixture.__doc__ = factory.__doc__
+        marker = fixture_register(_factory_fixture)
+        self.factory_class_map[marker] = factory
+        return marker
+
+
+def register_fixture(
+    factory: type[BaseFactory[Any]] | None = None,
+    *,
+    scope: Scope = "function",
+    autouse: bool = False,
+    name: str | None = None,
+) -> Any:
+    """A decorator that allows registering model factories as fixtures.
+
+    :param factory: An optional factory class to decorate.
+    :param scope: Pytest scope.
+    :param autouse: Auto use fixture.
+    :param name: Fixture name.
+
+    :returns: A fixture factory instance.
+    """
+    factory_fixture = FactoryFixture(scope=scope, autouse=autouse, name=name)
+    return factory_fixture(factory) if factory else factory_fixture