opensource/manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-03 04:04:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update the types in decorator methods using @wraps

Browse Source 
This is method to address issues flagged by pyright
This commit is contained in:
Grant Sanderson
2024-02-03 18:00:47 -06:00
parent cde709fcfa
commit 711438f625
1 changed files with 14 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
manimlib/mobject/mobject.py
View File

@ -48,11 +48,12 @@ from manimlib.utils.space_ops import rotation_matrix_transpose
from typing import TYPE_CHECKING
if TYPE_CHECKING:
from typing import Callable, Iterable, Iterator, Union, Tuple, Optional
from typing import Callable, Iterable, Iterator, Union, Tuple, Optional, TypeVar, Any
import numpy.typing as npt
from manimlib.typing import ManimColor, Vect3, Vect4, Vect3Array, UniformDict, Self
from moderngl.context import Context
T = TypeVar('T')
TimeBasedUpdater = Callable[["Mobject", float], "Mobject" | None]
NonTimeUpdater = Callable[["Mobject"], "Mobject" | None]
Updater = Union[TimeBasedUpdater, NonTimeUpdater]
@ -164,20 +165,23 @@ class Mobject(object):
mob.note_changed_data()
return self
def affects_data(func: Callable):
@staticmethod
def affects_data(func: Callable[..., T]) -> Callable[..., T]:
@wraps(func)
def wrapper(self, *args, **kwargs):
func(self, *args, **kwargs)
result = func(self, *args, **kwargs)
self.note_changed_data()
return result
return wrapper
def affects_family_data(func: Callable):
@staticmethod
def affects_family_data(func: Callable[..., T]) -> Callable[..., T]:
@wraps(func)
def wrapper(self, *args, **kwargs):
func(self, *args, **kwargs)
result = func(self, *args, **kwargs)
for mob in self.family_members_with_points():
mob.note_changed_data()
return self
return result
return wrapper
# Only these methods should directly affect points
@ -578,7 +582,8 @@ class Mobject(object):
# Copying and serialization
def stash_mobject_pointers(func: Callable):
@staticmethod
def stash_mobject_pointers(func: Callable[..., T]) -> Callable[..., T]:
@wraps(func)
def wrapper(self, *args, **kwargs):
uncopied_attrs = ["parents", "target", "saved_state"]
@ -1863,7 +1868,8 @@ class Mobject(object):
# Operations touching shader uniforms
def affects_shader_info_id(func: Callable):
@staticmethod
def affects_shader_info_id(func: Callable[..., T]) -> Callable[..., T]:
@wraps(func)
def wrapper(self, *args, **kwargs):
result = func(self, *args, **kwargs)