mirror of
https://github.com/Pardus-LiderAhenk/ahenk
synced 2024-11-22 15:32:19 +03:00
file transfer via sftp added
This commit is contained in:
parent
5e5b446f6b
commit
59cae7567f
1 changed files with 60 additions and 0 deletions
60
opt/ahenk/base/messaging/ssh_file_transfer.py
Normal file
60
opt/ahenk/base/messaging/ssh_file_transfer.py
Normal 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)))
|
||||
|
Loading…
Reference in a new issue