|
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)) |
|
|
