| from server import PromptServer
|
| from aiohttp import web
|
| from ..core import logger
|
| from ..general import cmonitor
|
|
|
| @PromptServer.instance.routes.patch("/crystools/monitor")
|
| async def newSettings(request):
|
| try:
|
| settings = await request.json()
|
|
|
|
|
| if 'rate' in settings is not None:
|
| rate = settings['rate']
|
| if type(rate) is not int and type(rate) is not float:
|
| raise Exception('Rate must be an number.')
|
|
|
| if cmonitor.rate == 0 and rate > 0:
|
| cmonitor.rate = rate
|
| cmonitor.startMonitor()
|
| else:
|
| cmonitor.rate = rate
|
|
|
|
|
| if 'switchCPU' in settings is not None:
|
| switchCPU = settings['switchCPU']
|
| if type(switchCPU) is not bool:
|
| raise Exception('switchCPU must be an boolean.')
|
|
|
| cmonitor.hardwareInfo.switchCPU = switchCPU
|
|
|
| if 'switchHDD' in settings is not None:
|
| switchHDD = settings['switchHDD']
|
| if type(switchHDD) is not bool:
|
| raise Exception('switchHDD must be an boolean.')
|
|
|
| cmonitor.hardwareInfo.switchHDD = switchHDD
|
|
|
| if 'switchRAM' in settings is not None:
|
| switchRAM = settings['switchRAM']
|
| if type(switchRAM) is not bool:
|
| raise Exception('switchRAM must be an boolean.')
|
|
|
| cmonitor.hardwareInfo.switchRAM = switchRAM
|
|
|
| if 'whichHDD' in settings is not None:
|
| whichHDD = settings['whichHDD']
|
| if type(whichHDD) is not str:
|
| raise Exception('whichHDD must be an string.')
|
|
|
| cmonitor.hardwareInfo.whichHDD = whichHDD
|
|
|
|
|
| return web.Response(status=200)
|
| except Exception as e:
|
| logger.error(e)
|
| return web.Response(status=400, text=str(e))
|
|
|
|
|
| @PromptServer.instance.routes.post("/crystools/monitor/switch")
|
| async def monitorSwitch(request):
|
| try:
|
| switch = await request.json()
|
|
|
| if 'monitor' in switch is not None:
|
| monitor = switch['monitor']
|
| if type(monitor) is not bool:
|
| raise Exception('monitor must be an boolean.')
|
|
|
| if monitor:
|
| cmonitor.startMonitor()
|
| else:
|
| cmonitor.stopMonitor()
|
|
|
| return web.Response(status=200)
|
| except Exception as e:
|
| logger.error(e)
|
| return web.Response(status=400, text=str(e))
|
|
|
|
|
| @PromptServer.instance.routes.get("/crystools/monitor/HDD")
|
| def getHDDs(request):
|
| try:
|
| return web.json_response(cmonitor.hardwareInfo.getHDDsInfo())
|
| except Exception as e:
|
| logger.error(e)
|
| return web.Response(status=400, text=str(e))
|
|
|
|
|
| @PromptServer.instance.routes.get("/crystools/monitor/GPU")
|
| def getGPUs(request):
|
| try:
|
| gpuInfo = cmonitor.hardwareInfo.getGPUInfo()
|
| return web.json_response(gpuInfo)
|
| except Exception as e:
|
| logger.error(e)
|
| return web.Response(status=400, text=str(e))
|
|
|
|
|
| @PromptServer.instance.routes.patch("/crystools/monitor/GPU/{index}")
|
| async def getGPUs(request):
|
| try:
|
| index = request.match_info["index"]
|
| settings = await request.json()
|
| if 'utilization' in settings is not None:
|
| if type(settings['utilization']) is not bool:
|
| raise Exception('utilization must be an boolean.')
|
|
|
| cmonitor.hardwareInfo.GPUInfo.gpusUtilization[int(index)] = settings['utilization']
|
|
|
| if 'vram' in settings is not None:
|
| if type(settings['vram']) is not bool:
|
| raise Exception('vram must be an boolean.')
|
|
|
| cmonitor.hardwareInfo.GPUInfo.gpusVRAM[int(index)] = settings['vram']
|
|
|
| if 'temperature' in settings is not None:
|
| if type(settings['temperature']) is not bool:
|
| raise Exception('temperature must be an boolean.')
|
|
|
| cmonitor.hardwareInfo.GPUInfo.gpusTemperature[int(index)] = settings['temperature']
|
|
|
| return web.Response(status=200)
|
| except Exception as e:
|
| logger.error(e)
|
| return web.Response(status=400, text=str(e))
|
|
|
