step_example.py

step_example.py

from stack.step import KlaviyoStackTerraformStep

class MailDumpsterStep(KlaviyoStackTerraformStep):
    STEP_NAME = 'mail_dumpster'
    MODULE_NAME = 'mail_dumpster'

    def __init__(self, stack):
        self.stack = stack

        super(MailDumpsterStep, self).__init__(
            self.stack,
            self.STEP_NAME,
            include_secrets=True,
            module=self.MODULE_NAME
        )

    def on_complete(self):
        step_outputs = self.get_step_outputs()

        self.update_zookeeper_znode_json_key_value(
            self.zookeeper_env_node_key,
            'hostname',
            step_outputs['mail_dumpster_domain']['value'],
        )