Added option to folllow symlinks to ftpfs, patch from Andrew

fs/ftpfs.py

@@ -337,6 +337,7 @@ class FTPListDataParser(object):
             i = 0
             while (i + 3) < len(result.name):
                 if result.name[i:i+4] == ' -> ':
+                    result.target = result.name[i+4:]
                     result.name = result.name[:i]
                     break
                 i += 1
@@ -891,7 +892,7 @@ class FTPFS(FS):
               'file.read_and_write' : False,
               }
 
-    def __init__(self, host='', user='', passwd='', acct='', timeout=_GLOBAL_DEFAULT_TIMEOUT, port=21, dircache=True):
+    def __init__(self, host='', user='', passwd='', acct='', timeout=_GLOBAL_DEFAULT_TIMEOUT, port=21, dircache=True, follow_symlinks=False):
         """Connect to a FTP server.
 
         :param host: Host to connect to
@@ -917,6 +918,7 @@ class FTPFS(FS):
         self.timeout = timeout
         self.default_timeout = timeout is _GLOBAL_DEFAULT_TIMEOUT
         self.use_dircache = dircache
+        self.follow_symlinks = follow_symlinks
 
         self.use_mlst = False
         self._lock = threading.RLock()
@@ -1018,6 +1020,30 @@ class FTPFS(FS):
             pass
         self.dircache[path] = dirlist
 
+        def is_symlink(info):
+            return info['try_retr'] and info['try_cwd'] and info.has_key('target')
+
+        def resolve_symlink(linkpath):
+            linkinfo = self.getinfo(linkpath)
+            if not linkinfo.has_key('resolved'):
+                linkinfo['resolved'] = linkpath
+            if is_symlink(linkinfo):
+                target = linkinfo['target']
+                base, fname = pathsplit(linkpath)
+                return resolve_symlink(pathjoin(base, target))
+            else:
+                return linkinfo
+
+        if self.follow_symlinks:
+            for name in dirlist:
+                if is_symlink(dirlist[name]):
+                    target = dirlist[name]['target']
+                    linkinfo = resolve_symlink(pathjoin(path, target))
+                    for key in linkinfo:
+                        if key != 'name':
+                            dirlist[name][key] = linkinfo[key]
+                    del dirlist[name]['target']
+
         return dirlist
 
     @synchronize

fs/opener.py

@@ -434,7 +434,7 @@ examples:
         dirpath, resourcepath = pathsplit(path)
         url = netloc
 
-        ftpfs = FTPFS(url, user=username or '', passwd=password or '')
+        ftpfs = FTPFS(url, user=username or '', passwd=password or '', follow_symlinks=(fs_name_params == "symlinks"))
         ftpfs.cache_hint(True)
 
         if create_dir and path:

fs/tests/test_ftpfs.py

@@ -16,7 +16,9 @@ from six import PY3
 
 
 try:
-    from pyftpdlib import ftpserver
+    from pyftpdlib.authorizers import DummyAuthorizer
+    from pyftpdlib.handlers import FTPHandler
+    from pyftpdlib.servers import FTPServer
 except ImportError:
     if not PY3:
         raise ImportError("Requires pyftpdlib <http://code.google.com/p/pyftpdlib/>")
@@ -89,21 +91,21 @@ if __name__ == "__main__":
 
     # Run an ftp server that exposes a given directory
     import sys
-    authorizer = ftpserver.DummyAuthorizer()    
+    authorizer = DummyAuthorizer()
     authorizer.add_user("user", "12345", sys.argv[1], perm="elradfmw")
     authorizer.add_anonymous(sys.argv[1])
 
-    def nolog(*args):
-        pass
-    ftpserver.log = nolog
-    ftpserver.logline = nolog
+    #def nolog(*args):
+    #    pass
+    #ftpserver.log = nolog
+    #ftpserver.logline = nolog
 
-    handler = ftpserver.FTPHandler
+    handler = FTPHandler
     handler.authorizer = authorizer
     address = ("127.0.0.1", int(sys.argv[2]))
     #print address
 
-    ftpd = ftpserver.FTPServer(address, handler)
+    ftpd = FTPServer(address, handler)
 
     sys.stdout.write('serving\n')
     sys.stdout.flush()