pandaharvester/harvesterzipper/ssh_zipper.py

0001 from pandaharvester.harvestercore import core_utils
0002
0003 from .base_zipper import BaseZipper
0004
0005 # logger
0006 _logger = core_utils.setup_logger("ssh_zipper")
0007
0008
0009 # ssh plugin for zipper
0010 class SshZipper(BaseZipper):
0011     """SSH zipper plugin that zips output files via SSH connection."""
0012     # constructor
0013     def __init__(self, **kwarg):
0014         BaseZipper.__init__(self, **kwarg)
0015
0016     # zip output files
0017     def zip_output(self, jobspec):
0018         tmpLog = self.make_logger(_logger, f"PandaID={jobspec.PandaID}", method_name="zip_output")
0019         return self.ssh_zip_output(jobspec, tmpLog)
0020
0021     # asynchronous zip output
0022     def async_zip_output(self, jobspec):
0023         tmpLog = self.make_logger(_logger, f"PandaID={jobspec.PandaID}", method_name="zip_output")
0024         # not really asynchronous as two staged zipping is not implemented in this plugin
0025         return self.ssh_zip_output(jobspec, tmpLog)
0026
0027     # post zipping
0028     def post_zip_output(self, jobspec):
0029         return True, ""