mirror/borg
1
0
Fork 0
mirror of https://github.com/borgbackup/borg.git synced 2025-02-22 06:01:54 +00:00
Code Issues Releases Wiki Activity

hashindex: read/write: accept file-like objects for path

Browse source
This commit is contained in:
Marian Beermann 2017-05-25 13:43:04 +02:00
parent 39051ac5f1
commit 06cf15cc6d
1 changed files with 10 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/borg/hashindex.pyx
View file

@ -67,8 +67,11 @@ cdef class IndexBase:
def __cinit__(self, capacity=0, path=None, key_size=32): def __cinit__(self, capacity=0, path=None, key_size=32):
self.key_size = key_size self.key_size = key_size
if path: if path:
if isinstance(path, (str, bytes)):
with open(path, 'rb') as fd: with open(path, 'rb') as fd:
self.index = hashindex_read(fd) self.index = hashindex_read(fd)
else:
self.index = hashindex_read(path)
assert self.index, 'hashindex_read() returned NULL with no exception set' assert self.index, 'hashindex_read() returned NULL with no exception set'
else: else:
self.index = hashindex_init(capacity, self.key_size, self.value_size) self.index = hashindex_init(capacity, self.key_size, self.value_size)
@ -84,8 +87,11 @@ cdef class IndexBase:
return cls(path=path) return cls(path=path)
def write(self, path): def write(self, path):
if isinstance(path, (str, bytes)):
with open(path, 'wb') as fd: with open(path, 'wb') as fd:
hashindex_write(self.index, fd) hashindex_write(self.index, fd)
else:
hashindex_write(self.index, path)
def clear(self): def clear(self):
hashindex_free(self.index) hashindex_free(self.index)