mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-09-21 03:01:27 -03:00
Will Miao
72 lines
1.9 KiB
Python
Executable File
72 lines
1.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
Wait for LoRa Manager server to become ready.
|
|
|
|
Timeout is configurable via --timeout (default 30s); the script polls the port
|
|
until the server accepts connections or the timeout expires.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import socket
|
|
import sys
|
|
import time
|
|
|
|
|
|
def is_server_ready(port: int, timeout: float = 2.0) -> bool:
|
|
"""Check if server is accepting connections."""
|
|
try:
|
|
with socket.create_connection(("127.0.0.1", port), timeout=timeout):
|
|
return True
|
|
except (socket.timeout, ConnectionRefusedError, OSError):
|
|
return False
|
|
|
|
|
|
def wait_for_server(port: int, timeout: int = 30) -> bool:
|
|
"""Wait for server to become ready."""
|
|
start = time.time()
|
|
last_report = 0.0
|
|
while time.time() - start < timeout:
|
|
if is_server_ready(port):
|
|
return True
|
|
# Report progress every ~5s so a slow boot is visible, not silent.
|
|
elapsed = time.time() - start
|
|
if elapsed - last_report >= 5:
|
|
print(f" ...still waiting ({int(elapsed)}s/{timeout}s)")
|
|
last_report = elapsed
|
|
time.sleep(0.5)
|
|
return False
|
|
|
|
|
|
def main() -> int:
|
|
parser = argparse.ArgumentParser(
|
|
description="Wait for LoRa Manager server to become ready"
|
|
)
|
|
parser.add_argument(
|
|
"--port",
|
|
type=int,
|
|
default=8188,
|
|
help="Server port (default: 8188)",
|
|
)
|
|
parser.add_argument(
|
|
"--timeout",
|
|
type=int,
|
|
default=30,
|
|
help="Timeout in seconds (default: 30)",
|
|
)
|
|
|
|
args = parser.parse_args()
|
|
|
|
print(f"Waiting for server on port {args.port} (timeout: {args.timeout}s)...")
|
|
|
|
if wait_for_server(args.port, args.timeout):
|
|
print(f"Server ready at http://127.0.0.1:{args.port}/loras")
|
|
return 0
|
|
print(f"Timeout: Server not ready after {args.timeout}s")
|
|
return 1
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|