#Bandaid for hardcoding of local files

## Be sure to install the paramiko package. 'pip install paramiko'

ip = ~~ip address~~
un = ~~username~~
pw = ~~password~~
localdir = "/home/user/sftpfiles/"  ## change to your local storage directory

import paramiko

class sftp_client:  #get local directory of a pulled file with "local_filename = ship.files[remote_filename]"

    def __init__(self, ip, un, pw, localdir):

        self.ip = ip
        self.un = un
        self.pw = pw
        self.localdir = localdir

        self.ssh = paramiko.SSHClient()

        self.ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        self.ssh.load_system_host_keys()

        self.ssh.connect(ip,username=un, password=pw, key_filename = key_file)

        self.sftp = self.ssh.open_sftp()
        self.files = {}

    def get_file(self,filename):

        localpath = localdir+filename.split('/')[-1]
        self.sftp.get(filename, localpath)
        self.files[filename]=localpath


ship = sftp_client(ip,un,pw,localdir)