3
0
Fork 0

fix: allow caller to specify default subject for email message

we already allow for some config fallbacks but sometimes the caller
needs to declare the default, to use as fallback when specific config
is not present
This commit is contained in:
Lance Edgar 2025-08-10 11:05:39 -05:00
parent 22d3ba97c9
commit eb6ad9884c
2 changed files with 26 additions and 7 deletions

View file

@ -370,7 +370,7 @@ class EmailHandler(GenericHandler):
""" """
return Message(**kwargs) return Message(**kwargs)
def make_auto_message(self, key, context={}, **kwargs): def make_auto_message(self, key, context={}, default_subject=None, **kwargs):
""" """
Make a new email message using config to determine its Make a new email message using config to determine its
properties, and auto-generating body from a template. properties, and auto-generating body from a template.
@ -386,6 +386,9 @@ class EmailHandler(GenericHandler):
:param context: Context dict used to render template(s) for :param context: Context dict used to render template(s) for
the message. the message.
:param default_subject: Optional :attr:`~Message.subject`
template/string to use, if config does not specify one.
:param \**kwargs: Any remaining kwargs are passed as-is to :param \**kwargs: Any remaining kwargs are passed as-is to
:meth:`make_message()`. More on this below. :meth:`make_message()`. More on this below.
@ -408,7 +411,7 @@ class EmailHandler(GenericHandler):
if 'sender' not in kwargs: if 'sender' not in kwargs:
kwargs['sender'] = self.get_auto_sender(key) kwargs['sender'] = self.get_auto_sender(key)
if 'subject' not in kwargs: if 'subject' not in kwargs:
kwargs['subject'] = self.get_auto_subject(key, context) kwargs['subject'] = self.get_auto_subject(key, context, default=default_subject)
if 'to' not in kwargs: if 'to' not in kwargs:
kwargs['to'] = self.get_auto_to(key) kwargs['to'] = self.get_auto_to(key)
if 'cc' not in kwargs: if 'cc' not in kwargs:
@ -449,7 +452,7 @@ class EmailHandler(GenericHandler):
# fall back to global default, if present # fall back to global default, if present
return self.config.get(f'{self.config.appname}.email.default.replyto') return self.config.get(f'{self.config.appname}.email.default.replyto')
def get_auto_subject(self, key, context={}, rendered=True, setting=None): def get_auto_subject(self, key, context={}, rendered=True, setting=None, default=None):
""" """
Returns automatic :attr:`~wuttjamaican.email.Message.subject` Returns automatic :attr:`~wuttjamaican.email.Message.subject`
line for a message, as determined by config. line for a message, as determined by config.
@ -470,15 +473,17 @@ class EmailHandler(GenericHandler):
instance. This is passed along to instance. This is passed along to
:meth:`get_auto_subject_template()`. :meth:`get_auto_subject_template()`.
:param default: Default subject to use if none is configured.
:returns: Final subject text, either "raw" or rendered. :returns: Final subject text, either "raw" or rendered.
""" """
template = self.get_auto_subject_template(key, setting=setting) template = self.get_auto_subject_template(key, setting=setting, default=default)
if not rendered: if not rendered:
return template return template
return Template(template).render(**context) return Template(template).render(**context)
def get_auto_subject_template(self, key, setting=None): def get_auto_subject_template(self, key, setting=None, default=None):
""" """
Returns the template string to use for automatic subject line Returns the template string to use for automatic subject line
of a message, as determined by config. of a message, as determined by config.
@ -497,6 +502,8 @@ class EmailHandler(GenericHandler):
optimization; otherwise it will be fetched if needed via optimization; otherwise it will be fetched if needed via
:meth:`get_email_setting()`. :meth:`get_email_setting()`.
:param default: Default subject to use if none is configured.
:returns: Final subject template, as raw text. :returns: Final subject template, as raw text.
""" """
# prefer configured subject specific to key # prefer configured subject specific to key
@ -504,6 +511,10 @@ class EmailHandler(GenericHandler):
if template: if template:
return template return template
# or use caller-specified default, if applicable
if default:
return default
# or subject from email setting, if defined # or subject from email setting, if defined
if not setting: if not setting:
setting = self.get_email_setting(key) setting = self.get_email_setting(key)

View file

@ -291,7 +291,7 @@ class TestEmailHandler(ConfigTestCase):
msg = handler.make_auto_message('foo', subject=None) msg = handler.make_auto_message('foo', subject=None)
get_auto_subject.assert_not_called() get_auto_subject.assert_not_called()
msg = handler.make_auto_message('foo') msg = handler.make_auto_message('foo')
get_auto_subject.assert_called_once_with('foo', {}) get_auto_subject.assert_called_once_with('foo', {}, default=None)
# to # to
with patch.object(handler, 'get_auto_to') as get_auto_to: with patch.object(handler, 'get_auto_to') as get_auto_to:
@ -376,7 +376,7 @@ class TestEmailHandler(ConfigTestCase):
template = handler.get_auto_subject_template('foo') template = handler.get_auto_subject_template('foo')
self.assertEqual(template, "Foo Message") self.assertEqual(template, "Foo Message")
# setting can provide default subject # EmailSetting can provide default subject
providers = { providers = {
'wuttatest': MagicMock(email_modules=['tests.test_email']), 'wuttatest': MagicMock(email_modules=['tests.test_email']),
} }
@ -385,6 +385,10 @@ class TestEmailHandler(ConfigTestCase):
template = handler.get_auto_subject_template('mock_foo') template = handler.get_auto_subject_template('mock_foo')
self.assertEqual(template, "MOCK FOO!") self.assertEqual(template, "MOCK FOO!")
# caller can provide default subject
template = handler.get_auto_subject_template('mock_foo', default="whatever is clever")
self.assertEqual(template, "whatever is clever")
def test_get_auto_subject(self): def test_get_auto_subject(self):
handler = self.make_handler() handler = self.make_handler()
@ -397,6 +401,10 @@ class TestEmailHandler(ConfigTestCase):
subject = handler.get_auto_subject('foo') subject = handler.get_auto_subject('foo')
self.assertEqual(subject, "Wutta Message") self.assertEqual(subject, "Wutta Message")
# caller can provide default subject
subject = handler.get_auto_subject('foo', default="whatever is clever")
self.assertEqual(subject, "whatever is clever")
# can configure just for key # can configure just for key
self.config.setdefault('wutta.email.foo.subject', "Foo Message") self.config.setdefault('wutta.email.foo.subject', "Foo Message")
subject = handler.get_auto_subject('foo') subject = handler.get_auto_subject('foo')