def mkdir(self, path):
try:
dst_path = self._get_realpath(path)
self._client.mkdir(dst_path)
except Exception as e:
raise IrmaSFTPError("{0}".format(e))
return
def list(self, path):
""" list remote directory <path>"""
try:
dst_path = self._get_realpath(path)
return self._client.listdir(dst_path)
except Exception as e:
raise IrmaSFTPError("{0}".format(e))
def rename(self, oldpath, newpath):
try:
old_realpath = self._get_realpath(oldpath)
new_realpath = self._get_realpath(newpath)
self._client.rename(old_realpath, new_realpath)
except Exception as e:
raise IrmaSFTPError("{0}".format(e))
def delete(self, path, filename):
""" Delete <filename> into directory <path>"""
try:
dstpath = self._get_realpath(path)
full_dstpath = self._tweaked_join(dstpath, filename)
self._client.remove(full_dstpath)
except Exception as e:
raise IrmaSFTPError("{0}".format(e))
def is_file(self, path, filename):
try:
dstpath = self._get_realpath(path)
full_dstpath = self._tweaked_join(dstpath, filename)
st = self._client.stat(full_dstpath)
return not stat.S_ISDIR(st.st_mode)
except Exception as e:
reason = "{0} [{1}]".format(e, path)
raise IrmaSFTPError(reason)
def upload_fobj(self, path, fobj):
""" Upload <data> to remote directory <path>"""
try:
dstname = self._hash(fobj)
path = self._tweaked_join(path, dstname)
dstpath = self._get_realpath(path)
self._client.putfo(fobj, dstpath)
return dstname
except Exception as e:
raise IrmaSFTPError("{0}".format(e))
def download_fobj(self, path, remotename, fobj):
""" returns <remotename> found in <path>"""
# self._client.getfo(fobj, dstpath)
try:
dstpath = self._get_realpath(path)
full_dstpath = self._tweaked_join(dstpath, remotename)
self._client.getfo(full_dstpath, fobj)
# remotename is hashvalue of data
self._check_hash(remotename, fobj)
except Exception as e:
raise IrmaSFTPError("{0}".format(e))
def _connect(self):
if self._conn is not None:
log.warn("Already connected to sftp server")
return
try:
self._conn = Transport((self._host, self._port))
self._conn.window_size = pow(2, 27)
self._conn.packetizer.REKEY_BYTES = pow(2, 32)
self._conn.packetizer.REKEY_PACKETS = pow(2, 32)
self._conn.connect(username=self._user, password=self._passwd)
self._client = SFTPClient.from_transport(self._conn)
except Exception as e:
raise IrmaSFTPError("{0}".format(e))
def deletepath(self, path, deleteParent=False):
# recursively delete all subdirs and files
try:
for f in self.list(path):
if self.is_file(path, f):
self.delete(path, f)
else:
self.deletepath(self._tweaked_join(path, f),
deleteParent=True)
if deleteParent:
dstpath = self._get_realpath(path)
self._client.rmdir(dstpath)
except Exception as e:
reason = "{0} [{1}]".format(str(e), path)
raise IrmaSFTPError(reason)
def download_fobj(self, path, remotename, fobj):
""" returns <remotename> found in <path>
if hash check is on:
remotename should be a <sha256 val> and
will be compared after transfer
"""
try:
dstpath = self._get_realpath(path)
full_dstpath = self._tweaked_join(dstpath, remotename)
self._client.getfo(full_dstpath, fobj)
if self.hash_check:
# remotename is hashvalue of data
self._check_hash(remotename, fobj)
except Exception as e:
raise IrmaSFTPError("{0}".format(e))
def upload_fobj(self, path, fobj):
""" Upload <data> to remote destination
if hash check is on:
remote dest is <path>/<sha256 value>
if hash check is off:
remote dest is <path>
"""
try:
if self.hash_check:
dstname = self._hash(fobj)
path = self._tweaked_join(path, dstname)
dstpath = self._get_realpath(path)
self._client.putfo(fobj, dstpath)
if self.hash_check:
return dstname
except Exception as e:
raise IrmaSFTPError("{0}".format(e))
