Merge remote-tracking branch 'origin/ISS-678-use-examples' into ISS-678-use-examples

Author: Michel Samia

.pre-commit-config.yaml

@@ -23,15 +23,20 @@ repos:
       - id: ruff
         args: ["--fix"]
       - id: ruff-format
-  - repo: https://github.com/codespell-project/codespell
-    rev: v2.4.1
+  - repo: https://github.com/crate-ci/typos
+    rev: v1.30.3
     hooks:
-      - id: codespell
+      - id: typos
   - repo: https://github.com/pycontribs/mirrors-prettier
     rev: "v3.5.3"
     hooks:
       - id: prettier
         exclude: ".all-contributorsrc"
+  - repo: https://github.com/ComPWA/taplo-pre-commit
+    rev: v0.9.3
+    hooks:
+      - id: taplo-format
+        exclude: "uv.lock"
   - repo: https://github.com/pre-commit/mirrors-mypy
     rev: "v1.15.0"
     hooks:

polyfactory/factories/base.py

@@ -835,7 +835,8 @@ def get_field_value(  # noqa: C901, PLR0911, PLR0912
                 build_context=build_context,
             )
 
-        if provider := cls.get_provider_map().get(unwrapped_annotation):
+        provider_map = cls.get_provider_map()
+        if provider := (provider_map.get(field_meta.annotation) or provider_map.get(unwrapped_annotation)):
             return provider()
 
         if isinstance(unwrapped_annotation, TypeVar):
@@ -914,7 +915,10 @@ def get_field_value_coverage(  # noqa: C901,PLR0912
 
                 yield handle_collection_type_coverage(child_meta, origin, cls, build_context=build_context)
 
-            elif provider := cls.get_provider_map().get(unwrapped_annotation):
+            elif provider := (
+                (provider_map := cls.get_provider_map()).get(field_meta.annotation)
+                or provider_map.get(unwrapped_annotation)
+            ):
                 yield CoverageContainerCallable(provider)
 
             elif isinstance(unwrapped_annotation, TypeVar):

polyfactory/field_meta.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import functools
 from dataclasses import asdict
 from typing import TYPE_CHECKING, Any, Hashable, Literal, Mapping, Pattern, TypedDict, cast
 
@@ -60,7 +61,7 @@ class Constraints(TypedDict):
 class FieldMeta:
     """Factory field metadata container. This class is used to store the data about a field of a factory's model."""
 
-    __slots__ = ("annotation", "children", "constraints", "default", "name", "random")
+    __slots__ = ("__dict__", "annotation", "children", "constraints", "default", "name", "random")
 
     annotation: Any
     random: Random
@@ -88,15 +89,18 @@ def __init__(
         self.name = name
         self.constraints = constraints
 
-    @property
+    @functools.cached_property
     def type_args(self) -> tuple[Any, ...]:
         """Return the normalized type args of the annotation, if any.
 
         :returns: a tuple of types.
         """
+        annotation = self.annotation
+        if is_annotated(annotation):
+            annotation = get_args(self.annotation)[0]
         return tuple(
             TYPE_MAPPING.get(arg, arg) if isinstance(arg, Hashable) else arg  # pyright: ignore[reportCallIssue,reportArgumentType]
-            for arg in get_args(self.annotation)
+            for arg in get_args(annotation)
         )
 
     @classmethod
@@ -140,9 +144,7 @@ def from_type(
             metadata = cls.get_constraints_metadata(annotation)
             constraints = cls.parse_constraints(metadata)
 
-        if annotated:
-            annotation = get_args(annotation)[0]
-        elif (origin := get_origin(annotation)) and origin in TYPE_MAPPING:  # pragma: no cover
+        if not annotated and (origin := get_origin(annotation)) and origin in TYPE_MAPPING:
             container = TYPE_MAPPING[origin]
             annotation = container[get_args(annotation)]  # type: ignore[index]