dag_factory_example.py

from dag_factory.process_invoices import ProcessInvoices
from dag_factory.process_messages import ProcessMessages

class DAGFactory():
    """The DAG Factory Class"""

    def __init__(self, environment_config=None):
        self.environment_config = environment_config

    def create(self, dag_type, config: list):
        supported_dags = {
            'ProcessInvoices': ProcessInvoices,
            'ProcessMessages': ProcessMessages
        }
        dag = supported_dags[dag_type](
            config=config, 
            environment_config=self.environment_config
        )
        return dag