Python 获取 Linux 控制台窗口宽度，灵活适配终端输出大小

2024-03-16 23:07:26

获取 Linux 控制台窗口宽度的 Python 技巧

简介

在 Python 开发中，了解控制台窗口的大小有时至关重要，尤其是在处理需要适应终端宽度的文本输出时。本文将详细介绍如何在 Linux 系统上使用 Python 获取控制台窗口宽度，让你轻松掌握这个有用的技巧。

使用 struct 和 fcntl 模块

第一步是导入 struct 和 fcntl 模块，它们提供操纵和获取结构化数据的强大工具。接下来，通过 sys.stdin.fileno() 获取终端文件符，然后使用 fcntl.ioctl 函数获取终端窗口大小。

import struct
import fcntl

fd = sys.stdin.fileno()
window_size = fcntl.ioctl(fd, fcntl.TIOCGWINSZ, struct.pack('HHHH', 0, 0, 0, 0))

width = window_size[1]
height = window_size[0]

将 window_size 解析为元组后，即可轻松提取窗口宽度和高度。

使用 os 模块

一种更简单的方法是使用 os 模块。它提供了一个名为 get_terminal_size() 的方便函数，可以立即返回终端窗口大小。

import os

window_size = os.get_terminal_size()
width = window_size.columns
height = window_size.lines

注意事项

这些方法仅适用于 Linux 系统。
Windows 系统需要不同的方法，例如使用 msvcrt 模块的 get_osfhandle 和 get_terminal_size 函数。
其他操作系统可能需要特定于操作系统的解决方法。

示例

为了展示这些方法的使用，让我们编写一个 Python 程序来获取控制台窗口宽度并打印出来。

import struct
import fcntl

fd = sys.stdin.fileno()
window_size = fcntl.ioctl(fd, fcntl.TIOCGWINSZ, struct.pack('HHHH', 0, 0, 0, 0))
width = window_size[1]

print(f"控制台窗口宽度：{width}")