HH-205308 fix recursion in json encoder #680

Powerrr · 2024-02-01T05:37:53Z

https://jira.hh.ru/browse/HH-205308

Пофиксил контракт default-метода _encode_value: если он не преобразует значение, то кидает ошибку (иначе получается бесконечная рекурсия), и он не преобразует вложенные объекты, сериализатор json сам вызовет его, если нужно.
Но в одном месте оказалось нужно преобразование вместе с вложенными объектами, для этого сделал рекурсивный метод _deep_encode_value.

Powerrr · 2024-02-01T05:45:39Z

frontik/json_builder.py

 FrontikJsonDecodeError = orjson.JSONDecodeError


-def _encode_value(value: Any) -> Any:
+def _deep_encode_value(value: Any) -> Any:


бывший рекурсивный _encode_value переименовал в _deep_encode_value и в паре мест добавил ещё рекурсии (т.к. странно, что некоторые вложенные объекты преобразовывались, а некоторые нет, так что сделал все), см. комменты ниже

Powerrr · 2024-02-01T05:46:35Z

frontik/json_builder.py

+        return None
+
+    elif isinstance(value, BaseModel):
+        return _deep_encode_value(value.model_dump())


тут добавил рекурсивный вызов

Powerrr · 2024-02-01T05:46:47Z

frontik/json_builder.py

+        return _deep_encode_value(value.model_dump())
+
+    elif hasattr(value, 'to_dict'):
+        return _deep_encode_value(value.to_dict())


и тут добавил рекурсивный вызов

Powerrr · 2024-02-01T05:47:17Z

frontik/json_builder.py

+    return value
+
+
+def _encode_value(value: Any) -> Any:


этот метод теперь без рекурсии. также убрал обработку dict, list, tuple, т.к. json-сериализатор сам их умеет

Powerrr · 2024-02-01T05:48:16Z

frontik/json_builder.py

@@ -76,7 +91,7 @@ def replace(self, *args: Any, **kwargs: Any) -> None:

    def to_dict(self) -> dict:
        """Return plain dict from all data appended to JsonBuilder"""
-        return _encode_value(self._concat_chunks())
+        return _deep_encode_value(self._concat_chunks())
