Question

如何以写模式打开FTP服务器上的文件？我知道我可以直接写入/创建文件（当我有数据时），但是我想先打开它进行写入，然后再像使用contextmanager在本地进行写入一样先写入它。

原因是，我想创建一个接口，该接口将具有用于传输协议服务器的统一方法。特别是SFTP和FTP。

因此，使用SFTP很容易（使用paramiko）：

def open(sftp, fname, mode='r'):
    return sftp.open(fname, mode=mode)

现在我可以这样做：

with open(sftp, 'some_file.txt', 'w') as f:
    f.write(data)

然后我可以阅读所写的内容

with open(sftp, 'some_file.txt', 'r') as f:
    print(f.read().decode('utf-8'))

如何为FTP（使用ftplib）执行相同的实现？

FTP的阅读部分，我能够实现，并且可以像SFTP一样以阅读模式打开文件。但是，如何在写入模式下打开它？ ftplib方法storbinary要求“立即”提供数据。我的意思是我应该已经通过open方法传递了我想写入的数据（但是那样会破坏统一方法的目的）？

import io

def open(ftp, filename, mode='r'):
    """Open a file on FTP server."""
    def handle_buffer(buffer_data):
        bio.write(buffer_data)

    # Reading implementation
    if mode == 'r':
        bio = io.BytesIO()
        ftp.retrbinary(
            'RETR %s' % filename, callback=handle_buffer)
        bio.seek(0)
        return bio
    # Writing implementation.
    if mode == 'w':
        # how to open in write mode?

更新

假设我们已经在FTP中立即编写实现：

bio = io.BytesIO
# Write some data
data = csv.writer(bio)
data.writerows(data_to_export)
bio.seek(0)
# Store. So it looks like storbinary does not open file in w mode, it does everything in one go?
ftp.storbinary("STOR " + file_name, sio)

所以问题是我如何才能将写入数据与仅以写入模式打开文件分开。甚至可以使用ftplib吗？

Answer 1

因此，经过一番努力，我得以完成这项工作。解决方案是在读取模式（必须重新实现读取模式，因为它仅适用于纯文件读取，但是如果让我说尝试使用csv读取器）失败时，则为open方法实现自定义上下文管理器。在写入模式下。

对于读取模式，我选择使用tempfile，因为使用其他方法，我无法使用其他读取器（普通文件读取器，csv阅读器等）正确读取数据。虽然在读取模式下使用打开的tempfile时，一切都能按预期进行。

对于写模式，我能够利用内存缓冲区-> io.BytesIO。因此，编写时不必使用tempfile。

import tempfile


class OpenRead(object):

    def _open_tempfile(self):
        self.tfile = tempfile.NamedTemporaryFile()
        # Write data on tempfile.
        self.ftp.retrbinary(
            'RETR %s' % self.filename, self.tfile.write)
        # Get back to start of file, so it would be possible to
        # read it.
        self.tfile.seek(0)
        return open(self.tfile.name, 'r')

    def __init__(self, ftp, filename):
        self.ftp = ftp
        self.filename = filename
        self.tfile = None

    def __enter__(self):
        return self._open_tempfile()

    def __exit__(self, exception_type, exception_value, traceback):
        # Remove temporary file.
        self.tfile.close()

class OpenWrite(object):
    def __init__(self, ftp, filename):
        self.ftp = ftp
        self.filename = filename
        self.data = ''

    def __enter__(self):
        return self

    def __exit__(self, exception_type, exception_value, traceback):
        bio = io.BytesIO()
        if isinstance(self.data, six.string_types):
            self.data = self.data.encode()
        bio.write(self.data)
        bio.seek(0)
        res = self.ftp.storbinary('STOR %s' % self.filename, bio)
        bio.close()
        return res

    def write(self, data):
        self.data += data

def open(ftp, filename, mode='r'):
    """Open a file on FTP server."""
    if mode == 'r':
        return OpenRead(ftp, filename)
    if mode == 'w':
        return OpenWrite(ftp, filename)

P.S。如果没有上下文管理器，这可能无法正常工作，但是就我而言，这是个好的解决方案。如果有人能更好地实施，欢迎与他们分享。

更新

决定使用ftputil软件包而不是标准ftplib。因此，不需要进行所有黑客攻击，因为ftputil会处理它，并且实际上使用了与paramiko相同的命名方法，它们执行相同的操作，因此统一协议的使用要容易得多。

Python ftplib：以写模式打开文件吗？

1 个答案: