fix(signarl): wrong msgpack encode

app/models/signalr.py

@@ -1,10 +1,12 @@
 from __future__ import annotations
 
 import datetime
-from typing import Any, get_origin
+from enum import Enum
+from typing import Any
 
 from pydantic import (
     BaseModel,
+    BeforeValidator,
     ConfigDict,
     Field,
     TypeAdapter,
@@ -17,22 +19,56 @@ def serialize_to_list(value: BaseModel) -> list[Any]:
     data = []
     for field, info in value.__class__.model_fields.items():
         v = getattr(value, field)
-        anno = get_origin(info.annotation)
-        if anno and issubclass(anno, BaseModel):
+        typ = v.__class__
+        if issubclass(typ, BaseModel):
             data.append(serialize_to_list(v))
-        elif anno and issubclass(anno, list):
+        elif issubclass(typ, list):
             data.append(
                 TypeAdapter(
                     info.annotation, config=ConfigDict(arbitrary_types_allowed=True)
                 ).dump_python(v)
             )
-        elif isinstance(v, datetime.datetime):
+        elif issubclass(typ, datetime.datetime):
             data.append([v, 0])
+        elif issubclass(typ, Enum):
+            list_ = list(typ)
+            data.append(list_.index(v) if v in list_ else v.value)
         else:
             data.append(v)
     return data
 
 
+def _by_index(v: Any, class_: type[Enum]):
+    enum_list = list(class_)
+    if not isinstance(v, int):
+        return v
+    if 0 <= v < len(enum_list):
+        return enum_list[v]
+    raise ValueError(
+        f"Value {v} is out of range for enum "
+        f"{class_.__name__} with {len(enum_list)} items"
+    )
+
+
+def EnumByIndex(enum_class: type[Enum]) -> BeforeValidator:
+    return BeforeValidator(lambda v: _by_index(v, enum_class))
+
+
+def msgpack_union(v):
+    data = v[1]
+    data.append(v[0])
+    return data
+
+
+def msgpack_union_dump(v: BaseModel) -> list[Any]:
+    _type = getattr(v, "type", None)
+    if _type is None:
+        raise ValueError(
+            f"Model {v.__class__.__name__} does not have a '_type' attribute"
+        )
+    return [_type, serialize_to_list(v)]
+
+
 class MessagePackArrayModel(BaseModel):
     model_config = ConfigDict(arbitrary_types_allowed=True)