precommit

Author: adhami3310

reflex/istate/proxy.py

@@ -198,7 +198,7 @@ def __getattr__(self, name: str) -> Any:
             )
             raise ImmutableStateError(msg)
 
-        value = super().__getattr__(name)
+        value = super().__getattr__(name)  # pyright: ignore[reportAttributeAccessIssue]
         if not name.startswith("_self_") and isinstance(value, MutableProxy):
             # ensure mutations to these containers are blocked unless proxy is _mutable
             return ImmutableMutableProxy(
@@ -397,7 +397,7 @@ class MutableProxy(wrapt.ObjectProxy):
     }
 
     # Dynamically generated classes for tracking dataclass mutations.
-    __dataclass_proxies__: dict[type, type] = {}
+    __dataclass_proxies__: dict[str, type] = {}
 
     def __new__(cls, wrapped: Any, *args, **kwargs) -> MutableProxy:
         """Create a proxy instance for a mutable object that tracks changes.
@@ -427,7 +427,7 @@ def __new__(cls, wrapped: Any, *args, **kwargs) -> MutableProxy:
                     },
                 )
             cls = cls.__dataclass_proxies__[wrapper_cls_name]
-        return super().__new__(cls)
+        return super().__new__(cls)  # pyright: ignore[reportArgumentType]
 
     def __init__(self, wrapped: Any, state: BaseState, field_name: str):
         """Create a proxy for a mutable object that tracks changes.
@@ -543,7 +543,7 @@ def __getattr__(self, __name: str) -> Any:
         Returns:
             The attribute value.
         """
-        value = super().__getattr__(__name)
+        value = super().__getattr__(__name)  # pyright: ignore[reportAttributeAccessIssue]
 
         if callable(value):
             if __name in self.__mark_dirty_attrs__:
@@ -554,7 +554,7 @@ def __getattr__(self, __name: str) -> Any:
                 # Wrap methods that may return mutable objects tied to the state.
                 value = wrapt.FunctionWrapper(
                     value,
-                    self._wrap_recursive_decorator,
+                    self._wrap_recursive_decorator,  # pyright: ignore[reportArgumentType]
                 )
 
             if (
@@ -564,8 +564,8 @@ def __getattr__(self, __name: str) -> Any:
             ):
                 # Wrap methods called on Base subclasses, which might do _anything_
                 return wrapt.FunctionWrapper(
-                    functools.partial(value.__func__, self),  # pyright: ignore [reportFunctionMemberAccess]
-                    self._wrap_recursive_decorator,
+                    functools.partial(value.__func__, self),  # pyright: ignore [reportFunctionMemberAccess, reportAttributeAccessIssue]
+                    self._wrap_recursive_decorator,  # pyright: ignore[reportArgumentType]
                 )
 
         if is_mutable_type(type(value)) and __name not in (
@@ -587,7 +587,7 @@ def __getitem__(self, key: Any) -> Any:
         Returns:
             The item value.
         """
-        value = super().__getitem__(key)
+        value = super().__getitem__(key)  # pyright: ignore[reportAttributeAccessIssue]
         if isinstance(key, slice) and isinstance(value, list):
             return [self._wrap_recursive(item) for item in value]
         # Recursively wrap mutable items retrieved through this proxy.
@@ -599,7 +599,7 @@ def __iter__(self) -> Any:
         Yields:
             Each item value (possibly wrapped in MutableProxy).
         """
-        for value in super().__iter__():
+        for value in super().__iter__():  # pyright: ignore[reportAttributeAccessIssue]
             # Recursively wrap mutable items retrieved through this proxy.
             yield self._wrap_recursive(value)
 
@@ -617,7 +617,7 @@ def __delitem__(self, key: str):
         Args:
             key: The key of the item.
         """
-        self._mark_dirty(super().__delitem__, args=(key,))
+        self._mark_dirty(super().__delitem__, args=(key,))  # pyright: ignore[reportAttributeAccessIssue]
 
     def __setitem__(self, key: str, value: Any):
         """Set the item on the proxied object and mark state dirty.
@@ -626,7 +626,7 @@ def __setitem__(self, key: str, value: Any):
             key: The key of the item.
             value: The value of the item.
         """
-        self._mark_dirty(super().__setitem__, args=(key, value))
+        self._mark_dirty(super().__setitem__, args=(key, value))  # pyright: ignore[reportAttributeAccessIssue]
 
     def __setattr__(self, name: str, value: Any):
         """Set the attribute on the proxied object and mark state dirty.

tests/units/test_state.py

@@ -2432,7 +2432,7 @@ def assert_array_dirty():
     assert isinstance(mutable_state.array[0], MutableProxy)
     for item in mutable_state.array:
         assert isinstance(item, MutableProxy)
-        item["foo"] = "bar"
+        item["foo"] = "bar"  # pyright: ignore[reportArgumentType, reportCallIssue]
         assert_array_dirty()
 
 
@@ -2507,7 +2507,7 @@ def assert_hashmap_dirty():
     mutable_value_third_ref.append("baz")  # pyright: ignore[reportAttributeAccessIssue]
     assert not mutable_state.dirty_vars
     # Unfortunately previous refs still will mark the state dirty... nothing doing about that
-    assert mutable_value.pop()
+    assert mutable_value.pop()  # pyright: ignore[reportCallIssue]
     assert_hashmap_dirty()