INNER CODE UNIT · Python
load_data
mitre-attack/attack-scripts · scripts/diff_stix.py:176
def load_data(self):
"""
Load data from files into data dict.
"""
if self.verbose:
pbar = tqdm(total=len(self.types) * len(self.domains), desc="loading data", bar_format="{l_bar}{bar}| [{elapsed}<{remaining}, {rate_fmt}{postfix}]")
for obj_type in self.types:
for domain in self.domains:
# handle data loaded from either a directory or the TAXII server
def load_datastore(data_store):
raw_data = list(chain.from_iterable(
data_store.query(f) for f in attackTypeToStixFilter[obj_type]
))
raw_data = self.deep_copy_stix(raw_data)
id_to_obj = {item['id']: item for item in raw_data}
return {
"id_to_obj": id_to_obj,