Break out transport protocol and tcp specifics into their own submodules under tractor.ipc

tractor/ipc/__init__.py

@@ -21,12 +21,12 @@ from ._transport import MsgTransport as MsgTransport
 
 from ._tcp import (
     get_stream_addrs as get_stream_addrs,
-    get_msg_transport as get_msg_transport,
     MsgpackTCPStream as MsgpackTCPStream
 )
 
 from ._chan import (
     _connect_chan as _connect_chan,
+    get_msg_transport as get_msg_transport,
     Channel as Channel
 )
 

tractor/ipc/_chan.py

@@ -29,14 +29,15 @@ from pprint import pformat
 import typing
 from typing import (
     Any,
+    Type
 )
 
 import trio
 
-from tractor.ipc import (
-    MsgTransport,
-    get_stream_addrs,
-    get_msg_transport
+from tractor.ipc._transport import MsgTransport
+from tractor.ipc._tcp import (
+    MsgpackTCPStream,
+    get_stream_addrs
 )
 from tractor.log import get_logger
 from tractor._exceptions import (
@@ -51,6 +52,17 @@ log = get_logger(__name__)
 _is_windows = platform.system() == 'Windows'
 
 
+def get_msg_transport(
+
+    key: tuple[str, str],
+
+) -> Type[MsgTransport]:
+
+    return {
+        ('msgpack', 'tcp'): MsgpackTCPStream,
+    }[key]
+
+
 class Channel:
     '''
     An inter-process channel for communication between (remote) actors.

tractor/ipc/_tcp.py

@@ -15,9 +15,8 @@
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
 '''
 TCP implementation of tractor.ipc._transport.MsgTransport protocol 
+
 '''
-
-
 from __future__ import annotations
 from collections.abc import (
     AsyncGenerator,
@@ -405,14 +404,3 @@ class MsgpackTCPStream(MsgTransport):
 
     def connected(self) -> bool:
         return self.stream.socket.fileno() != -1
-
-
-def get_msg_transport(
-
-    key: tuple[str, str],
-
-) -> Type[MsgTransport]:
-
-    return {
-        ('msgpack', 'tcp'): MsgpackTCPStream,
-    }[key]

tractor/ipc/_transport.py

@@ -16,8 +16,8 @@
 '''
 typing.Protocol based generic msg API, implement this class to add backends for
 tractor.ipc.Channel
-'''
 
+'''
 import trio
 from typing import (
     runtime_checkable,