o Ďi"@sdZddlmZddlmZddlmZddlZddlZddlmZddZ d d Z d d Z d dZ ddZ dZdZdZdZdZddZddZddZdS)z*OS specific console_attr helper functions.)absolute_import)division)unicode_literalsN)encodingcCs<d}ttttfD]}z |}|rWnWqYq|pdS)zGets the terminal x and y dimensions in characters. _GetTermSize*() helper functions taken from: http://stackoverflow.com/questions/263890/ Returns: (columns, lines): A tuple containing the terminal x and y dimensions. N)P)_GetTermSizePosix_GetTermSizeWindows_GetTermSizeEnvironment_GetTermSizeTput)xyget_terminal_sizerX/home/jeff/fluffinator/venv/lib/python3.10/site-packages/fire/console/console_attr_os.py GetTermSizes rc sddlddlddlfdd}|dp|dp|d}|sRd}z"ztttj}||}Wnd}YW|durEt||S|durQt|ww|S)z.Returns the Posix terminal x and y dimensions.rNcsBzd|jd}|r|d|dfWSdWSYdS)zReturns the terminal (x,y) size for fd. Args: fd: The terminal file descriptor. Returns: The terminal (x,y) size for fd or None on error. shhsjunkrN)unpackioctl TIOCGWINSZ)fdrcfcntlstructtermiosrr_GetXY=s  z!_GetTermSizePosix.._GetXYr)rrrosopenctermidO_RDONLYclose)rr rrrrr4s(   rc Csddl}ddlm}ddlm}|jd}|d}|j||s#dS|d|j\ }}}}} } } } } }}| | d}| | d}||fS) z0Returns the Windows terminal x and y dimensions.rN)create_string_buffer)windllis hhhhHhhhhhhr) rctypesr"r#kernel32 GetStdHandleGetConsoleScreenBufferInforraw)rr"r#hcsbi unused_bufx unused_bufy unused_curx unused_cury unused_wattrlefttoprightbottom unused_maxx unused_maxyxyrrrr \s       r cCsttjdttjdfS)z=Returns the terminal x and y dimensions from the environment.COLUMNSLINES)intrenvironrrrrr usr cCsTddl}t|jddg|jd}t|}t|jddg|jd}t|}||fS)z5Returns the terminal x and y dimensions from tput(1).rNtputcols)stderrlines) subprocessrDecode check_outputSTDOUTr;)rAoutputr>rowsrrrr zsr àcCs,ttfD] }z|WSYqddS)zReturns a function that reads one keypress from stdin with no echo. Returns: A function that reads one keypress from stdin with no echo or a function that always returns None if stdin does not support it. cSsdSNrrrrrsz#GetRawKeyFunction..)_GetRawKeyFunctionPosix_GetRawKeyFunctionWindows)get_raw_key_functionrrrGetRawKeyFunctions rQcs"ddlddlfdd}|S)z+_GetRawKeyFunction helper using Posix APIs.rNc sdddddddddddd }tjtjfd d }}zUzD|}|tkrg|} |tkrE|WWj |S| rJn |}|}|d krV|}nq5| |dWWj |SWnd}YWj |n j |w|t t fvrdS|S)zReads and returns one keypress from stdin, no echo, using Posix APIs. Returns: The key name, None for EOF, <*> for function keys, otherwise a character. ) ABDC56HFMSTcsttdS)Nr)rrBrreadrrrr _GetKeyCharszE_GetRawKeyFunctionPosix.._GetRawKeyPosix.._GetKeyCharT~N)sysstdoutflushstdinfileno tcgetattrsetraw _ANSI_CSI tcsetattr TCSADRAINisalphaget _CONTROL_D _CONTROL_Z) ansi_to_keyrg old_settingscprev_crttyrfr_GetRawKeyPosixsR        $z0_GetRawKeyFunctionPosix.._GetRawKeyPosix)r}r)r~rr|rrNs5rNcsddlfdd}|S)z-_GetRawKeyFunction helper using Windows APIs.rNc s`ddddddddd }tjfd d }|}|ttfvr&||d S|ttfvr.d S|S)zReads and returns one keypress from stdin, no echo, using Windows APIs. Returns: The key name, None for EOF, <*> for function keys, otherwise a character. rRrSrTrUrVrWrXrY)r`PKrbIQGOcstSrL)rrBgetchrmsvcrtrrrgszI_GetRawKeyFunctionWindows.._GetRawKeyWindows.._GetKeyCharriN)rjrkrl_WINDOWS_CSI_1_WINDOWS_CSI_2rurvrw)windows_to_keyrgrzrrr_GetRawKeyWindowss  z4_GetRawKeyFunctionWindows.._GetRawKeyWindowsr)rrrrrOs rO)__doc__ __future__rrrrrj fire.consolerrrr r r rqrvrwrrrQrNrOrrrrs(    (  ?