U ,c@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|}|r&Wq8WqYqXq|p>dS)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_sizer@/tmp/pip-unpacked-wheel-1xt1w7un/fire/console/console_attr_os.py GetTermSizes  rc sddlddlddlfdd}|dp>|dp>|d}|sd}z4ztttj}||}Wnd}YnXW5|dk rt|X|S)z.Returns the Posix terminal x and y dimensions.rNcsFz2d|jd}|r.|d|dfndWSYdSXdS)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. shhZjunkrN)unpackZioctl TIOCGWINSZ)fdrcfcntlstructtermiosrr_GetXY@s z!_GetTermSizePosix.._GetXYr)rrroscloseopenctermidO_RDONLY)rr rrrrr7s  rc Csddl}ddlm}ddlm}|jd}|d}|j||sFdS|d|j\ }}}}} } } } } }}| | d}| | d}||fS) z0Returns the Windows terminal x and y dimensions.rN)create_string_buffer)windllis hhhhHhhhhhhr) rctypesr!r"kernel32Z GetStdHandleGetConsoleScreenBufferInforraw)rr!r"hcsbiZ unused_bufxZ unused_bufyZ unused_curxZ unused_curyZ unused_wattrlefttoprightZbottomZ unused_maxxZ unused_maxyxyrrrr _s$      r cCsttjdttjdfS)z=Returns the terminal x and y dimensions from the environment.COLUMNSLINES)intrenvironrrrrr xsr cCsTddl}t|jddg|jd}t|}t|jddg|jd}t|}||fS)z5Returns the terminal x and y dimemsions from tput(1).rNZtputcols)stderrlines) subprocessrDecode check_outputSTDOUTr1)r6outputr3Zrowsrrrr }sr àcCs0ttfD]}z |WSYqXqddS)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. cSsdSNrrrrrz#GetRawKeyFunction..)_GetRawKeyFunctionPosix_GetRawKeyFunctionWindows)Zget_raw_key_functionrrrGetRawKeyFunctions rEcs"ddlddlfdd}|S)z+_GetRawKeyFunction helper using Posix APIs.rNc sdddddddddddd }tjtjfd d }}zzj|}|t kr|}|t krz|WWPS| rq|}|}|d krh|}qqh| |d WWSWnd}YnXW5j|X|t t fkrdS|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)rr7rreadrrrr _GetKeyCharszE_GetRawKeyFunctionPosix.._GetRawKeyPosix.._GetKeyChar~N)sysstdoutflushstdinfileno tcgetattr tcsetattr TCSADRAINZsetraw _ANSI_CSIisalphaget _CONTROL_D _CONTROL_Z)Z ansi_to_keyr[Z old_settingscZprev_crttyrZr_GetRawKeyPosixsF      z0_GetRawKeyFunctionPosix.._GetRawKeyPosix)rmr)rnrrlrrCs5rCcsddlfdd}|S)z-_GetRawKeyFunction helper using Windows APIs.rNc s`ddddddddd }tjfd d }|}|ttfkrL||d S|ttfkr\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. rFrGrHrIrJrKrLrM)rTPKrVIQGOcstSr@)rr7Zgetchrmsvcrtrrr[szI_GetRawKeyFunctionWindows.._GetRawKeyWindows.._GetKeyCharr]N)r^r_r`_WINDOWS_CSI_1_WINDOWS_CSI_2rhrirj)Zwindows_to_keyr[rkrurr_GetRawKeyWindowss   z4_GetRawKeyFunctionWindows.._GetRawKeyWindowsru)ryrrurrDs rD)__doc__ __future__rrrrr^Z fire.consolerrrr r r rfrirjrwrxrErCrDrrrrs&    ( ?