执行python脚本并将输出保存为CSV文件

时间:2016-12-23 21:59:03

标签: python csv ssh paramiko

我想将python脚本(sshClient.py)的输出保存为CSV文件。你能帮助我怎么做吗?提前致谢。我在这里分享了代码和结果输出,以便您更好地理解。

import sys
import time
import select
import paramiko

host = '169.254.115.1'
i = 1

#
# Try to connect to the host.
# Retry a few times if it fails.
#
while True:
    print ('Trying to connect to %s (%i/3)' % (host, i))

    try:
        ssh = paramiko.SSHClient()
        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        ssh.connect(host, port=22, username='user', password='user')
        print ("Connected to %s" % host)
        break
    except paramiko.AuthenticationException:
        print ("Authentication failed when connecting to %s") % host
        sys.exit(1)
    except:
        print ("Could not SSH to %s, waiting for it to start" % host)
        i += 1
        time.sleep(2)

    # If we could not connect within time limit
    if i == 3:
        print ("Could not connect to %s. Giving up") % host
        sys.exit(1)

# Send the command (non-blocking)
stdin, stdout, stderr = ssh.exec_command("cd /opt/cohda/test; sudo ./runtest_iperf_tx.sh")

# Wait for the command to terminate
while not stdout.channel.exit_status_ready():
    # Only print data if there is data to read in the channel
    if stdout.channel.recv_ready():
        rl, wl, xl = select.select([stdout.channel], [], [], 0.0)
        if len(rl) > 0:
            # Print data from stdout
            print (stdout.channel.recv(1024)),
#
# Disconnect from the host
#
print ("Command done, closing SSH connection")
ssh.close()

结果输出:Throughput Vs Time

1 个答案:

答案 0 :(得分:-1)

未经测试,但如果数据是ascii ......

,那么这种效果应该有效
# Wait for the command to terminate
csv_str = ""
while not stdout.channel.exit_status_ready():
    # Only print data if there is data to read in the channel
    if stdout.channel.recv_ready():
        rl, wl, xl = select.select([stdout.channel], [], [], 0.0)
        if len(rl) > 0:
            # Print data from stdout
            print (stdout.channel.recv(1024)),
            csv_str += str.split(str(stdout.channel.recv(1024)),",")
#
# Disconnect from the host
#
PathFileName = "/hard/path/filename.csv"
f = open(PathFileName,'a')
f.write(csv_str)
f.close()