Support context manager for opening blend file

This commit is contained in:
2015-06-09 03:08:18 +10:00
parent 88eefcf6bc
commit 7d7fede5b2
3 changed files with 57 additions and 51 deletions

View File

@@ -145,6 +145,12 @@ class BlendFile:
# cache (could lazy init, incase we never use?)
self.block_from_offset = {block.addr_old: block for block in self.blocks if block.code != b'ENDB'}
def __enter__(self):
return self
def __exit__(self, type, value, traceback):
self.close()
def find_blocks_from_code(self, code):
assert(type(code) == bytes)
if code not in self.code_index: