Source code for aztk.spark.models.plugins.spark_ui_proxy.configuration

import os
from aztk.models.plugins.plugin_configuration import PluginConfiguration, PluginPort, PluginTargetRole
from aztk.models.plugins.plugin_file import PluginFile
from aztk.utils import constants

dir_path = os.path.dirname(os.path.realpath(__file__))


class SparkUIProxyPlugin(PluginConfiguration):
    def __init__(self):
        super().__init__(
            name="spark_ui_proxy",
            ports=[
                PluginPort(
                    internal=9999,
                    public=True
                )
            ],
            target_role=PluginTargetRole.Master,
            execute="spark_ui_proxy.sh",
            args=["localhost:8080", "9999"],
            files=[
                PluginFile("spark_ui_proxy.sh", os.path.join(dir_path, "spark_ui_proxy.sh")),
                PluginFile("spark_ui_proxy.py", os.path.join(dir_path, "spark_ui_proxy.py")),
            ],
        )