Skip to content

fix(examples): lint fixes in demo_runner.py #2

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
explosivebit merged 1 commit into from
Apr 1, 2026
Merged

fix(examples): lint fixes in demo_runner.py #2

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
33 changes: 19 additions & 14 deletions examples/demo_runner.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@
from __future__ import annotations

import asyncio
import contextlib
import signal
import sys
import time
Expand Down Expand Up @@ -72,7 +73,8 @@ def __init__(self, node_id: str = "mock-node", heartbeat_interval: float = 0):
self._heartbeat_task = None

# Initialize logger factory for Moleculer-style logging
from moleculerpy_repl.logger import LoggerFactory, LogLevel, LogFormat
from moleculerpy_repl.logger import LogFormat, LoggerFactory, LogLevel

self.logger_factory = LoggerFactory(
node_id=node_id,
level=LogLevel.INFO,
Expand Down Expand Up @@ -100,6 +102,7 @@ async def start(self) -> None:
async def _heartbeat_loop(self) -> None:
"""Background heartbeat loop."""
import psutil

while True:
try:
await asyncio.sleep(self.heartbeat_interval)
Expand All @@ -115,10 +118,8 @@ async def stop(self) -> None:
# Stop heartbeat task
if self._heartbeat_task:
self._heartbeat_task.cancel()
try:
with contextlib.suppress(asyncio.CancelledError):
await self._heartbeat_task
except asyncio.CancelledError:
pass
self._discovery_log.debug("Heartbeat stopped")

self.logger.info("ServiceBroker is stopping...")
Expand Down Expand Up @@ -158,7 +159,7 @@ async def broadcast(self, event: str, params: dict = None, **kwargs) -> None:

async def run_single_worker_demo(node_id: str, with_repl: bool = True, heartbeat: float = 0):
"""Run a single worker demo with mock broker."""
from moleculerpy_repl.repl import REPL as MoleculerPyREPL
from moleculerpy_repl.repl import REPL

print("=" * 60)
print("MoleculerPy Runner Demo (Single Worker Mode)")
Expand All @@ -176,7 +177,7 @@ async def run_single_worker_demo(node_id: str, with_repl: bool = True, heartbeat
# Start REPL
# Note: REPL.run() calls broker.stop() in finally block,
# so we don't need to stop it here
repl = MoleculerPyREPL(
repl = REPL(
broker=broker,
delimiter=f"{node_id} $ ",
)
Expand Down Expand Up @@ -220,14 +221,18 @@ def main():
For testing REPL and commands, use single worker mode.
""",
)
parser.add_argument("-i", "--instances", type=int, default=1,
help="Number of worker instances (requires MoleculerPy for >1)")
parser.add_argument("-r", "--repl", action="store_true",
help="Start REPL")
parser.add_argument("-n", "--node-id", type=str, default="demo-node",
help="Node ID prefix")
parser.add_argument("--heartbeat", type=float, default=0,
help="Heartbeat interval in seconds (0 = disabled)")
parser.add_argument(
"-i",
"--instances",
type=int,
default=1,
help="Number of worker instances (requires MoleculerPy for >1)",
)
parser.add_argument("-r", "--repl", action="store_true", help="Start REPL")
parser.add_argument("-n", "--node-id", type=str, default="demo-node", help="Node ID prefix")
parser.add_argument(
"--heartbeat", type=float, default=0, help="Heartbeat interval in seconds (0 = disabled)"
)

args = parser.parse_args()

Expand Down
Loading