import logging
from . import run_vm_script
from ...util.base import BaseStep

LOG = logging.getLogger(__name__)


class ChangeDatabaseConfigFile(BaseStep):

    def __unicode__(self):
        return "Changing database config file..."

    def do(self, workflow_dict):
        context_dict = {
            'CONFIGFILE': True,
            'IS_HA': workflow_dict['databaseinfra'].plan.is_ha
        },

        ret_script = run_vm_script(
            workflow_dict=workflow_dict,
            context_dict=context_dict,
            script=workflow_dict['cloudstackpack'].script,
        )

        return ret_script

    def undo(self, workflow_dict):
        context_dict = {
            'CONFIGFILE': True,
        }

        ret_script = run_vm_script(
            workflow_dict=workflow_dict,
            context_dict=context_dict,
            script=workflow_dict['original_cloudstackpack'].script,
        )

        return ret_script

import logging
from workflow.steps.mysql.resize import run_vm_script
from workflow.steps.util.base import BaseStep

LOG = logging.getLogger(__name__)


class ChangeDatabaseConfigFile(BaseStep):

    def __unicode__(self):
        return "Changing database config file..."

    def do(self, workflow_dict):
        context_dict = {
            'CONFIGFILE': True,
            'IS_HA': workflow_dict['databaseinfra'].plan.is_ha
        },

        ret_script = run_vm_script(
            workflow_dict=workflow_dict,
            context_dict=context_dict,
            script=workflow_dict['cloudstackpack'].script,
        )

        return ret_script

    def undo(self, workflow_dict):
        context_dict = {
            'CONFIGFILE': True,
            'IS_HA': workflow_dict['databaseinfra'].plan.is_ha,
        }

        ret_script = run_vm_script(
            workflow_dict=workflow_dict,
            context_dict=context_dict,
            script=workflow_dict['original_cloudstackpack'].script,
        )

        return ret_script

Add is_ha variable to change config rollback
Add is_ha variable to change config rollback

  import logging
- from . import run_vm_script
+ from workflow.steps.mysql.resize import run_vm_script
- from ...util.base import BaseStep
+ from workflow.steps.util.base import BaseStep
  
  LOG = logging.getLogger(__name__)
  
  
  class ChangeDatabaseConfigFile(BaseStep):
  
      def __unicode__(self):
          return "Changing database config file..."
  
      def do(self, workflow_dict):
          context_dict = {
              'CONFIGFILE': True,
              'IS_HA': workflow_dict['databaseinfra'].plan.is_ha
          },
  
          ret_script = run_vm_script(
              workflow_dict=workflow_dict,
              context_dict=context_dict,
              script=workflow_dict['cloudstackpack'].script,
          )
  
          return ret_script
  
      def undo(self, workflow_dict):
          context_dict = {
              'CONFIGFILE': True,
+             'IS_HA': workflow_dict['databaseinfra'].plan.is_ha,
          }
  
          ret_script = run_vm_script(
              workflow_dict=workflow_dict,
              context_dict=context_dict,
              script=workflow_dict['original_cloudstackpack'].script,
          )
  
          return ret_script
  
Add is_ha variable to change config rollback
## Code Before:
import logging
from . import run_vm_script
from ...util.base import BaseStep

LOG = logging.getLogger(__name__)


class ChangeDatabaseConfigFile(BaseStep):

    def __unicode__(self):
        return "Changing database config file..."

    def do(self, workflow_dict):
        context_dict = {
            'CONFIGFILE': True,
            'IS_HA': workflow_dict['databaseinfra'].plan.is_ha
        },

        ret_script = run_vm_script(
            workflow_dict=workflow_dict,
            context_dict=context_dict,
            script=workflow_dict['cloudstackpack'].script,
        )

        return ret_script

    def undo(self, workflow_dict):
        context_dict = {
            'CONFIGFILE': True,
        }

        ret_script = run_vm_script(
            workflow_dict=workflow_dict,
            context_dict=context_dict,
            script=workflow_dict['original_cloudstackpack'].script,
        )

        return ret_script

## Instruction:
Add is_ha variable to change config rollback
## Code After:
import logging
from workflow.steps.mysql.resize import run_vm_script
from workflow.steps.util.base import BaseStep

LOG = logging.getLogger(__name__)


class ChangeDatabaseConfigFile(BaseStep):

    def __unicode__(self):
        return "Changing database config file..."

    def do(self, workflow_dict):
        context_dict = {
            'CONFIGFILE': True,
            'IS_HA': workflow_dict['databaseinfra'].plan.is_ha
        },

        ret_script = run_vm_script(
            workflow_dict=workflow_dict,
            context_dict=context_dict,
            script=workflow_dict['cloudstackpack'].script,
        )

        return ret_script

    def undo(self, workflow_dict):
        context_dict = {
            'CONFIGFILE': True,
            'IS_HA': workflow_dict['databaseinfra'].plan.is_ha,
        }

        ret_script = run_vm_script(
            workflow_dict=workflow_dict,
            context_dict=context_dict,
            script=workflow_dict['original_cloudstackpack'].script,
        )

        return ret_script
