Renamed internal func and removed an iteration over self.blocks

This commit is contained in:
Sybren A. Stüvel 2018-02-23 11:43:51 +01:00
parent 8d32ab513f
commit 74ef3877c6

View File

@ -102,9 +102,9 @@ class BlendFile:
self.sdna_index_from_id = {} self.sdna_index_from_id = {}
self.block_from_addr = {} self.block_from_addr = {}
self.load_dna1_block() self._load_blocks()
def load_dna1_block(self): def _load_blocks(self):
"""Read the blend file to load its DNA structure to memory.""" """Read the blend file to load its DNA structure to memory."""
while True: while True:
block = BlendFileBlock(self) block = BlendFileBlock(self)
@ -118,14 +118,12 @@ class BlendFile:
self.blocks.append(block) self.blocks.append(block)
self.code_index[block.code].append(block) self.code_index[block.code].append(block)
self.block_from_addr[block.addr_old] = block
if not self.structs: if not self.structs:
raise exceptions.NoDNA1Block("No DNA1 block in file, not a valid .blend file", raise exceptions.NoDNA1Block("No DNA1 block in file, not a valid .blend file",
self.filepath) self.filepath)
self.block_from_addr = {block.addr_old: block for block in self.blocks
if block.code != b'ENDB'}
def __repr__(self): def __repr__(self):
clsname = self.__class__.__qualname__ clsname = self.__class__.__qualname__
if self.filepath == self.raw_filepath: if self.filepath == self.raw_filepath: