diff --git a/airflow/executors/local_executor.py b/airflow/executors/local_executor.py index a29342f2c8e46..b4848ed15ebe0 100644 --- a/airflow/executors/local_executor.py +++ b/airflow/executors/local_executor.py @@ -31,7 +31,7 @@ from queue import Empty, Queue # pylint: disable=unused-import # noqa: F401 from typing import Any, List, Optional, Tuple, Union # pylint: disable=unused-import # noqa: F401 -from setproctitle import setproctitle # pylint: disable=no-name-in-module +from setproctitle import getproctitle, setproctitle # pylint: disable=no-name-in-module from airflow import settings from airflow.exceptions import AirflowException @@ -340,7 +340,10 @@ def end(self): def start(self) -> None: """Starts the executor""" + old_proctitle = getproctitle() + setproctitle("airflow executor -- LocalExecutor") self.manager = Manager() + setproctitle(old_proctitle) self.result_queue = self.manager.Queue() self.workers = [] self.workers_used = 0 diff --git a/airflow/utils/serve_logs.py b/airflow/utils/serve_logs.py index 0fefa420b7d84..e7890f7810186 100644 --- a/airflow/utils/serve_logs.py +++ b/airflow/utils/serve_logs.py @@ -19,6 +19,7 @@ import os import flask +from setproctitle import setproctitle from airflow.configuration import conf @@ -27,6 +28,7 @@ def serve_logs(): """Serves logs generated by Worker""" print("Starting flask") flask_app = flask.Flask(__name__) + setproctitle("airflow serve-logs") @flask_app.route('/log/') def serve_logs_view(filename): # pylint: disable=unused-variable