Python3：有沒有辦法像在 python2 中一樣使用 telnetlib，沒有 ascii 編碼和 b 前綴？

Question

有沒有辦法使用 telnetlib 使 python2 腳本與 python3 兼容？

我注意到我需要在 read_until() 前面加上字母 b，並且當我想 write() 時我需要在字符串上使用 encode('ascii')。

蟒蛇2

tn = telnetlib.Telnet("192.168.1.45")
tn.write("ls " + dirname + "\n")
answer = tn.read_until(":/$")

蟒蛇3

tn = telnetlib.Telnet("192.168.1.45")
cmd_str = "ls " + dirname + "\n"
tn.write(cmd_str.encode('ascii'))
answer = tn.read_until(b":/$")

這將幫助我將許多腳本更新到 3.x，因為它是唯一的主要變化。

謝謝！

Answer 1

您可以在encodingtelnetlib.py編寫自己的子類

class Telnet(Telnet):

    def __init__(self, host=None, port=0,
                 timeout=socket._GLOBAL_DEFAULT_TIMEOUT,
                 encoding='ascii'):
         self.encoding = encoding
         super().__init__(host, port, timeout)

    def write(self, buffer):
        if isinstance(buffer, str):
            buffer = buffer.encode(self.encoding)
        return super().write(buffer)

    # and etc.... for other methods

現在是將import telnetlib更改為import encodingtelnetlib as telnetlib 。 這比找到每個讀取和寫入更容易。

Answer 2

可能不是因為 Python3 默認使用utf-8編碼，而 Python2 使用ascii 。 因此，如果您需要一個ascii編碼的字符串，您需要手動更改編碼。