from airflow import version
from airflow.utils.log.logging_mixin import LoggingMixin
from airflow.configuration import conf
from airflow.exceptions import AirflowException
from airflow.models import DAG
__version__ = version.version
settings.initialize()
login = None # type: Optional[Callable]
from airflow import executors
from airflow import hooks
from airflow import macros
from airflow import operators
from airflow import sensors
class AirflowMacroPlugin:
# pylint:disable=missing-docstring
def __init__(self, namespace):
self.namespace = namespace
operators._integrate_plugins() # pylint:disable=protected-access
sensors._integrate_plugins() # pylint:disable=protected-access
hooks._integrate_plugins() # pylint:disable=protected-access
executors._integrate_plugins() # pylint:disable=protected-access
macros._integrate_plugins() # pylint:disable=protected-access
except ImportError as err:
log.critical(
"Cannot import authentication module %s. "
"Please correct your authentication backend or disable authentication: %s",
auth_backend, err
)
if conf.getboolean('webserver', 'AUTHENTICATE'):
raise AirflowException("Failed to import authentication backend")
class AirflowViewPlugin(BaseView):
pass
class AirflowMacroPlugin(object):
def __init__(self, namespace):
self.namespace = namespace
from airflow import operators # noqa: E402
from airflow import sensors # noqa: E402
from airflow import hooks # noqa: E402
from airflow import executors # noqa: E402
from airflow import macros # noqa: E402
operators._integrate_plugins()
sensors._integrate_plugins() # noqa: E402
hooks._integrate_plugins()
executors._integrate_plugins()
macros._integrate_plugins()
global login
login = import_module(auth_backend)
except ImportError as err:
_log.critical(
"Cannot import authentication module %s. "
"Please correct your authentication backend or disable authentication: %s",
auth_backend, err
)
if conf.getboolean('webserver', 'AUTHENTICATE'):
raise AirflowException("Failed to import authentication backend")
class AirflowViewPlugin(BaseView):
pass
class AirflowMacroPlugin(object):
def __init__(self, namespace):
self.namespace = namespace
from airflow import operators
from airflow import hooks
from airflow import executors
from airflow import macros
from airflow import contrib
operators._integrate_plugins()
hooks._integrate_plugins()
executors._integrate_plugins()
macros._integrate_plugins()
