imporve py

HapticX · Jan 13, 2024 · 33468fb · 33468fb
1 parent 07577d2
commit 33468fb
Showing 1 changed file with 14 additions and 9 deletions.
diff --git a/src/happyx/bindings/python_types.nim b/src/happyx/bindings/python_types.nim
@@ -71,15 +71,20 @@ type
     wssHandshakeError,
     wssMismatchProtocol,
     wssError
-  WebSocket* = ref object of PyNimObjectExperimental
-    id*: uint64
-    ws*:
-      when enableHttpBeast:
-        websocketx.WebSocket
-      else:
-        websocket.AsyncWebSocket
-    data*: string
-    state*: WebSocketState
+when enableHttpBeast:
+  type
+    WebSocket* = ref object of PyNimObjectExperimental
+      id*: uint64
+      ws*: websocket.AsyncWebSocket
+      data*: string
+      state*: WebSocketState
+else:
+  type
+    WebSocket* = ref object of PyNimObjectExperimental
+      id*: uint64
+      ws*: websocketx.WebSocket
+      data*: string
+      state*: WebSocketState
 
 
 var