From d68b57baa8dadb14a14db5f5a32a2cc197c88c75 Mon Sep 17 00:00:00 2001 From: Lance Edgar Date: Sun, 8 Jan 2023 13:04:46 -0600 Subject: [PATCH] Allow for custom mail recipients in luigi crontab --- rattail_fabric2/deploy/luigi/crontab.mako | 4 ++++ rattail_fabric2/luigi.py | 5 +++-- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/rattail_fabric2/deploy/luigi/crontab.mako b/rattail_fabric2/deploy/luigi/crontab.mako index f27c223..55b6f77 100644 --- a/rattail_fabric2/deploy/luigi/crontab.mako +++ b/rattail_fabric2/deploy/luigi/crontab.mako @@ -1,4 +1,8 @@ ## -*- mode: conf; -*- +% if mailto: +MAILTO="${','.join(mailto)}" +% endif + # rotate logs and restart Luigi at *just before* 12:00am midnight 55 23 * * * root ${appdir}/rotate-luigi-logs.sh diff --git a/rattail_fabric2/luigi.py b/rattail_fabric2/luigi.py index 1e1e1c7..ef56099 100644 --- a/rattail_fabric2/luigi.py +++ b/rattail_fabric2/luigi.py @@ -33,7 +33,7 @@ deploy_common = make_deploy(__file__) def install_luigi(c, envroot, luigi='luigi', user='rattail', db_connection=None, - crontab=False): + crontab=False, crontab_mailto=None): """ Install and configure Luigi to the given virtualenv. """ @@ -73,7 +73,8 @@ def install_luigi(c, envroot, luigi='luigi', user='rattail', db_connection=None, context={'appdir': appdir}) if crontab: deploy_common(c, 'luigi/crontab.mako', '/etc/cron.d/luigi', - use_sudo=True, context={'appdir': appdir}) + use_sudo=True, context={'appdir': appdir, + 'mailto': crontab_mailto}) def install_overnight_script(c, envroot, user='rattail', automation='All',