Use SaveHooks as explicit generic type alias

fcurella · Jun 18, 2024 · d51b94b · d51b94b
1 parent f7414aa
commit d51b94b
Showing 1 changed file with 24 additions and 24 deletions.
diff --git a/django_fakery/faker_factory.py b/django_fakery/faker_factory.py
@@ -85,7 +85,7 @@ def build_one(
         self,
         model: T,
         fields: Opt[FieldMap],
-        pre_save: Opt[SaveHooks],
+        pre_save: Opt[SaveHooks[T]],
         seed: Opt[Seed],
         make_fks: bool,
         iteration: Opt[int],
@@ -234,8 +234,8 @@ def build(
         self,
         model: T,
         fields: Opt[FieldMap],
-        pre_save: Opt[SaveHooks],
-        seed: Opt[SaveHooks],
+        pre_save: Opt[SaveHooks[T]],
+        seed: Opt[SaveHooks[T]],
         quantity: None,
         make_fks: bool,
     ) -> Built:  # pragma: no cover
@@ -258,7 +258,7 @@ def build(
         self,
         model: T,
         fields: Opt[FieldMap],
-        pre_save: Opt[SaveHooks],
+        pre_save: Opt[SaveHooks[T]],
         seed: Opt[Seed],
         quantity: Opt[int],
         make_fks: bool,
@@ -302,8 +302,8 @@ def make_one(
         self,
         model: T,
         fields: Opt[FieldMap],
-        pre_save: Opt[SaveHooks],
-        post_save: Opt[SaveHooks],
+        pre_save: Opt[SaveHooks[T]],
+        post_save: Opt[SaveHooks[T]],
         seed: Opt[Seed],
         iteration: Opt[int],
     ) -> T:  # pragma: no cover
@@ -358,8 +358,8 @@ def get_or_make(
         model: T,
         lookup: Opt[Lookup],
         fields: Opt[FieldMap],
-        pre_save: Opt[SaveHooks],
-        post_save: Opt[SaveHooks],
+        pre_save: Opt[SaveHooks[T]],
+        post_save: Opt[SaveHooks[T]],
         seed: Opt[Seed],
     ) -> Tuple[T, bool]:  # pragma: no cover
         pass
@@ -406,8 +406,8 @@ def g_m(
         self,
         model: T,
         lookup: Opt[Lookup],
-        pre_save: Opt[SaveHooks],
-        post_save: Opt[SaveHooks],
+        pre_save: Opt[SaveHooks[T]],
+        post_save: Opt[SaveHooks[T]],
         seed: Opt[Seed],
     ) -> Callable[..., T]:  # pragma: no cover
         pass
@@ -445,8 +445,8 @@ def update_or_make(
         model: T,
         lookup: Opt[Lookup],
         fields: Opt[FieldMap],
-        pre_save: Opt[SaveHooks],
-        post_save: Opt[SaveHooks],
+        pre_save: Opt[SaveHooks[T]],
+        post_save: Opt[SaveHooks[T]],
         seed: Opt[Seed],
     ) -> Tuple[T, bool]:  # pragma: no cover
         pass
@@ -501,8 +501,8 @@ def u_m(
         self,
         model: T,
         lookup: Opt[Lookup],
-        pre_save: Opt[SaveHooks],
-        post_save: Opt[SaveHooks],
+        pre_save: Opt[SaveHooks[T]],
+        post_save: Opt[SaveHooks[T]],
         seed: Opt[Seed],
     ) -> Callable[..., T]:  # pragma: no cover
         pass
@@ -539,8 +539,8 @@ def make(
         self,
         model: T,
         fields: Opt[FieldMap],
-        pre_save: Opt[SaveHooks],
-        post_save: Opt[SaveHooks],
+        pre_save: Opt[SaveHooks[T]],
+        post_save: Opt[SaveHooks[T]],
         seed: Opt[Seed],
         quantity: None,
     ) -> T:  # pragma: no cover
@@ -563,8 +563,8 @@ def make(
         self,
         model: T,
         fields: Opt[FieldMap],
-        pre_save: Opt[SaveHooks],
-        post_save: Opt[SaveHooks],
+        pre_save: Opt[SaveHooks[T]],
+        post_save: Opt[SaveHooks[T]],
         seed: Opt[Seed],
         quantity: Opt[int],
     ) -> List[T]:  # pragma: no cover
@@ -604,8 +604,8 @@ def m(
     def m(
         self,
         model: T,
-        pre_save: Opt[SaveHooks],
-        post_save: Opt[SaveHooks],
+        pre_save: Opt[SaveHooks[T]],
+        post_save: Opt[SaveHooks[T]],
         seed: Opt[Seed],
         quantity: None,
     ) -> Callable[..., T]:  # pragma: no cover
@@ -626,8 +626,8 @@ def m(
     def m(
         self,
         model: T,
-        pre_save: Opt[SaveHooks],
-        post_save: Opt[SaveHooks],
+        pre_save: Opt[SaveHooks[T]],
+        post_save: Opt[SaveHooks[T]],
         seed: Opt[Seed],
         quantity: Opt[int],
     ) -> Callable[..., List[T]]:  # pragma: no cover
@@ -663,7 +663,7 @@ def b(
     def b(
         self,
         model: T,
-        pre_save: Opt[SaveHooks],
+        pre_save: Opt[SaveHooks[T]],
         seed: Opt[Seed],
         quantity: None,
         make_fks: bool,
@@ -685,7 +685,7 @@ def b(
     def b(
         self,
         model: T,
-        pre_save: Opt[SaveHooks],
+        pre_save: Opt[SaveHooks[T]],
         seed: Opt[Seed],
         quantity: Opt[int],
         make_fks: bool,