asdict is weird

Author: adhami3310

reflex/components/component.py

@@ -516,7 +516,9 @@ def _deterministic_hash(value: object) -> str:
             str((value._js_expr, _deterministic_hash(value._get_all_var_data())))
         )
     if isinstance(value, VarData):
-        return _hash_dict(dataclasses.asdict(value))
+        return _hash_dict({
+            k.name: getattr(value, k.name) for k in dataclasses.fields(value)
+        })
     if isinstance(value, BaseComponent):
         # If the value is a component, hash its rendered code.
         return _hash_dict(value.render())

reflex/istate/data.py

@@ -185,7 +185,7 @@ def from_router_data(cls, router_data: dict) -> "PageData":
 
 @serializer(to=dict)
 def _serialize_page_data(obj: PageData) -> dict:
-    return dataclasses.asdict(obj)
+    return {key.name: getattr(obj, key.name) for key in dataclasses.fields(obj)}
 
 
 @dataclasses.dataclass(frozen=True)
@@ -215,7 +215,7 @@ def from_router_data(cls, router_data: dict) -> "SessionData":
 
 @serializer(to=dict)
 def _serialize_session_data(obj: SessionData) -> dict:
-    return dataclasses.asdict(obj)
+    return {key.name: getattr(obj, key.name) for key in dataclasses.fields(obj)}
 
 
 @dataclasses.dataclass(frozen=True)

reflex/vars/base.py

@@ -1508,8 +1508,8 @@ def _create_literal_var(
         if dataclasses.is_dataclass(value) and not isinstance(value, type):
             return LiteralObjectVar.create(
                 {
-                    k: (None if callable(v) else v)
-                    for k, v in dataclasses.asdict(value).items()
+                    k.name: (None if callable(v := getattr(value, k.name)) else v)
+                    for k in dataclasses.fields(value)
                 },
                 _var_type=type(value),
                 _var_data=_var_data,
@@ -1591,8 +1591,8 @@ def _get_all_var_data_without_creating_var_dispatch(
 
         if dataclasses.is_dataclass(value) and not isinstance(value, type):
             return LiteralObjectVar._get_all_var_data_without_creating_var({
-                k: (None if callable(v) else v)
-                for k, v in dataclasses.asdict(value).items()
+                k.name: (None if callable(v := getattr(value, k.name)) else v)
+                for k in dataclasses.fields(value)
             })
 
         if isinstance(value, range):

tests/units/test_state.py

@@ -28,6 +28,7 @@
 from reflex.constants.state import FIELD_MARKER
 from reflex.environment import environment
 from reflex.event import Event, EventHandler
+from reflex.istate.data import HeaderData, _FrozenDictStrStr
 from reflex.istate.manager import StateManager
 from reflex.istate.manager.disk import StateManagerDisk
 from reflex.istate.manager.memory import StateManagerMemory
@@ -925,7 +926,11 @@ def test_get_sid(test_state, router_data):
     assert test_state.router.session.session_id == "9fpxSzPb9aFMb4wFAAAH"
 
 
-def test_get_headers(test_state, router_data, router_data_headers):
+def test_get_headers(
+    test_state: TestState,
+    router_data: dict[str, str | dict],
+    router_data_headers: dict[str, str],
+):
     """Test getting client headers.
 
     Args:
@@ -936,13 +941,10 @@ def test_get_headers(test_state, router_data, router_data_headers):
     print(router_data_headers)
     test_state.router = RouterData.from_router_data(router_data)
     print(test_state.router.headers)
-    assert dataclasses.asdict(test_state.router.headers) == {
-        format.to_snake_case(k): v for k, v in router_data_headers.items()
-    } | {
-        "raw_headers": {
-            "_data": tuple(sorted((k, v) for k, v in router_data_headers.items()))
-        }
-    }
+    assert test_state.router.headers == HeaderData(
+        **{format.to_snake_case(k): v for k, v in router_data_headers.items()},
+        raw_headers=_FrozenDictStrStr(**router_data_headers),
+    )
 
 
 def test_get_client_ip(test_state, router_data):