mist.mist_job.py Maven / Gradle / Ivy

Go to download
from abc import ABCMeta, abstractmethod

class ContextSupport:
    __metaclass__ = ABCMeta

    def __init__(self):
        pass

    context = None

    @abstractmethod
    def setup(self, context_wrapper):
        self.context = context_wrapper.context

class PublisherSupport:
    __metaclass__ = ABCMeta

    def __init__(self):
        pass

    publisher = None

    @abstractmethod
    def set_publisher(self, publisher_wrapper):
        self.publisher = publisher_wrapper

class MistJob(ContextSupport, PublisherSupport):
    __metaclass__ = ABCMeta

    @abstractmethod
    def do_stuff(self, params):
        raise NotImplementedError()

    def setup(self, context_wrapper):
        super(MistJob, self).setup(context_wrapper)

    def set_publisher(self, publisher_wrapper):
        super(MistJob, self).set_publisher(publisher_wrapper)

class WithSQLSupport(ContextSupport):
    __metaclass__ = ABCMeta

    sql_context = None
    session = None

    @abstractmethod
    def setup(self, context_wrapper):
        super(WithSQLSupport, self).setup(context_wrapper)
        try:
            from pyspark.sql import SparkSession
            self.session = context_wrapper.session
        except ImportError:
            self.sql_context = context_wrapper.sql_context

class WithHiveSupport(ContextSupport):
    __metaclass__ = ABCMeta

    hive_context = None
    session = None

    @abstractmethod
    def setup(self, context_wrapper):
        super(WithHiveSupport, self).setup(context_wrapper)
        try:
            from pyspark.sql import SparkSession
            self.session = context_wrapper.session
        except ImportError:
            self.hive_context = context_wrapper.hive_context

class WithMQTTPublisher(PublisherSupport):
    __metaclass__ = ABCMeta

    mqtt = None

    @abstractmethod
    def set_publisher(self, publisher_wrapper):
        super(WithMQTTPublisher, self).set_publisher(publisher_wrapper)
        self.mqtt = publisher_wrapper.mqtt