Python - transferring files with SFTP using paramiko

sftp.py

#!/usr/bin/env python3
#
# transferring files with SFTP
# https://www.paramiko.org

import paramiko


class SFTP:
    def __init__(self, host, user, password, port=22):
        self.host = host
        self.user = user
        self.password = password
        self.port = port
        self.sftp = None
        self._ssh = paramiko.SSHClient()
        self._ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())

    def __enter__(self):
        print(f"Connecting to server: {self.host}")
        self._ssh.connect(
            hostname=self.host,
            port=self.port,
            username=self.user,
            password=self.password,
        )
        self.sftp = self._ssh.open_sftp()
        return self.sftp

    def __exit__(self, exc_type, exc_value, traceback):
        if self.sftp:
            self.sftp.close()
        if self._ssh:
            self._ssh.close()
        print("Connection closed.")


# example usage
if __name__ == "__main__":
    with SFTP("some-ftp-site.com", "username", "secret") as sftp:
        for file in sftp.listdir():
            print(f"  - {file}")
        sftp.get("example.txt")
        sftp.put("example.txt")