Ahenk/usr/share/ahenk/base/file/ssh_file_transfer.py

100 lines
3.9 KiB
Python
Raw Normal View History

2019-05-17 09:53:14 +03:00
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Author: Volkan Şahin <volkansah.in> <bm.volkansahin@gmail.com>
from base.util.util import Util
from base.system.system import System
from base.scope import Scope
import paramiko
import logging
class Ssh(object):
def __init__(self, parameter_map):
scope = Scope().get_instance()
self.logger = scope.get_logger()
self.configuration_manager = scope.get_configuration_manager()
logging.getLogger("paramiko").setLevel(logging.INFO)
try:
self.target_hostname = parameter_map['host']
self.target_port = parameter_map['port']
self.target_username = parameter_map['username']
self.target_path = parameter_map['path']
self.target_password = None
self.p_key = None
if Util.has_attr_json(parameter_map, 'password'):
self.target_password = parameter_map['password']
else:
self.p_key = parameter_map['pkey']
except Exception as e:
self.logger.error(
'A problem occurred while parsing ssh connection parameters. Error Message: {0}'.format(str(e)))
self.connection = None
self.logger.debug('Parameters set up')
def send_file(self, local_path, md5):
self.logger.debug(' {0} is sending to {1}'.format(local_path, self.target_path + md5))
try:
sftp = paramiko.SFTPClient.from_transport(self.connection)
try:
sftp.chdir(self.target_path) # Test if remote_path exists
except IOError:
sftp.mkdir(self.target_path) # Create remote_path
sftp.chdir(self.target_path)
sftp.put(local_path, self.target_path + md5)
self.logger.debug('File was sent to {0} from {1}'.format(local_path, self.target_path))
return True
except Exception as e:
self.logger.error('A problem occurred while sending file. Exception message: {0}'.format(str(e)))
return False
def get_file(self, local_path=None, file_name=None):
self.logger.debug('Getting file ...')
try:
tmp_file_name = str(Util.generate_uuid())
local_full_path = System.Ahenk.received_dir_path() + tmp_file_name
sftp = paramiko.SFTPClient.from_transport(self.connection)
sftp.get(self.target_path, local_full_path)
if local_path is None:
receive_path = System.Ahenk.received_dir_path()
else:
receive_path = local_path
if file_name is not None:
f_name = file_name
else:
f_name = str(Util.get_md5_file(local_full_path))
Util.rename_file(local_full_path, receive_path + f_name)
self.logger.debug('File was downloaded to {0} from {1}'.format(receive_path, self.target_path))
except Exception as e:
self.logger.warning('A problem occurred while downloading file. Exception message: {0}'.format(str(e)))
raise
return f_name
def connect(self):
self.logger.debug('Connecting to {0} via {1}'.format(self.target_hostname, self.target_port))
try:
connection = paramiko.Transport(self.target_hostname, int(self.target_port))
connection.connect(username=self.target_username, password=self.target_password, pkey=self.p_key)
self.connection = connection
self.logger.debug('Connected.')
except Exception as e:
self.logger.error(
'A problem occurred while connecting to {0} . Exception message: {1}'.format(
self.target_hostname, str(e)))
def disconnect(self):
self.connection.close()
self.logger.debug('Connection is closed.')
# TODO
pass
def is_connected(self):
# TODO
pass