mirror of
https://github.com/Chia-Network/chia-blockchain.git
synced 2024-09-20 16:08:51 +03:00
67 lines
2.2 KiB
Python
67 lines
2.2 KiB
Python
import asyncio
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
|
|
from pathlib import Path
|
|
from typing import Optional
|
|
|
|
from chia.daemon.client import DaemonProxy, connect_to_daemon_and_validate
|
|
from chia.util.service_groups import services_for_groups
|
|
|
|
|
|
def launch_start_daemon(root_path: Path) -> subprocess.Popen:
|
|
os.environ["CHIA_ROOT"] = str(root_path)
|
|
# TODO: use startupinfo=subprocess.DETACHED_PROCESS on windows
|
|
chia = sys.argv[0]
|
|
process = subprocess.Popen(f"{chia} run_daemon".split(), stdout=subprocess.PIPE)
|
|
return process
|
|
|
|
|
|
async def create_start_daemon_connection(root_path: Path) -> Optional[DaemonProxy]:
|
|
connection = await connect_to_daemon_and_validate(root_path)
|
|
if connection is None:
|
|
print("Starting daemon")
|
|
# launch a daemon
|
|
process = launch_start_daemon(root_path)
|
|
# give the daemon a chance to start up
|
|
if process.stdout:
|
|
process.stdout.readline()
|
|
await asyncio.sleep(1)
|
|
# it prints "daemon: listening"
|
|
connection = await connect_to_daemon_and_validate(root_path)
|
|
if connection:
|
|
return connection
|
|
return None
|
|
|
|
|
|
async def async_start(root_path: Path, group: str, restart: bool) -> None:
|
|
daemon = await create_start_daemon_connection(root_path)
|
|
if daemon is None:
|
|
print("Failed to create the chia daemon")
|
|
return
|
|
|
|
for service in services_for_groups(group):
|
|
if await daemon.is_running(service_name=service):
|
|
print(f"{service}: ", end="", flush=True)
|
|
if restart:
|
|
if not await daemon.is_running(service_name=service):
|
|
print("not running")
|
|
elif await daemon.stop_service(service_name=service):
|
|
print("stopped")
|
|
else:
|
|
print("stop failed")
|
|
else:
|
|
print("Already running, use `-r` to restart")
|
|
continue
|
|
print(f"{service}: ", end="", flush=True)
|
|
msg = await daemon.start_service(service_name=service)
|
|
success = msg["data"]["success"]
|
|
|
|
if success is True:
|
|
print("started")
|
|
else:
|
|
error = msg["data"]["error"]
|
|
print(f"{service} failed to start. Error: {error}")
|
|
await daemon.close()
|