Merge bfxapi.rest.types and bfxapi.websocket.types in bfxapi.tests sub-package.

bfxapi/types/notification.py

@@ -0,0 +1,38 @@
+from typing import List, Optional, Any, Generic, TypeVar, cast
+from dataclasses import dataclass
+from . labeler import _Type, _Serializer
+
+T = TypeVar("T")
+
+@dataclass
+class Notification(_Type, Generic[T]):
+    mts: int
+    type: str
+    message_id: Optional[int]
+    data: T
+    code: Optional[int]
+    status: str
+    text: str
+
+class _Notification(_Serializer, Generic[T]):
+    __LABELS = [ "mts", "type", "message_id", "_PLACEHOLDER", "data", "code", "status", "text" ]
+
+    def __init__(self, serializer: Optional[_Serializer] = None, is_iterable: bool = False):
+        super().__init__("Notification", Notification, _Notification.__LABELS, ignore = [ "_PLACEHOLDER" ])
+
+        self.serializer, self.is_iterable = serializer, is_iterable
+
+    def parse(self, *values: Any) -> Notification[T]:
+        notification = cast(Notification[T], Notification(**dict(self._serialize(*values))))
+
+        if isinstance(self.serializer, _Serializer):
+            data = cast(List[Any], notification.data)
+
+            if not self.is_iterable:
+                if len(data) == 1 and isinstance(data[0], list):
+                    data = data[0]
+
+                notification.data = self.serializer.parse(*data)
+            else: notification.data = cast(T, [ self.serializer.parse(*sub_data) for sub_data in data ])
+
+        return notification