U ÷Õ/e> ã@s’UddlmZddlZddlmZGdd„deƒZGdd„deƒZGdd „d eƒZGd d „d eƒZd eifd eifdeifdeifgZ de d<dS)é)Ú annotationsN)ÚRequestHandlerc@seZdZdd„ZdS)Ú APIHandlercCs| d¡| dd¡dS)NzAPI V1ú Content-Typez text/plain)ÚwriteÚ set_header)Úself©r úB/tmp/pip-unpacked-wheel-g426oqom/distributed/http/scheduler/api.pyÚget s zAPIHandler.getN©Ú__name__Ú __module__Ú __qualname__r r r r r rsrc@seZdZdd„ZdS)ÚRetireWorkersHandlerc ÃsÎ| dd¡|j}zpt |jj¡}| dd¡}|rR|j|d}|j|dIdH}n| di¡}|j|dIdH}|  t  |¡¡WnFt k rÈ}z(|  dt |ƒ¡|  t  d d i¡¡W5d}~XYnXdS) Nrúapplication/jsonÚnr)r)ÚworkersréôÚErrorúInternal Server Error)rÚserverÚjsonÚloadsÚrequestÚbodyr Zworkers_to_closeZretire_workersrÚdumpsÚ ExceptionÚ set_statusÚstr)rÚ schedulerÚparamsZ n_workersrZ workers_infoÚer r r Úposts    zRetireWorkersHandler.postN)r rrr#r r r r rsrc@seZdZdd„ZdS)ÚGetWorkersHandlerc Cs’| dd¡|j}z4t|jƒdd„|j ¡Dƒdœ}| t |¡¡WnFtk rŒ}z(|  dt |ƒ¡| t ddi¡¡W5d}~XYnXdS) NrrcSsg|]}|j|jdœ‘qS)©ÚnameÚaddressr%)Ú.0Úwsr r r Ú (sÿz)GetWorkersHandler.get..)Z num_workersrrrr) rrÚlenrÚvaluesrrrrrr)rr Úresponser"r r r r "s þþzGetWorkersHandler.getNr r r r r r$!sr$c@seZdZdd„ZdS)ÚAdaptiveTargetHandlerc Cs‚| dd¡|j}z$| ¡}d|i}| t |¡¡WnFtk r|}z(| dt|ƒ¡| t ddi¡¡W5d}~XYnXdS)Nrrrrrr) rrZadaptive_targetrrrrrr)rr Zdesired_workersr-r"r r r r 4s ÿzAdaptiveTargetHandler.getNr r r r r r.3sr.z/api/v1z/api/v1/retire_workersz/api/v1/get_workersz/api/v1/adaptive_targetz list[tuple]Úroutes) Ú __future__rrZdistributed.http.utilsrrrr$r.r/Ú__annotations__r r r r Ús  ü