plugins/mozmenu.py

"""
Plugin that adds a "Mozilla" entry to the top bar with some useful links.

Based on an example at
https://github.com/airflow-plugins/Getting-Started/blob/master/Tutorial/creating-ui-modification.md
"""
from airflow.plugins_manager import AirflowPlugin

telemetry_airflow = {
    "name": "telemetry-airflow on GitHub",
    "category": "Mozilla",
    "href": "https://github.com/mozilla/telemetry-airflow",
}

wtmo_dev = {
    "name": "WTMO Developer Guide",
    "category": "Mozilla",
    "href": "https://mozilla-hub.atlassian.net/wiki/spaces/SRE/pages/27922811/WTMO+Developer+Guide",
}

airflow_triage_guide = {
    "name": "Airflow Triage Guide",
    "category": "Mozilla",
    "href": "https://mozilla-hub.atlassian.net/wiki/spaces/DATA/pages/175603730/Airflow+Triage+Guide",
}

gke_cluster = {
    "name": "GKE cluster",
    "category": "Mozilla",
    "href": "https://console.cloud.google.com/kubernetes/workload/overview?project=moz-fx-data-airflow-gke-prod",
}


# ruff: noqa: RUF012
class MozMenuPlugin(AirflowPlugin):
    name = "Mozilla"
    operators = []
    flask_blueprints = []
    hooks = []
    executors = []
    appbuilder_views = []
    appbuilder_menu_items = [
        telemetry_airflow,
        wtmo_dev,
        airflow_triage_guide,
        gke_cluster,
    ]