mirror of
https://github.com/python/cpython.git
synced 2025-07-23 19:25:40 +00:00
gh-43414: os.get_terminal_size() now uses the actual file descriptor on Windows instead of mapping to standard handles (#93203)
This commit is contained in:
parent
390847922c
commit
edb72047f3
3 changed files with 17 additions and 15 deletions
|
@ -3665,6 +3665,19 @@ class TermsizeTests(unittest.TestCase):
|
||||||
raise
|
raise
|
||||||
self.assertEqual(expected, actual)
|
self.assertEqual(expected, actual)
|
||||||
|
|
||||||
|
@unittest.skipUnless(sys.platform == 'win32', 'Windows specific test')
|
||||||
|
def test_windows_fd(self):
|
||||||
|
"""Check if get_terminal_size() returns a meaningful value in Windows"""
|
||||||
|
try:
|
||||||
|
conout = open('conout$', 'w')
|
||||||
|
except OSError:
|
||||||
|
self.skipTest('failed to open conout$')
|
||||||
|
with conout:
|
||||||
|
size = os.get_terminal_size(conout.fileno())
|
||||||
|
|
||||||
|
self.assertGreaterEqual(size.columns, 0)
|
||||||
|
self.assertGreaterEqual(size.lines, 0)
|
||||||
|
|
||||||
|
|
||||||
@unittest.skipUnless(hasattr(os, 'memfd_create'), 'requires os.memfd_create')
|
@unittest.skipUnless(hasattr(os, 'memfd_create'), 'requires os.memfd_create')
|
||||||
@support.requires_linux_version(3, 17)
|
@support.requires_linux_version(3, 17)
|
||||||
|
|
|
@ -0,0 +1,2 @@
|
||||||
|
:func:`os.get_terminal_size` now attempts to read the size from any provided
|
||||||
|
handle, rather than only supporting file descriptors 0, 1 and 2.
|
|
@ -13247,24 +13247,11 @@ os_get_terminal_size_impl(PyObject *module, int fd)
|
||||||
|
|
||||||
#ifdef TERMSIZE_USE_CONIO
|
#ifdef TERMSIZE_USE_CONIO
|
||||||
{
|
{
|
||||||
DWORD nhandle;
|
|
||||||
HANDLE handle;
|
HANDLE handle;
|
||||||
CONSOLE_SCREEN_BUFFER_INFO csbi;
|
CONSOLE_SCREEN_BUFFER_INFO csbi;
|
||||||
switch (fd) {
|
handle = _Py_get_osfhandle(fd);
|
||||||
case 0: nhandle = STD_INPUT_HANDLE;
|
|
||||||
break;
|
|
||||||
case 1: nhandle = STD_OUTPUT_HANDLE;
|
|
||||||
break;
|
|
||||||
case 2: nhandle = STD_ERROR_HANDLE;
|
|
||||||
break;
|
|
||||||
default:
|
|
||||||
return PyErr_Format(PyExc_ValueError, "bad file descriptor");
|
|
||||||
}
|
|
||||||
handle = GetStdHandle(nhandle);
|
|
||||||
if (handle == NULL)
|
|
||||||
return PyErr_Format(PyExc_OSError, "handle cannot be retrieved");
|
|
||||||
if (handle == INVALID_HANDLE_VALUE)
|
if (handle == INVALID_HANDLE_VALUE)
|
||||||
return PyErr_SetFromWindowsErr(0);
|
return NULL;
|
||||||
|
|
||||||
if (!GetConsoleScreenBufferInfo(handle, &csbi))
|
if (!GetConsoleScreenBufferInfo(handle, &csbi))
|
||||||
return PyErr_SetFromWindowsErr(0);
|
return PyErr_SetFromWindowsErr(0);
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue