extern/django-helpdesk
1
0
Fork 1
mirror of https://github.com/django-helpdesk/django-helpdesk.git synced 2025-06-20 17:47:58 +02:00
Code Issues Packages Projects Releases Wiki Activity

Support optional passing in of body text.

Browse Source
This commit is contained in:
Christopher Broderick 2025-04-02 10:14:26 +01:00
parent a8f26ff46f
commit c68b3d933d
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
helpdesk/tests/utils.py
View File

@ -288,15 +288,18 @@ def add_simple_email_headers(
def generate_mime_part( def generate_mime_part(
locale: str = "en_US", locale: str = "en_US",
part_type: str = "plain", part_type: str = "plain",
body: str = None,
) -> typing.Optional[Message]: ) -> typing.Optional[Message]:
""" """
Generates amime part of the sepecified type Generates a mime part of the specified type
:param locale: change this to generate locale specific strings :param locale: change this to generate locale specific strings
:param text_type: options are plain, html, image (attachment), file (attachment) :param text_type: options are plain, html, image (attachment), file (attachment)
:param body: if provided then will be added to the plain mime part only
""" """
if "plain" == part_type: if "plain" == part_type:
body = get_fake("text", locale=locale, min_length=1024) if body is None:
body = get_fake("text", locale=locale, min_length=1024)
msg = MIMEText(body, part_type) msg = MIMEText(body, part_type)
elif "html" == part_type: elif "html" == part_type:
body = get_fake_html(locale=locale, wrap_in_body_tag=True) body = get_fake_html(locale=locale, wrap_in_body_tag=True)
@ -317,6 +320,7 @@ def generate_multipart_email(
type_list: typing.List[str] = ["plain", "html", "image"], type_list: typing.List[str] = ["plain", "html", "image"],
sub_type: str = None, sub_type: str = None,
use_short_email: bool = False, use_short_email: bool = False,
body: str = None,
) -> typing.Tuple[Message, typing.Tuple[str, str], typing.Tuple[str, str]]: ) -> typing.Tuple[Message, typing.Tuple[str, str], typing.Tuple[str, str]]:
""" """
Generates an email including headers with the defined multiparts Generates an email including headers with the defined multiparts
@ -325,10 +329,11 @@ def generate_multipart_email(
:param type_list: options are plain, html, image (attachment), file (attachment), and executable (.exe attachment) :param type_list: options are plain, html, image (attachment), file (attachment), and executable (.exe attachment)
:param sub_type: multipart sub type that defaults to "mixed" if not specified :param sub_type: multipart sub type that defaults to "mixed" if not specified
:param use_short_email: produces a "To" or "From" that is only the email address if True :param use_short_email: produces a "To" or "From" that is only the email address if True
:param body: if provided then will be added to the plain mime part only
""" """
msg = MIMEMultipart(sub_type) if sub_type else MIMEMultipart() msg = MIMEMultipart(sub_type) if sub_type else MIMEMultipart()
for part_type in type_list: for part_type in type_list:
msg.attach(generate_mime_part(locale=locale, part_type=part_type)) msg.attach(generate_mime_part(locale=locale, part_type=part_type, body=body))
from_meta, to_meta = add_simple_email_headers( from_meta, to_meta = add_simple_email_headers(
msg, locale=locale, use_short_email=use_short_email msg, locale=locale, use_short_email=use_short_email
) )
@ -336,12 +341,13 @@ def generate_multipart_email(
def generate_text_email( def generate_text_email(
locale: str = "en_US", use_short_email: bool = False locale: str = "en_US", use_short_email: bool = False, body: str = None,
) -> typing.Tuple[Message, typing.Tuple[str, str], typing.Tuple[str, str]]: ) -> typing.Tuple[Message, typing.Tuple[str, str], typing.Tuple[str, str]]:
""" """
Generates an email including headers Generates an email including headers
""" """
body = get_fake("text", locale=locale, min_length=1024) if body is None:
body = get_fake("text", locale=locale, min_length=1024)
msg = MIMEText(body) msg = MIMEText(body)
from_meta, to_meta = add_simple_email_headers( from_meta, to_meta = add_simple_email_headers(
msg, locale=locale, use_short_email=use_short_email msg, locale=locale, use_short_email=use_short_email