|
22 | 22 | from csp.impl.wiring.delayed_node import DelayedNodeWrapperDef |
23 | 23 |
|
24 | 24 | try: |
25 | | - _HAS_WS_ADAPTER = True |
26 | 25 | from csp.lib import _websocketadapterimpl |
27 | 26 | except ImportError: |
28 | 27 | raise ImportError( |
@@ -406,9 +405,6 @@ def __init__( |
406 | 405 | headers: Dict[str, str] = None |
407 | 406 | headers to apply to the request during the handshake |
408 | 407 | """ |
409 | | - if not _HAS_WS_ADAPTER: |
410 | | - raise ModuleNotFoundError("`csp` build does not include websocket C++ adapter") |
411 | | - |
412 | 408 | assert reconnect_interval >= timedelta(seconds=1) |
413 | 409 | self._properties = dict( |
414 | 410 | uri=uri, |
@@ -455,30 +451,25 @@ def _create(self, engine, memo): |
455 | 451 | return _websocketadapterimpl._websocket_adapter_manager(engine, self._properties) |
456 | 452 |
|
457 | 453 |
|
458 | | -if _HAS_WS_ADAPTER: |
459 | | - _websocket_input_adapter_def = input_adapter_def( |
460 | | - "websocket_input_adapter", |
461 | | - _websocketadapterimpl._websocket_input_adapter, |
462 | | - ts["T"], |
463 | | - WebsocketAdapterManager, |
464 | | - typ="T", |
465 | | - properties=dict, |
466 | | - ) |
| 454 | +_websocket_input_adapter_def = input_adapter_def( |
| 455 | + "websocket_input_adapter", |
| 456 | + _websocketadapterimpl._websocket_input_adapter, |
| 457 | + ts["T"], |
| 458 | + WebsocketAdapterManager, |
| 459 | + typ="T", |
| 460 | + properties=dict, |
| 461 | +) |
467 | 462 |
|
468 | | - _websocket_output_adapter_def = output_adapter_def( |
469 | | - "websocket_output_adapter", |
470 | | - _websocketadapterimpl._websocket_output_adapter, |
471 | | - WebsocketAdapterManager, |
472 | | - input=ts["T"], |
473 | | - ) |
| 463 | +_websocket_output_adapter_def = output_adapter_def( |
| 464 | + "websocket_output_adapter", |
| 465 | + _websocketadapterimpl._websocket_output_adapter, |
| 466 | + WebsocketAdapterManager, |
| 467 | + input=ts["T"], |
| 468 | +) |
474 | 469 |
|
475 | | - _websocket_header_update_adapter_def = output_adapter_def( |
476 | | - "websocket_header_update_adapter", |
477 | | - _websocketadapterimpl._websocket_header_update_adapter, |
478 | | - WebsocketAdapterManager, |
479 | | - input=ts[List[WebsocketHeaderUpdate]], |
480 | | - ) |
481 | | -else: |
482 | | - _websocket_input_adapter_def = None |
483 | | - _websocket_output_adapter_def = None |
484 | | - _websocket_header_update_adapter_def = None |
| 470 | +_websocket_header_update_adapter_def = output_adapter_def( |
| 471 | + "websocket_header_update_adapter", |
| 472 | + _websocketadapterimpl._websocket_header_update_adapter, |
| 473 | + WebsocketAdapterManager, |
| 474 | + input=ts[List[WebsocketHeaderUpdate]], |
| 475 | +) |
0 commit comments