KittyPatch/kitty
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add a tell() method to PatchFile to track how much has been written

Browse Source
This commit is contained in:
Kovid Goyal 2021-10-03 10:18:28 +05:30
parent f6e518d2f9
commit 3a373a200c
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
kittens/transfer/librsync.py
View File

@ -120,6 +120,11 @@ class PatchFile(StreamingJob):
job = begin_patch(self.read_from_src) job = begin_patch(self.read_from_src)
super().__init__(job, output_buf_size=4 * IO_BUFFER_SIZE) super().__init__(job, output_buf_size=4 * IO_BUFFER_SIZE)
def tell(self) -> int:
if self.dest_file.closed:
return os.path.getsize(self.src_file.name if self.overwrite_src else self.dest_file.name)
return self.dest_file.tell()
def read_from_src(self, b: memoryview, pos: int) -> int: def read_from_src(self, b: memoryview, pos: int) -> int:
self.src_file.seek(pos) self.src_file.seek(pos)
return self.src_file.readinto(b) # type: ignore return self.src_file.readinto(b) # type: ignore