From 8d1e31ac5f4e1b83c727ae477c11e60420e2e5d3 Mon Sep 17 00:00:00 2001 From: Eden Kirin Date: Thu, 23 Mar 2023 00:21:20 +0100 Subject: [PATCH] Prettify output --- src/funnel/main.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/src/funnel/main.py b/src/funnel/main.py index 1b395d3..6be79a2 100644 --- a/src/funnel/main.py +++ b/src/funnel/main.py @@ -28,12 +28,9 @@ TIMEZONE = "Europe/Zagreb" @asynccontextmanager async def get_serve_segments_stub() -> AsyncGenerator[ServeSegmentsStub, None]: """Connect to segments server and create stub""" - async with grpc.aio.insecure_channel( - f"{SERVE_SEGMENTS_HOST}:{SERVE_SEGMENTS_PORT}" - ) as channel: - print( - f"🖥 Connected to ServeSegments server on {SERVE_SEGMENTS_HOST}:{SERVE_SEGMENTS_PORT}" - ) + serve_segments_addr = f"{SERVE_SEGMENTS_HOST}:{SERVE_SEGMENTS_PORT}" + async with grpc.aio.insecure_channel(serve_segments_addr) as channel: + print(f"🖥 Connected to ServeSegments server on {serve_segments_addr}") yield ServeSegmentsStub(channel)