Source code for est.core.process.write

from .base import Process




class WriteXasObject(
    Process,
    input_names=["xas_obj"],
    optional_input_names=[
        "output_file",
        "overwrite",
        "output_group_template",
        "output_group_start_index",
    ],
    output_names=["result"],
):


    @staticmethod
    def definition() -> str:
        return "write XAS object to a file"




    def run(self):
        if not self.inputs.output_file:
            self.outputs.result = None
            return

        xas_obj = self.getXasObject(xas_obj=self.inputs.xas_obj)

        overwrite = self.get_input_value("overwrite", True)
        output_group_template = self.get_input_value("output_group_template", "1.{}")
        output_group_start_index = self.get_input_value("output_group_start_index", 1)

        urls = xas_obj.to_file(
            self.inputs.output_file,
            overwrite=overwrite,
            output_group_template=output_group_template,
            output_group_start_index=output_group_start_index,
        )
        self.outputs.result = urls