通过termios.TIOCSTI注入unicode字符

时间:2016-01-06 11:36:24

标签: python python-3.x unicode console termios

我有一段python代码,它将bash历史记录中的条目注入命令提示符。

在切换到Python 3之前,一切都运行良好。 现在德国的Umlaute出现了错误。

例如

python3 console_test.py mööp

结果:

$ m�

以下是相关代码:

import fcntl
import sys
import termios

command = sys.argv[1]

fd = sys.stdin.fileno()
old = termios.tcgetattr(fd)
new = termios.tcgetattr(fd)
new[3] = new[3] & ~termios.ECHO  # disable echo
termios.tcsetattr(fd, termios.TCSANOW, new)
for c in command:
    fcntl.ioctl(fd, termios.TIOCSTI, c)
termios.tcsetattr(fd, termios.TCSANOW, old)

我尝试将输入编码为utf-8,但这给了我:

OSError: [Errno 14] Bad address

1 个答案:

答案 0 :(得分:1)

我自己找到了答案,Python3自动使用文件系统编码对参数进行解码,因此在调用ioctl之前我必须将其反转:

import fcntl
import sys
import termios
import struct
import os

command = sys.argv[1]

if sys.version_info >= (3,):
    # reverse the automatic encoding and pack into a list of bytes
    command = (struct.pack('B', c) for c in os.fsencode(command))

fd = sys.stdin.fileno()
old = termios.tcgetattr(fd)
new = termios.tcgetattr(fd)
new[3] = new[3] & ~termios.ECHO  # disable echo
termios.tcsetattr(fd, termios.TCSANOW, new)
for c in command:
    fcntl.ioctl(fd, termios.TIOCSTI, c)

termios.tcsetattr(fd, termios.TCSANOW, old)