31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
#!/usr/bin/env python
|
|
# vim:fileencoding=utf-8
|
|
# License: GPLv3 Copyright: 2021, Kovid Goyal <kovid at kovidgoyal.net>
|
|
|
|
import os
|
|
from typing import Iterator
|
|
|
|
from .rsync import IO_BUFFER_SIZE, RsyncError, begin_create_signature, iter_job
|
|
|
|
|
|
def signature_of_file(path: str) -> Iterator[bytes]:
|
|
with open(path, 'rb') as f:
|
|
f.seek(0, os.SEEK_END)
|
|
fsz = f.tell()
|
|
job = begin_create_signature(fsz)
|
|
f.seek(0)
|
|
finished = False
|
|
prev_unused_input = b''
|
|
while not finished:
|
|
input_data = f.read(IO_BUFFER_SIZE)
|
|
no_more_data = not input_data
|
|
if prev_unused_input:
|
|
input_data = prev_unused_input + input_data
|
|
prev_unused_input = b''
|
|
output, finished, sz_of_unused_input = iter_job(job, input_data)
|
|
if sz_of_unused_input > 0 and not finished:
|
|
if no_more_data:
|
|
raise RsyncError(f"{sz_of_unused_input} bytes of input data were not used")
|
|
prev_unused_input = input_data[-sz_of_unused_input:]
|
|
yield output
|