python腳本實現scp上傳下載功能

普通版本

 1 # -*- coding:utf-8 -*-
import paramiko,os,sys,time

port = 22
user = 'root'
def ssh_scp_put(ip,port,user,password,local_file,remote_file):
    ssh = paramiko.SSHClient()
    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    ssh.connect(ip, 22, 'root', password)
    a = ssh.exec_command('date')
    stdin, stdout, stderr = a
    print stdout.read()
    sftp = paramiko.SFTPClient.from_transport(ssh.get_transport())
    sftp = ssh.open_sftp()
    sftp.put(local_file, remote_file)

def ssh_scp_get(ip, port, user, password, remote_file, local_file):
    ssh = paramiko.SSHClient()
    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    ssh.connect(ip, 22, 'root', password)
    a = ssh.exec_command('date')
    stdin, stdout, stderr = a
    print stdout.read()
    sftp = paramiko.SFTPClient.from_transport(ssh.get_transport())
    sftp = ssh.open_sftp()
    sftp.get(remote_file, local_file)



ip = raw_input("請輸入遠端主機的IP地址：")
password = raw_input("請輸入遠端主機的密碼：")

while True:
    print '''
    -------歡迎使用 scp software--------
    上傳文件請輸入  [ 1 ]:
    下載文件請輸入  [ 2 ]:
    退出SCP請輸入   [ q ]:
    ------------------------------------
    '''
    choice = raw_input("請輸入 [ ]")
    if choice == "1":
        local_file = raw_input("請輸入本地文件的絕對路徑：")
        remote_file = raw_input("請輸入文件上傳的絕對路徑：")
        ssh_scp_put(ip,port,user,password,local_file,remote_file)
    elif choice == "2":
        remote_file = raw_input("請輸入遠端文件的絕對路徑：")
        local_file = raw_input("請輸入要放到本地的絕對路徑：")
        ssh_scp_get(ip,port,user,password,remote_file,local_file)
    elif choice == "q":
        print "感謝使用，再見"
        exit()
    else:
        print "輸入錯誤，請重新輸入："

 裝逼版本

import paramiko
import os
import sys
import getpass
print("\033[32;1m****開始配置目標機器信息*****\033[0m")
#ips = input("主機IP:")
#user = input("主機賬號:")
#password = getpass.getpass("主機密碼:")
#port = 22
user = "root"
ips = "10.10.123.96"
password = "B^Dc%4LSBvhZZK3B"
port = 22
class Tools(object):
    def __init__(self, user, password, port, ips):
        self.user = user
        self.password = password
        self.port = port
        self.ip = ips
    def connect(self):
        try:
            self.ssh = paramiko.SSHClient()
            self.ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
            self.ssh.connect(self.ip, self.port, self.user, self.password)
            print("連接已建立")
        except Exception as e:
            print("未能連接到主機")
    def cmd(self):
        cmd = input("請輸入要執行的命令:>>")
        stdout, stdin, stderr = self.ssh.exec_command(cmd)
        #print(sys.stdout.read())
    def input(self):
        self.local_file_abs = input("本地文件的絕對路徑:>>")
        self.remote_file_abs = input("遠程文件的絕對路徑:>>")
    def put(self):
        sftp = paramiko.SFTPClient.from_transport(self.ssh.get_transport())
        sftp = self.ssh.open_sftp()
        self.input()
        sftp.put(self.local_file_abs,self.remote_file_abs)
    def get(self):
        sftp = paramiko.SFTPClient.from_transport(self.ssh.get_transport())
        sftp = self.ssh.open_sftp()
        self.input()
        sftp.get(self.remote_file_abs,self.local_file_abs)
    def close(self):
        self.ssh.close()
        print("連接關閉")
obj = Tools(user, password, port, ips)
if __name__ == "__main__":
    msg = '''\033[32;1m
    執行命令 >>輸入cmd
    上傳文件 >>輸入put
    下載文件 >>輸入get
    退出     >>輸入q\033[0m
    '''
    getattr(obj, "connect")()
    while True:
        print(msg)
        inp = input("action:>>")
        if hasattr(obj,inp):
            getattr(obj,inp)()
        if inp == "q":
            getattr(obj,"close")()
            exit()
           else:print("沒有該選項，請重新輸入:>>")