o Th@sdZddlmZddlmZmZmZmZddlZddl Z ddl Z ddl m Z m Z e jdkZ e jdkrPddlZddlmZmZmZmZmZdd lmZmZmZnddlZddlZddlZddlZe jdkrud eeeeeffd d Z nd eeeeeffd d Z d eeeffddZ!ded e"fddZ#ded e"fddZ$eded edfddZ%ded e"fddZ&ded efddZ'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_RECTreturncCsGdddt}tjj}tjj}t|_tt|g|_|d}|}||t |}|rB|j j |j j d}|j j |j jd}||fSdS)Nc@s.eZdZdefdefdefdefdefgZdS)z-_pty_size..CONSOLE_SCREEN_BUFFER_INFOdwSizedwCursorPosition wAttributessrWindowdwMaximumWindowSizeN)__name__ __module__ __qualname__rr r_fields_rrC/var/www/html/venv/lib/python3.10/site-packages/invoke/terminals.pyCONSOLE_SCREEN_BUFFER_INFO3srirNN)r r kernel32 GetStdHandleGetConsoleScreenBufferInforrestyperargtypesrrRightLeftBottomTop)rr"r#hstdcsbiretsizexsizeyrrr _pty_size2s r/c Csbd}d}t|dd}zttjtj|}t||\}}||fWStj t t t fy0Y|Sw)zS Suitable for most POSIX platforms. .. versionadded:: 1.0 r HHr) structpackfcntlioctlsysstdouttermios TIOCGWINSZunpackerror TypeErrorIOErrorAttributeError)sizefmtbufresultrowscolsrrrr/Qs  cCst\}}|pd|p dfS)z Determine current local pseudoterminal dimensions. :returns: A ``(num_cols, num_rows)`` two-tuple describing PTY size. Defaults to ``(80, 24)`` if unable to get a sensible result dynamically. .. versionadded:: 1.0 P)r/)rCrBrrrpty_sizess rFstreamcCs"t|sdStt|kS)a Detect if given stdin ``stream`` seems to be in the foreground of a TTY. Specifically, compares the current Python process group ID to that of the stream's file descriptor to see if they match; if they do not match, it is likely that the process has been placed in the background. This is used as a test to determine whether we should manipulate an active stdin so it runs in a character-buffered mode; touching the terminal in this way when the process is backgrounded, causes most shells to pause execution. .. note:: Processes that aren't attached to a terminal to begin with, will always fail this test, as it starts with "do you have a real ``fileno``?". .. versionadded:: 1.0 F)rosgetpgrp tcgetpgrpfileno)rGrrrstdin_is_foregrounded_ttysrLcCsdt|}|d|d}}t|tj@}t|tj@}| | |tjdv|tjdvf}t|S)N)r)r)r7 tcgetattrboolECHOICANONVMINVTIMEall)rGattrslflagsccechoicanon sentinelsrrrcbreak_already_sets   r^)NNNc csltst|rt|rt|rdVdSt|}t|zdVWt|tj |dSt|tj |w)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 rLr^r7rQtty setcbreak tcsetattr TCSADRAIN)rG old_settingsrrrcharacter_buffereds    &reinput_cCsJt|sdStjdkrtSt|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)rr5platformmsvcrtkbhitselectrR)rfreads_rrrready_for_readings  rmcCs<tst|rt|rt|tjd}tt 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) r_r rr3r4r7FIONREADintr1r9)rffionreadrrr bytes_to_readsrr)(__doc__ contextlibrtypingrrrrrHrjr5utilrr rgr_rhctypesr r r rrctypes.wintypesrrrr3r1r7r`rpr/rFrRrLr^rermrrrrrrs>     "