#!/usr/bin/python
# coding=utf-8

import paramiko
import os

def sftp_upload(host,port,username,password,local,remote):
    sf = paramiko.Transport((host,port))
    sf.connect(username = username,password = password)
    sftp = paramiko.SFTPClient.from_transport(sf)
    try:
        if os.path.isdir(local):#判斷本地參數是目錄還是文件
            for f in os.listdir(local):#遍歷本地目錄
                sftp.put(os.path.join(local+f),os.path.join(remote+f))#上傳目錄中的文件
        else:
            sftp.put(local,remote)#上傳文件
    except Exception,e:
        print('upload exception:',e)
    sf.close()

def sftp_download(host,port,username,password,local,remote):
    sf = paramiko.Transport((host,port))
    sf.connect(username = username,password = password)
    sftp = paramiko.SFTPClient.from_transport(sf)
    try:
        if os.path.isdir(local):#判斷本地參數是目錄還是文件
            for f in sftp.listdir(remote):#遍歷遠程目錄
                 sftp.get(os.path.join(remote+f),os.path.join(local+f))#下載目錄中文件
        else:
            sftp.get(remote,local)#下載文件
    except Exception,e:
        print('download exception:',e)
    sf.close()

if __name__ == '__main__':
    host = '192.168.1.2'#主機
    port = 22 #端口
    username = 'root' #用戶名
    password = '123456' #密碼
    local = 'F:\\sftptest\\'#本地文件或目錄，與遠程一致，當前為windows目錄格式，window目錄中間需要使用雙斜線
    remote = '/opt/tianpy5/python/test/'#遠程文件或目錄，與本地一致，當前為linux目錄格式
    sftp_upload(host,port,username,password,local,remote)#上傳
    #sftp_download(host,port,username,password,local,remote)#下載