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

__RCSID__ = "3c530efac (2020-12-05 00:59:33 +0100) 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)