From 96cf4b6e9fd04809205589c6c294542feaba7127 Mon Sep 17 00:00:00 2001
From: Tyler Goodlet <jgbt@protonmail.com>
Date: Thu, 23 Jul 2020 13:23:36 -0400
Subject: [PATCH] Allow opening a portal through an existing channel

---
 tractor/_portal.py | 24 ++++++++++++++----------
 1 file changed, 14 insertions(+), 10 deletions(-)

diff --git a/tractor/_portal.py b/tractor/_portal.py
index 89cab39..c84a7fe 100644
--- a/tractor/_portal.py
+++ b/tractor/_portal.py
@@ -314,7 +314,8 @@ class LocalPortal:
 @asynccontextmanager
 async def open_portal(
     channel: Channel,
-    nursery: Optional[trio.Nursery] = None
+    nursery: Optional[trio.Nursery] = None,
+    start_msg_loop: bool = True,
 ) -> typing.AsyncGenerator[Portal, None]:
     """Open a ``Portal`` through the provided ``channel``.
 
@@ -332,15 +333,17 @@ async def open_portal(
         if channel.uid is None:
             await actor._do_handshake(channel)
 
-        msg_loop_cs: trio.CancelScope = await nursery.start(
-            partial(
-                actor._process_messages,
-                channel,
-                # if the local task is cancelled we want to keep
-                # the msg loop running until our block ends
-                shield=True,
+        msg_loop_cs = None
+        if start_msg_loop:
+            msg_loop_cs: trio.CancelScope = await nursery.start(
+                partial(
+                    actor._process_messages,
+                    channel,
+                    # if the local task is cancelled we want to keep
+                    # the msg loop running until our block ends
+                    shield=True,
+                )
             )
-        )
         portal = Portal(channel)
         try:
             yield portal
@@ -352,6 +355,7 @@ async def open_portal(
                 await channel.send(None)
 
             # cancel background msg loop task
-            msg_loop_cs.cancel()
+            if msg_loop_cs:
+                msg_loop_cs.cancel()
 
             nursery.cancel_scope.cancel()