我收到缓冲区错误，但我在第123行的'with open（）'块中有一个行缓冲区。这是缓冲区的正确位置吗？我是否应该在连接类中使用某些东西，或者也可以使用解析器参数。
我正在尝试使用带有地址或主机名的文件，使用'--host_file'参数在多个ASA上运行SNMP命令。它可以正常使用'--host'参数。
提供的任何帮助都会受到极大的欢迎。
Traceback (most recent call last):
File "asaos-snmpv3-tool.py", line 145, in <module>
main()
File "asaos-snmpv3-tool.py", line 124, in main
with open(hosts, mode='r', buffering=-1):
TypeError: coercing to Unicode: need string or buffer, file found
import pexpect
import argparse
PROMPT = ['# ', '>>> ', '>', '\$ ']
SNMPGROUPCMD = ' snmp-server group '
V3PRIVCMD = ' v3 priv '
SNMPSRVUSRCMD = ' snmp-server user '
V3AUTHCMD = ' v3 auth '
PRIVCMD = ' priv '
SNMPSRVHOSTCMD = ' snmp-server host '
VERSION3CMD = ' version 3 '
SHAHMACCMD = ' sha '
SNMPSRVENTRAP = ' snmp-server enable traps all '
WRME = ' write memory '
def send_command(child, cmd):
child.sendline(cmd)
child.expect(PROMPT)
print child.before
def connect(user, host, passwd, en_passwd):
ssh_newkey = 'Are you sure you want to continue connecting?'
constr = 'ssh ' + user + '@' + host
child = pexpect.spawn(constr)
ret = child.expect([pexpect.TIMEOUT, ssh_newkey, '[P|p]assword:'])
if ret == 0:
print '[-] Error Connecting'
return
if ret == 1:
child.sendline('yes')
ret = child.expect([pexpect.TIMEOUT, '[P|p]assword:'])
if ret == 0:
print '[-] Error Connecting'
return
child.sendline(passwd)
child.expect(PROMPT)
child.sendline('enable')
child.sendline(en_passwd)
child.expect(PROMPT)
child.sendline('config t')
child.expect(PROMPT)
return child
def connects(user, hosts, passwd, en_passwd):
ssh_newkey = 'Are you sure you want to continue connecting?'
constr = 'ssh ' + user + '@' + hosts
child = pexpect.spawn(constr)
ret = child.expect([pexpect.TIMEOUT, ssh_newkey, '[P|p]assword:'])
if ret == 0:
print '[-] Error Connecting'
return
if ret == 1:
child.sendline('yes')
ret = child.expect([pexpect.TIMEOUT, '[P|p]assword:'])
if ret == 0:
print '[-] Error Connecting'
return
child.sendline(passwd)
child.expect(PROMPT)
child.sendline('enable')
child.sendline(en_passwd)
child.expect(PROMPT)
child.sendline('config t')
child.expect(PROMPT)
return child
def main():
parser = argparse.ArgumentParser('usage %prog ' + '--host --host_file --username --password--enable --group --snmp_user --snmp_host --int_name --snmp_v3_auth --snmp_v3_hmac --snmp_v3_priv --snmp_v3_encr')
parser.add_argument('--host', dest='host', type=str, help='specify a target host')
parser.add_argument('--host_file', dest='hosts', type=file, help='specify a target host file')
parser.add_argument('--username', dest='user', type=str, help='specify a user name')
parser.add_argument('--password', dest='passwd', type=str, help='specify a passwd')
parser.add_argument('--enable', dest='en_passwd', type=str, help='specify an enable passwd')
parser.add_argument('--group', dest='group', type=str, help='specify an snmp group')
parser.add_argument('--snmp_user', dest='snmpuser', type=str, help='specify an snmp user')
parser.add_argument('--snmp_host', dest='snmphost', type=str, help='specify an snmp server host')
parser.add_argument('--int_name', dest='intname', type=str, help='specify interface name')
parser.add_argument('--snmp_v3_auth', dest='snmpauth', type=str, help='specify the snmp user authentication')
parser.add_argument('--snmp_v3_hmac', dest='snmphmac', type=str, help='set snmp HMAC, md5 or sha')
parser.add_argument('--snmp_v3_priv', dest='snmppriv', type=str, help='specify the snmp priv password')
parser.add_argument('--snmp_v3_encr', dest='snmpencrypt', type=str, help='specify encryption, des, 3des, or aes(128/192/256)')
args = parser.parse_args()
host = args.host
hosts = args.hosts
user = args.user
passwd = args.passwd
en_passwd = args.en_passwd
group = args.group
snmpuser = args.snmpuser
snmphost = args.snmphost
intname = args.intname
snmpauth = args.snmpauth
snmppriv = args.snmppriv
snmpencrypt = args.snmpencrypt
if hosts:
with open(hosts, mode='r', buffering=1):
for line in hosts:
hosts = line.rstrip
child = connects(user, hosts, passwd, en_passwd)
send_command(child, SNMPGROUPCMD + group + V3PRIVCMD)
send_command(child, SNMPSRVUSRCMD + snmpuser + ' ' + group + V3AUTHCMD + SHAHMACCMD + snmpauth + PRIVCMD + snmpencrypt + ' ' + snmppriv)
send_command(child, SNMPSRVHOSTCMD + intname + ' ' + snmphost + VERSION3CMD + snmpuser)
send_command(child, SNMPSRVENTRAP)
send_command(child, WRME)
elif host:
child = connect(user, host, passwd, en_passwd)
send_command(child, SNMPGROUPCMD + group + V3PRIVCMD)
send_command(child, SNMPSRVUSRCMD + snmpuser + ' ' + group + V3AUTHCMD + SHAHMACCMD + snmpauth + PRIVCMD + snmpencrypt + ' ' + snmppriv)
send_command(child, SNMPSRVHOSTCMD + intname + ' ' + snmphost + VERSION3CMD + snmpuser)
send_command(child, SNMPSRVENTRAP)
send_command(child, WRME)
else:
print ('Specify either --host or --host_file or I have nothing to do')
if __name__ == '__main__':
main()
答案 0 :(得分：4)
这里有一系列问题：
if hosts:
with open(hosts, mode='r', buffering=1):
在这里，hosts显然是一个文件名。你打开那个文件......然后不要把它存储在任何地方的变量中，然后继续使用文件名就好像它是一个文件一样。
但是，在你到达那里之前，hosts并非实际一个文件名。 argparse的一个漂亮功能是它可以自动为您打开文件，并且您要求它在参数规范中使用type=file来执行此操作。这意味着hosts实际上是一个文件对象，并且您正在尝试将该文件对象本身用作文件名！这实际上导致TypeError: coercing to Unicode: need string or buffer, file found：open尝试将文件名转换为Unicode字符串，而且它不知道如何使用文件对象。如果你想要open主机，不要告诉argparse为你做这件事;只需将其类型保留为str。
for line in hosts:
由于hosts应该是一个文件名，作为一个字符串，这将为文件名中的每个字符提供一个字符串。哪个没用。
hosts = line.rstrip
这会覆盖您需要的hosts变量。更糟糕的是，它使用绑定方法line.rstrip覆盖它，而不是调用line.rstrip的结果。
child = connects(user, hosts, passwd, en_passwd)
在这里，您将该绑定方法传递给需要字符串的函数，该函数不起作用。
不要一遍又一遍地使用相同的变量名来表示不同的东西。为每件事使用不同的名称。理想情况下，其名称在某种程度上与其持有的东西有关。例如，名为hosts的文件中的ech（剥离）行可能是host或hosts_entry ...有一件事情不就是你的全部内容hosts。
无论如何，要解决所有这些问题：
if hosts:
for line in hosts:
host = line.rstrip()
child = connects(user, host, passwd, en_passwd)
或者，如果你想控制hosts的打开方式（我不确定为什么你认为指定buffering=1很重要，但可能你有一些理由？）：
parser.add_argument('--host_file', dest='hosts', type=str, help='specify a target host file')
# …
if hosts:
with open(hosts, buffering=1) as hosts_file:
for line in hosts_file:
host = line.rstrip()
child = connects(user, host, passwd, en_passwd)
答案 1 :(得分：1)
它看起来就像你在这里迭代错误的东西一样：
with open(hosts, mode='r', buffering=1):
for line in hosts:
hosts = line.rstrip
你可能想要这个：
with open(hosts, mode='r', buffering=1) as lines:
for line in lines:
host = line.rstrip