"""Module for storing and loading data files."""
import json
import importlib_resources
from ase_notebook import data
from ase_notebook.atoms_convert import deserialize_atoms
[docs]def load_data_file(name, load_json=True, as_binary=False):
"""Load a data file."""
if as_binary:
return importlib_resources.read_binary(data, name)
string = importlib_resources.read_text(data, name)
if load_json and name.endswith(".json"):
return json.loads(string)
return string
[docs]def get_example_atoms(name="pyrite"):
"""Load an example ase.Atoms instance."""
data = load_data_file(f"example_{name}.atoms.json", load_json=False)
return deserialize_atoms(data)