diff --git a/scouter/activities/activities.py b/scouter/activities/activities.py index 775f958..9ada6d3 100644 --- a/scouter/activities/activities.py +++ b/scouter/activities/activities.py @@ -8,6 +8,7 @@ with workflow.unsafe.imports_passed_through(): from scouter.activities.gates import Gates from scouter.activities.mongodb import MongoDB from typing import Any + from os import getenv class Activities(Postgres, Redis, Gates, MongoDB,): @@ -61,6 +62,8 @@ class Activities(Postgres, Redis, Gates, MongoDB,): notification_handler=notification_handler ) + self.pod_id = getenv("HOSTNAME", "localhost") + def shutdown(self): Postgres.close(self) MongoDB.shutdown(self) diff --git a/scouter/workflow/scouter.py b/scouter/workflow/scouter.py index c2afbff..3f15e13 100644 --- a/scouter/workflow/scouter.py +++ b/scouter/workflow/scouter.py @@ -39,7 +39,7 @@ class Scouter: 'model_name': input_data['model_name'], 'schedule_name': input_data['schedule_name'], 'workflow_name': input_data['workflow_name'], - 'pod_id': getenv("HOSTNAME", "localhost") + 'pod_id': Activities.pod_id } }