Python之telnetlib模块

telnetlib是python标准库中的一员,我们可以使用该模块以telnet的方式与服务器交互。请观察下面 示例 了解它的用法:

import telnetlib


def run_telnet(host, username, password, commands):
    """
    连接并登录服务器后,执行命令串
    """
    # 连接Telnet服务器
    tn = telnetlib.Telnet(host, port=23, timeout=10)
    tn.set_debuglevel(1)

    # 输入登录用户名
    tn.read_until(b'login: ')
    tn.write(username.encode('utf-8') + b'\n')

    # 输入登录密码
    tn.read_until(b'Password: ')
    tn.write(password.encode('utf-8') + b'\n')
    tn.read_until(b'\r\n$')

    # 登录完毕后执行命令
    for command in commands:
        tn.write(command.encode('utf-8') + b'\n')
        tn.read_until(b'\r\n$')

    # 终止Telnet连接
    tn.close()


if __name__ == '__main__':
    command_list = ['pwd', 'ls']
    run_telnet('172.31.0.151', 'username', 'password', command_list)

上述示例中依次完成了客户端与服务器的telnet连接,通过用户名及密码登录服务器,登陆后执行指定的命令串。telnetlib实现了一个Telnet类,该类实现了Telnet协议。该类在初始化过程中,调用了Telnet.open()函数,帮助客户端与服务器建立telnet连接。连接完成后,可以通过以下函数与服务器进行交互:

  • Telnet.read_until() :读取直到遇到给定的预期字节字符串,或者超时 。
  • Telnet.write() :将字节字符串写入套接字。
  • Telnet.close() :用于关闭telnet连接。