如何在telnetlib的write函数字符串中嵌入变量实现循环配置

问题描述 #

我编写了一份用于Telnet配置网络设备的Python脚本，当前脚本可正常运行，但希望将创建loop接口及配置IP的代码段改为循环实现。尝试通过字符串拼接嵌入变量时，出现TypeError: can't concat str to bytes错误——使用print可正常输出，但telnetlib.write函数无法运行，需解决该问题。

原脚本片段 #

tn.write(b"int loop 0\n")
tn.write(b"ip address 1.1.1.1 255.255.255.255\n")
tn.write(b"int loop 1\n")
tn.write(b"ip address 2.2.2.2 255.255.255.255\n")

尝试的循环代码 #

for i in range(2):
    tn.write(b"int loop "+ str(i)+"\n")
    tn.write(b"ip address "+str(i)+'.'+str(i)+'.'+str(i)+'.'+str(i) + " 255.255.255.255\n")

错误信息 #

Traceback (most recent call last):
  File "R1_scr1.py", line 22, in <module>
    tn.write(b"int loop "+ str(i)+"\n")
TypeError: can't concat str to bytes

解决方案 #

错误根源是字节串与普通字符串直接拼接：b"int loop "是字节类型，str(i)是字符串类型，Python不允许不同类型的字符串直接相加，而telnetlib的write方法仅接受字节串作为参数。

提供两种可行修复方式：

方式1：先拼接字符串，再转字节串 #

用f-string先构造完整的命令字符串，再通过encode()方法转成字节串：

for i in range(2):
    tn.write(f"int loop {i}\n".encode())
    ip_str = f"{i}.{i}.{i}.{i}"
    tn.write(f"ip address {ip_str} 255.255.255.255\n".encode())

方式2：直接使用字节串格式化 #

利用字节串的格式化语法，直接生成符合要求的字节串：

for i in range(2):
    tn.write(b"int loop %d\n" % i)
    tn.write(b"ip address %d.%d.%d.%d 255.255.255.255\n" % (i, i, i, i))

两种方式都能解决类型不匹配问题，方式1更直观易读，适合复杂命令拼接；方式2无需转换字符串，效率略高，可根据个人习惯选择。

内容的提问来源于stack exchange，提问作者user1993416