createDirectory an removeDirectory operations for sFTP

/reviewed-on nexedi/erp5!756

bt5/erp5_web_service/DocumentTemplateItem/portal_components/document.erp5.FTPConnector.py

@@ -31,6 +31,7 @@ from AccessControl import ClassSecurityInfo
 from Products.ERP5Type import Permissions, PropertySheet
 from Products.ERP5Type.XMLObject import XMLObject
 
+
 class FTPConnector(XMLObject):
   # CMF Type Definition
   meta_type = 'FTP Connector'
@@ -111,3 +112,25 @@ class FTPConnector(XMLObject):
         conn.writeFile(remotepath, '%s' % filename, data, confirm=confirm)
     finally:
       conn.logout()
+
+  def createDirectory(self, path, mode=0777):
+    """Create a directory `path`, with file mode `mode`.
+
+    The directory is created immediatly, even if transaction is aborted.
+    """
+    conn = self.getConnection()
+    try:
+      conn.createDirectory(path, mode)
+    finally:
+      conn.logout()
+
+  def removeDirectory(self, path):
+    """Create a directory `path`, with file mode `mode`.
+
+    The directory is removed immediatly, even if transaction is aborted.
+    """
+    conn = self.getConnection()
+    try:
+      conn.removeDirectory(path)
+    finally:
+      conn.logout()
\ No newline at end of file

bt5/erp5_web_service/TestTemplateItem/portal_components/test.erp5.testFTPConnection.py

@@ -92,6 +92,12 @@ class TestSFTPConnection(ERP5TypeTestCase):
           self.connection.listFiles(".", sort_on="st_size")
       )
 
+    def test_create_remove_directory(self):
+      self.connection.createDirectory("foo")
+      self.assertItemsEqual(["foo"], self.connection.listFiles("."))
+      self.connection.removeDirectory("foo")
+      self.assertItemsEqual([], self.connection.listFiles("."))
+
   else:
     def test_no_SFTP_URL_in_environ(self):
       raise unittest.SkipTest(

product/ERP5Type/ConnectionPlugin/SFTPConnection.py

@@ -171,6 +171,16 @@ class SFTPConnection:
       raise SFTPError('%s while trying to rename "%s" to "%s" on %s.' % \
                      (str(msg), old_path, new_path, self.url))
 
+  def createDirectory(self, path, mode=0777):
+    """Create a directory `path` with mode `mode`.
+    """
+    return self.conn.mkdir(path, mode)
+
+  def removeDirectory(self, path):
+    """Remove directory `path`.
+    """
+    return self.conn.rmdir(path)
+
   def logout(self):
     """Logout of the SFTP Server"""
     self.conn.close()