mirror
/
borg
mirror of https://github.com/borgbackup/borg.git
1
0
Fork 0
Code Issues Releases Wiki Activity

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

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

14
src/borg/hashindex.pyx
View File

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