[project @ robey@lag.net-20060822025601-aab3b723b95d3a9c]

add 'flags' param to SFTPHandle and make the default impl avoid calling tell() when in append mode; add proper append-mode support to stub_sftp

paramiko/sftp_handle.py

@@ -35,7 +35,16 @@ class SFTPHandle (object):
     Server implementations can (and should) subclass SFTPHandle to implement
     features of a file handle, like L{stat} or L{chattr}.
     """
-    def __init__(self):
+    def __init__(self, flags=0):
+        """
+        Create a new file handle representing a local file being served over
+        SFTP.  If C{flags} is passed in, it's used to determine if the file
+        is open in append mode.
+        
+        @param flags: optional flags as passed to L{SFTPServerInterface.open}
+        @type flags: int
+        """
+        self.__flags = flags
         self.__name = None
         # only for handles to folders:
         self.__files = { }
@@ -121,6 +130,8 @@ class SFTPHandle (object):
         if writefile is None:
             return SFTP_OP_UNSUPPORTED
         try:
+            # in append mode, don't care about seeking
+            if (self.__flags & os.O_APPEND) == 0:
                 if self.__tell is None:
                     self.__tell = writefile.tell()
                 if offset != self.__tell:
@@ -131,6 +142,7 @@ class SFTPHandle (object):
         except IOError, e:
             self.__tell = None
             return SFTPServer.convert_errno(e.errno)
+        if self.__tell is not None:
             self.__tell += len(data)
         return SFTP_OK
 

tests/stub_sftp.py

@@ -104,8 +104,14 @@ class StubSFTPServer (SFTPServerInterface):
             attr._flags &= ~attr.FLAG_PERMISSIONS
             SFTPServer.set_file_attr(path, attr)
         if flags & os.O_WRONLY:
+            if flags & os.O_APPEND:
+                fstr = 'ab'
+            else:
                 fstr = 'wb'
         elif flags & os.O_RDWR:
+            if flags & os.O_APPEND:
+                fstr = 'a+b'
+            else:
                 fstr = 'r+b'
         else:
             # O_RDONLY (== 0)
@@ -114,7 +120,7 @@ class StubSFTPServer (SFTPServerInterface):
             f = os.fdopen(fd, fstr)
         except OSError, e:
             return SFTPServer.convert_errno(e.errno)
-        fobj = StubSFTPHandle()
+        fobj = StubSFTPHandle(flags)
         fobj.filename = path
         fobj.readfile = f
         fobj.writefile = f