xjdZddlmZddlmZmZmZmZddlZddl Z ddl Z ddl m Z m Z e jdkZ e jdkrddlZddlmZmZmZmZmZdd lmZmZmZnddlZddlZddlZddlZe jdkrd eeeeeffd Z nd eeeeeffd Z d eeeffd Z!ded e"fdZ#ded e"fdZ$eded edfdZ%ded e"fdZ&ded efdZ'dS)z Utility functions surrounding terminal devices & I/O. Much of this code performs platform-sensitive branching, e.g. Windows support. This is its own module to abstract away what would otherwise be distracting logic-flow interruptions. )contextmanager) GeneratorIOOptionalTupleN) has_filenoisattywin32) Structurec_ushortwindllPOINTERbyref)HANDLE_COORD _SMALL_RECTreturncGddt}tjj}tjj}t |_t t|g|_|d}|}||t|}|r<|j j |j j z dz}|j j |j jz dz}||fSdS)Nc0eZdZdefdefdefdefdefgZdS)-_pty_size..CONSOLE_SCREEN_BUFFER_INFOdwSizedwCursorPosition wAttributessrWindowdwMaximumWindowSizeN)__name__ __module__ __qualname__rr r_fields_AC:\Projects\Historical\venv\Lib\site-packages\invoke/terminals.pyCONSOLE_SCREEN_BUFFER_INFOr3s=6"#V,)[)&/ HHHr"r$irNN)r rkernel32 GetStdHandleGetConsoleScreenBufferInforrestyperargtypesrrRightLeftBottomTop)r$r'r(hstdcsbiretsizexsizeys r# _pty_sizer42s        3 %+_%O"%   . / // "+ |C  ))++((uT{{;;  M'$-*<) * *F    7<L( 7=\) I y>>r")NNNc#bKts-t|rt|rt|rdVdSt j|}t j| dVt j|tj |dS#t j|tj |wxYw)z Force local terminal ``stream`` be character, not line, buffered. Only applies to Unix-based systems; on Windows this is a no-op. .. versionadded:: 1.0 N) WINDOWSr rUrhr=r[tty setcbreak tcsetattr TCSADRAIN)rO old_settingss r#character_bufferedrps  Gf~~ G)00 G f % % G (00  f G EEE  fg&7 F F F F FG fg&7 F F F Fs &B "B.input_ct|sdStjdkrtjSt j|gggd\}}}t |o |d|uS)z Test ``input_`` to determine whether a read action will succeed. :param input_: Input stream object (file-like). :returns: ``True`` if a read should succeed, ``False`` otherwise. .. versionadded:: 1.0 Tr gr)r r;platformmsvcrtkbhitselectr\)rqreadsrJs r#ready_for_readingrxsk f  t |w|~~mVHb"c:: q!E0eAh&0111r"ctsft|rWt|rHtj|t jd}ttj d|dSdS)a| Query stream ``input_`` to see how many bytes may be readable. .. note:: If we are unable to tell (e.g. if ``input_`` isn't a true file descriptor or isn't a valid TTY) we fall back to suggesting reading 1 byte only. :param input: Input stream object (file-like). :returns: `int` number of bytes to read. .. versionadded:: 1.0 s hrr) rjr r r9r:r=FIONREADintr7r?)rqfionreads r# bytes_to_readr~sb$ 4vf~~4*V*<*<4;vw'7??6=h//2333 1r")(__doc__ contextlibrtypingrrrrrQrvr;utilr r rsrjrtctypesr r rrrctypes.wintypesrrrr9r7r=rkr|r4rNr\rUrhrprxr~r!r"r#rs&%%%%%111111111111 %$$$$$$$ ,' !<7MMM<;;;;;;;;;;LLLMMMNNNJJJ<7 uXc]HSM9:     > uXc]HSM9:    F $%S/ $ $ $ $9b9T99990rd,G G GGGG42b2T2222. "       r"