file transfer via sftp added

This commit is contained in:
Volkan Şahin 2016-04-20 18:06:12 +03:00
parent 5e5b446f6b
commit 59cae7567f

View file

@ -0,0 +1,60 @@
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Author: Volkan Şahin <volkansah.in> <bm.volkansahin@gmail.com>
import paramiko
from base.Scope import Scope
class FileTransfer(object):
def __init__(self, hostname, port, username, password, pkey=None):
scope = Scope().getInstance()
self.logger = scope.getLogger()
self.configuration_manager = scope.getConfigurationManager()
self.target_hostname = hostname
self.port = port
self.target_username = username
self.target_password = password
self.p_key = pkey
self.connection = None
self.logger.debug('[FileTransfer] Parameters set up')
def send_file(self, local_path, remote_path):
self.logger.debug('[FileTransfer] Sending file ...')
try:
sftp = paramiko.SFTPClient.from_transport(self.connection)
sftp.put(local_path, remote_path)
self.logger.debug('[FileTransfer] File was sent to %s from %s'.format(local_path,remote_path))
except Exception as e:
self.logger.error('[FileTransfer] A problem occurred while sending file. Exception message: %s'.format(str(e)))
finally:
self.connection.close()
self.logger.debug('[FileTransfer] Connection is closed successfully')
def get_file(self, local_path, remote_path):
self.logger.debug('[FileTransfer] Getting file ...')
try:
sftp = paramiko.SFTPClient.from_transport(self.connection)
sftp.get(remote_path, local_path)
self.logger.debug('[FileTransfer] File was downloaded to %s from %s'.format(local_path,remote_path))
except Exception as e:
self.logger.error('[FileTransfer] A problem occurred while downloading file. Exception message: %s'.format(str(e)))
finally:
self.connection.close()
self.logger.debug('[FileTransfer] Connection is closed successfully')
def connect(self):
self.logger.debug('[FileTransfer] Connecting to %s via %d'.format(self.target_hostname,self.port))
try:
connection = paramiko.Transport((self.target_hostname, self.port))
connection.connect(username=self.target_username, password=self.target_password, pkey=self.p_key)
self.connection = connection
self.logger.debug('[FileTransfer] Connected.')
except Exception as e:
self.logger.error('[FileTransfer] A problem occurred while connecting to %s . Exception message: %s'.format(self.target_hostname, str(e)))