Merge branch 'master' of wiscigt.powertheword.com:andrej.studen/labkeySlicerPythonAPI

labkeySlicerPythonExtension/fileIO.py

@@ -4,7 +4,7 @@ import qt
 import ctk
 import json
 import os
-
+import shutil
 
 class fileIO(slicer.ScriptedLoadableModule.ScriptedLoadableModule):
     def __init__(self,parent):
@@ -157,6 +157,11 @@ class fileIOWidget(slicer.ScriptedLoadableModule.ScriptedLoadableModuleWidget):
         copyToRemoteButton.connect('clicked(bool)',self.onCopyToRemoteButtonClicked)
         fileDialogFormLayout.addRow(copyToRemoteButton)
 
+        copyToLocalButton=qt.QPushButton("Remote -> Local")
+        copyToLocalButton.toolTip="Remote -> Local"
+        copyToLocalButton.connect('clicked(bool)',self.onCopyToLocalButtonClicked)
+        fileDialogFormLayout.addRow(copyToLocalButton)
+
         removeRemoteButton=qt.QPushButton("Remove remote")
         removeRemoteButton.toolTip="Remove Remote"
         removeRemoteButton.connect('clicked(bool)',self.onRemoveRemoteButtonClicked)
@@ -213,9 +218,40 @@ class fileIOWidget(slicer.ScriptedLoadableModule.ScriptedLoadableModuleWidget):
             localfile=os.path.join(localPath,f)
             self.copyLocalFileToRemote(localfile,remotePath)
 
+    def copyRemoteToLocal(self,localPath,remotePath):
+        ok,files=self.network.listRemoteDir(remotePath)
+        if not ok:
+            print 'Troubles getting remote dir content for {}'.format(remotePath)
+            return
+        #remove trailing slash
+        for f in files:
+            if f[-1]=='/':
+                f=f[:-1]
+
+            bf=f[f.rfind('/')+1:]
+            dirUrl=self.network.GetLabkeyWebdavUrl()+"/"+f
+            if self.network.isDir(dirUrl):
+                lp=os.path.join(localPath,bf)
+                if not os.path.isdir(lp):
+                    os.mkdir(lp)
+                    print 'Creating {}'.format(lp)
+                print 'Copying directory {} to {}'.format(f,lp)
+                self.copyRemoteToLocal(lp,f)
+            else:
+                rf=self.network.readFile(f)
+                fout=os.path.join(localPath,bf)
+                print 'Copying file {} to {}'.format(f,fout)
+                with open (fout, 'w') as fd:
+                    rf.seek (0)
+                    shutil.copyfileobj (rf, fd)
+                fd.close()
+
     def onCopyToRemoteButtonClicked(self):
         self.copyLocalFileToRemote(self.localPath.text,self.remotePath.text)
 
+    def onCopyToLocalButtonClicked(self):
+        self.copyRemoteToLocal(self.localPath.text,self.remotePath.text)
+
     def onRemoveRemoteButtonClicked(self):
         remotePath=self.remotePath.text
         if self.network.isRemoteDir(remotePath):