Source code for aiida_bands_inspect.parsers.plot

# -*- coding: utf-8 -*-

# © 2017-2019, ETH Zurich, Institut für Theoretische Physik
# Author: Dominik Gresch <greschd@gmx.ch>

from aiida.plugins import DataFactory
from aiida.parsers.parser import Parser

from ..calculations.plot import PlotCalculation

__all__ = ('PlotParser', )


class PlotParser(Parser):
    """
    Parse ``bands-inspect plot_bands`` output to a 'singlefile' data.

    Returns
    -------
    plot : aiida.orm.nodes.data.singlefile.SinglefileData
        File containing the generated plot.
    """
    def parse(self, **kwargs):  # pylint: disable=inconsistent-return-statements
        try:
            out_folder = self.retrieved
        except KeyError:
            return self.exit_codes.ERROR_NO_RETRIEVED_FOLDER

        try:
            with out_folder.open(
                PlotCalculation._OUTPUT_FILE_NAME,  # pylint: disable=protected-access
                'rb'
            ) as handle:
                self.out('plot', DataFactory('singlefile')(file=handle))
        except IOError:
            return self.exit_codes.ERROR_OUTPUT_FILE_MISSING