已知mac或ip远程H3C三层交换机查询ip或mac

#已知ip查mac
import telnetlib
import re

host = b'172.21.0.254'
user = b'admin\n'
password = b'password\n'
command = bytes((input('enter your ip:')),encoding = "utf8")

tn = telnetlib.Telnet(host)
tn.read_until(b'name:')
tn.write(user)
tn.read_until(b'ssword:')
tn.write(password)

tn.write(b'dis arp ' + command + b'\n')
tn.write(b'quit\n')

content = tn.read_all().decode('ascii')
match_mac = r'\w{4}-\w{4}-\w{4}'
mac = re.search(match_mac,content)
#print(content)
print('mac = ' + mac.group())

#已知mac查ip
import telnetlib
import re

host = b'172.21.0.254'
user = b'admin\n'
password = b'c$de6ase\n'
command = bytes((input('enter your mac:')),encoding = "utf8")

tn = telnetlib.Telnet(host)
tn.read_until(b'name:')
tn.write(user)
tn.read_until(b'ssword:')
tn.write(password)

tn.write(b'dis arp | inc  ' + command + b'\n')
tn.write(b'quit\n')

content = tn.read_all().decode('ascii')
match_ip = r'\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}'
ip = re.search(match_ip,content)
#print(content)
print('ip = ' + ip.group())