relinkall.py

# coding=utf-8
from earthdiagnostics.diagnostic import Diagnostic, DiagnosticOption, DiagnosticDomainOption, DiagnosticBoolOption
from earthdiagnostics.modelingrealm import ModelingRealm


class RelinkAll(Diagnostic):
    """
    Recreates the links for the variable specified

    :original author: Javier Vegas-Regidor<javier.vegas@bsc.es>

    :created: September 2016

    :param data_manager: data management object
    :type data_manager: DataManager
    :param startdate: startdate
    :type startdate: str
    """

    alias = 'relinkall'
    "Diagnostic alias for the configuration file"

    def __init__(self, data_manager, startdate):
        Diagnostic.__init__(self, data_manager)
        self.startdate = startdate

    def __str__(self):
        return 'Relink all output Startdate: {0}'.format(self.startdate)

    def __eq__(self, other):
        return self.startdate == other.startdate

    @classmethod
    def generate_jobs(cls, diags, options):
        """
            Creates a job for each chunk to compute the diagnostic

            :param diags: Diagnostics manager class
            :type diags: Diags
            :param options: variable, domain, move_old=False
            :type options: list[str]
            :return:
            """
        if len(options) > 1:
            raise Exception('The Relink All diagnostic has no options')
        job_list = list()
        for startdate in diags.config.experiment.startdates:
            job_list.append(RelinkAll(diags.data_manager, startdate))
        return job_list

    def compute(self):
        """
        Runs the diagnostic
        """
        self.data_manager.create_links(self.startdate)