added progressbar

Author: jhn-nt

docksing.py

@@ -18,6 +18,8 @@
 import time
 import cmd
 
+def progress(filename, size, sent):
+    sys.stdout.write("%s\'s progress: %.2f%%   \r" % (filename, float(sent)/float(size)*100) )
 
 @dataclass
 class CLICompose:
@@ -200,7 +202,6 @@ def push(self,tag:str,remotedir:str):
             tag (str): Name of the target image tag.
             remotedir (str): Absolute path of working directory on host.
         """
-        
         image=self.docker.images.get(tag)
         iid=image.short_id.split(":")[1]
         with io.BytesIO() as file:
@@ -209,7 +210,7 @@ def push(self,tag:str,remotedir:str):
             file.seek(0)
 
             if self.ssh:
-                with SCPClient(self.ssh.get_transport()) as scp:
+                with SCPClient(self.ssh.get_transport(),progress=progress) as scp:
                     scp.putfo(file,f"{remotedir}/{iid}.tar")
             else:
                 with open(f"{remotedir}/{iid}.tar","wb") as f:
@@ -240,7 +241,7 @@ def map_remote_volume(self,remote_dir:str,local_dir:str,container_dir:str,send_p
             if send_payload and Path(local_dir).is_dir():
                 # If the mapping points to an existing local folder,
                 # we copy its content in remote_dir/local_host
-                with SCPClient(self.ssh.get_transport()) as scp:
+                with SCPClient(self.ssh.get_transport(),progress=progress) as scp:
                     scp.put(
                         local_dir,
                         remote_path=remote_dir,

setup.py

@@ -5,12 +5,12 @@
 
 setup(
     name="docksing",
-    version="0.2.35",
+    version="0.2.36",
     description="CLI Utility for deployment of containerized jobs on SLURM HPC ",
     long_description=long_description,
     long_description_content_type="text/markdown",
     author="G. Angelotti",
-    author_email="giovanni.angelotti@idsia.ch",
+    author_emairel="giovanni.angelotti@idsia.ch",
     py_modules=["docksing"],
     install_requires=[
         "paramiko==3.4.0",