python - 按字节数划分文件

Question

我正在制作一个文件共享程序，当我发送一个文件时，我只想发送 1024 个字节，然后发送接下来的 1024 个字节，依此类推。我使用的是 scapy 而不是套接字，所以我需要手动执行此操作。如何将文件分成每个 1024 字节的 block ？

Answer 1

您可以编写一个简单的循环:

while True:
    block = f.read(1024)
    if not block:
        break
    ... do stuff with block ...

每次调用read都会读取数据并前进文件位置，因此每次都会获得一个新 block 。或者，您可以更聪明一点，使用两个参数 iter 来制作一个执行相同逻辑的 Python 生成器，但与 for 循环一起使用，看起来更干净一些:

from functools import partial

for block in iter(partial(f.read, 1024), b''):  # Change b'' to '' for text mode file
    ... do stuff with block ...