Allow overriding the mail instances from the outside

matthiask · Nov 17, 2018 · 8e0b04f · 8e0b04f
1 parent 8a4936c
commit 8e0b04f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/spark/spark_mails/api.py b/spark/spark_mails/api.py
@@ -13,8 +13,8 @@ def process_mail_events(iterable, *, defaults=None, fail_silently=False):
         mail.send(fail_silently=fail_silently)
 
 
-def mails_from_events(iterable, *, defaults=None):
-    mails = Mail.objects.as_mails()
+def mails_from_events(iterable, *, defaults=None, mails=None):
+    mails = Mail.objects.as_mails() if mails is None else mails
     defaults = {} if defaults is None else defaults
 
     for e in iterable: