o úÜÓh ã@s\dZddlmZddlmZmZdefdd„Zdeefd d „ZGd d „d e ƒZ d d„Z dS)zž This file contains utilities for handling input from the user and registering specific keys to specific functions, based on https://github.com/bchao1/bullet é)ÚListé)ÚKEYMAPÚ get_characterÚkeycó‡fdd„}|S)zR Mark the function with the key code so it can be handled in the register cs&t|dgƒ}|ˆg7}t|d|ƒ|S©NÚ handle_key©ÚgetattrÚsetattr©ÚfuncÚhandle©r©úT/var/www/html/ai/venv/lib/python3.10/site-packages/accelerate/commands/menu/input.pyÚ decorators   zmark..decoratorr)rrrrrÚmarkó rÚkeyscr)zS Mark the function with the key codes so it can be handled in the register cs$t|dgƒ}|ˆ7}t|d|ƒ|Srr r ©rrrr,s  z mark_multiple..decoratorr)rrrrrÚ mark_multiple'rrcs,eZdZdZ‡fdd„Zedd„ƒZ‡ZS)Ú KeyHandlerz; Metaclass that adds the key handlers to the class cshtƒ ||||¡}t|dƒst|diƒt|dtjƒ| ¡D]}t|dgƒ}|D]}||j|<q)q|S)NÚ key_handlerÚ handle_inputr ) ÚsuperÚ__new__Úhasattrr rrÚvaluesr r)ÚclsÚnameÚbasesÚattrsÚnew_clsÚvalueÚ handled_keysr©Ú __class__rrr:s     ÿzKeyHandler.__new__cCs<tƒ}|tdkr t|ƒ}|j |¡}|r||_||ƒSdS)zDFinds and returns the selected character if it exists in the handlerÚ undefinedN)rrÚordrÚgetÚcurrent_selection)r ÚcharÚhandlerrrrrFs  zKeyHandler.handle_input)Ú__name__Ú __module__Ú __qualname__Ú__doc__rÚ staticmethodrÚ __classcell__rrr'rr5s   rcCst|j|j|j ¡ƒS)z&Adds KeyHandler metaclass to the class)rr/Ú __bases__Ú__dict__Úcopy)r rrrÚregisterTsr8N) r2ÚtypingrÚkeymaprrÚstrrrÚtyperr8rrrrÚs