""" Class defining a production step """
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

__RCSID__ = "c59810311 (2021-10-01 10:44:25 +0200) Chris Burr <christopher.burr@cern.ch>"

import json

from DIRAC import S_OK, S_ERROR


class ProductionStep(object):

    """Define the Production Step object"""

    def __init__(self, **kwargs):
        """Simple constructor"""
        # Default values for transformation step parameters
        self.Name = ""
        self.Description = "description"
        self.LongDescription = "longDescription"
        self.Type = "MCSimulation"
        self.Plugin = "Standard"
        self.AgentType = "Manual"
        self.FileMask = ""
        #########################################
        self.ParentStep = None
        self.Inputquery = None
        self.Outputquery = None
        self.GroupSize = 1
        self.Body = "body"

    def getAsDict(self):
        """It returns the Step description as a dictionary"""
        prodStepDict = {}
        prodStepDict["name"] = self.Name
        prodStepDict["parentStep"] = []
        # check the ParentStep format
        if self.ParentStep:
            if isinstance(self.ParentStep, list):
                prodStepDict["parentStep"] = []
                for parentStep in self.ParentStep:  # pylint: disable=not-an-iterable
                    if not parentStep.Name:
                        return S_ERROR("Parent Step does not exist")
                    prodStepDict["parentStep"].append(parentStep.Name)
            elif isinstance(self.ParentStep, ProductionStep):
                if not self.ParentStep.Name:
                    return S_ERROR("Parent Step does not exist")
                prodStepDict["parentStep"] = [self.ParentStep.Name]
            else:
                return S_ERROR("Invalid Parent Step")

        prodStepDict["description"] = self.Description
        prodStepDict["longDescription"] = self.LongDescription
        prodStepDict["stepType"] = self.Type
        prodStepDict["plugin"] = self.Plugin
        prodStepDict["agentType"] = self.AgentType
        prodStepDict["fileMask"] = self.FileMask
        # Optional fields
        prodStepDict["inputquery"] = json.dumps(self.Inputquery)
        prodStepDict["outputquery"] = json.dumps(self.Outputquery)
        prodStepDict["groupsize"] = self.GroupSize
        prodStepDict["body"] = json.dumps(self.Body)

        return S_OK(prodStepDict)