SFTP 工具类

# Author: 富士康民工
# Created: 2024/1/13
# Description: 
"""
Tool class for SFTP
"""
import paramiko


class SftpTools:

    def __init__(self, host, username, password):
        self.host = host
        self.port = 22
        self.username = username
        self.password = password
        self.client = None
        self.sftp_session = None

    def connect(self):
        self.client = paramiko.SSHClient()
        self.client.load_system_host_keys()

        self.client.connect(self.host, self.port, self.username, self.password)
        self.sftp_session = self.client.open_sftp()

        return self.sftp_session

    def close(self):
        if self.sftp_session is not None:
            self.sftp_session.close()
            self.client.close()
            self.sftp_session = None
            self.client = None

    def list_files(self):
        if self.sftp_session is not None:
            files = self.sftp_session.listdir()
            return files
        else:
            return None

    def download(self, remote_path, local_path):
        if self.sftp_session is not None:
            self.sftp_session.get(remote_path, local_path)
        else:
            raise TypeError('Please connect SFTP before download!')

    def upload(self, local_path, remote_path):
        if self.sftp_session is not None:
            self.sftp_session.put(local_path, remote_path)
        else:
            raise TypeError('Please connect SFTP before upload!')


# 使用示例
tools = SftpTools('10.10.10.10', 'uploader', '123456')
# sftp_session = tools.connect()
tools.connect()
files = tools.list_files()
print(files)
tools.close()