Load setproctitle as runtime.

setup.py

@@ -1,11 +1,5 @@
 from setuptools import setup
 
-SETPROCTITLE_GITHUB = (
-    "setproctitle @ "
-    "https://github.com/Chia-Network/py-setproctitle/tarball/"
-    "d2ed86c5080bb645d8f6b782a4a86706c860d9e6#egg=setproctitle-50.0.0"
-)
-
 CLVM_TOOLS_GITHUB = (
     "clvm-tools @ "
     "https://github.com/Chia-Network/clvm_tools/tarball/6ff53bcfeb0c970647b6cfdde360d32b316b1326#egg=clvm-tools"
@@ -26,7 +20,6 @@ dependencies = [
     "chiapos==0.12.3",  # proof of space
     "sortedcontainers==2.1.0",  # For maintaining sorted mempools
     "websockets==8.1.0",  # For use in wallet RPC and electron UI
-    SETPROCTITLE_GITHUB,  # custom internal version of setproctitle, this should go away
     CLVM_TOOLS_GITHUB,  # clvm compiler tools
 ]
 

src/server/start_farmer.py

@@ -13,7 +13,7 @@ from src.server.server import ChiaServer
 from src.types.peer_info import PeerInfo
 from src.util.logging import initialize_logging
 from src.util.config import load_config, load_config_cli
-from setproctitle import setproctitle
+from src.util.setproctitle import setproctitle
 
 
 async def main():

src/server/start_full_node.py

@@ -25,7 +25,7 @@ from src.full_node.coin_store import CoinStore
 from src.util.logging import initialize_logging
 from src.util.config import load_config_cli
 from src.util.pip_import import pip_import
-from setproctitle import setproctitle
+from src.util.setproctitle import setproctitle
 
 
 async def main():

src/server/start_harvester.py

@@ -13,7 +13,7 @@ from src.server.server import ChiaServer
 from src.types.peer_info import PeerInfo
 from src.util.logging import initialize_logging
 from src.util.config import load_config, load_config_cli
-from setproctitle import setproctitle
+from src.util.setproctitle import setproctitle
 
 
 async def main():

src/server/start_introducer.py

@@ -12,7 +12,7 @@ from src.server.outbound_message import NodeType
 from src.server.server import ChiaServer
 from src.util.logging import initialize_logging
 from src.util.config import load_config_cli
-from setproctitle import setproctitle
+from src.util.setproctitle import setproctitle
 
 
 async def main():

src/server/start_timelord.py

@@ -16,7 +16,7 @@ from src.timelord import Timelord
 from src.types.peer_info import PeerInfo
 from src.util.logging import initialize_logging
 from src.util.config import load_config_cli
-from setproctitle import setproctitle
+from src.util.setproctitle import setproctitle
 
 
 async def main():

src/simulator/start_simulator.py

@@ -22,7 +22,7 @@ from src.types.full_block import FullBlock
 from src.full_node.coin_store import CoinStore
 from src.util.logging import initialize_logging
 from src.util.config import load_config_cli
-from setproctitle import setproctitle
+from src.util.setproctitle import setproctitle
 
 
 async def main():

src/timelord_launcher.py

@@ -7,7 +7,7 @@ from src.util.logging import initialize_logging
 from src.util.config import load_config
 from asyncio import Lock
 from typing import List
-from setproctitle import setproctitle
+from src.util.setproctitle import setproctitle
 
 config = load_config("config.yaml", "timelord_launcher")
 

src/util/setproctitle.py

@@ -0,0 +1,13 @@
+from .pip_import import pip_import
+
+
+SETPROCTITLE_GITHUB = (
+    "setproctitle @ "
+    "https://github.com/Chia-Network/py-setproctitle/tarball/"
+    "d2ed86c5080bb645d8f6b782a4a86706c860d9e6#egg=setproctitle-50.0.0"
+)
+
+
+def setproctitle(ps_name):
+    pysetproctitle = pip_import("setproctitle", SETPROCTITLE_GITHUB)
+    pysetproctitle.setproctitle(ps_name)

src/wallet/websocket_server.py

@@ -29,7 +29,7 @@ from src.wallet.util.wallet_types import WalletType
 from src.wallet.wallet_info import WalletInfo
 from src.wallet.wallet_node import WalletNode
 from src.types.mempool_inclusion_status import MempoolInclusionStatus
-from setproctitle import setproctitle
+from src.util.setproctitle import setproctitle
 
 # Timeout for response from wallet/full node for sending a transaction
 TIMEOUT = 5

tests/test_setproctitle.py

@@ -1,5 +1,5 @@
 import unittest
-from setproctitle import setproctitle
+from src.util.setproctitle import setproctitle
 
 
 class TestSetProcTitle(unittest.TestCase):