# Django 5.2 documentation

# Source code for django.core.mail.message

    import mimetypes
    from collections import namedtuple
    from email import charset as Charset
    from email import encoders as Encoders
    from email import generator, message_from_bytes
    from email.errors import HeaderParseError
    from email.header import Header
    from email.headerregistry import Address, parser
    from email.message import Message
    from email.mime.base import MIMEBase
    from email.mime.message import MIMEMessage
    from email.mime.multipart import MIMEMultipart
    from email.mime.text import MIMEText
    from email.utils import formataddr, formatdate, getaddresses, make_msgid
    from io import BytesIO, StringIO
    from pathlib import Path

    from django.conf import settings
    from django.core.mail.utils import DNS_NAME
    from django.utils.encoding import force_bytes, force_str, punycode

    # Don't BASE64-encode UTF-8 messages so that we avoid unwanted attention from
    # some spam filters.
    utf8_charset = Charset.Charset("utf-8")
    utf8_charset.body_encoding = None  # Python defaults to BASE64
    utf8_charset_qp = Charset.Charset("utf-8")
    utf8_charset_qp.body_encoding = Charset.QP

    # Default MIME type to use on attachments (if it is not explicitly given
    # and cannot be guessed).
    DEFAULT_ATTACHMENT_MIME_TYPE = "application/octet-stream"

    RFC5322_EMAIL_LINE_LENGTH_LIMIT = 998

    class BadHeaderError(ValueError):
        pass

    # Header names that contain structured address data (RFC 5322).
    ADDRESS_HEADERS = {
        "from",
        "sender",
        "reply-to",
        "to",
        "cc",
        "bcc",
        "resent-from",
        "resent-sender",
        "resent-to",
        "resent-cc",
        "resent-bcc",
    }

    def forbid_multi_line_headers(name, val, encoding):
        """Forbid multi-line headers to prevent header injection."""
        encoding = encoding or settings.DEFAULT_CHARSET
        val = str(val)  # val may be lazy
        if "\n" in val or "\r" in val:
            raise BadHeaderError(
                "Header values can't contain newlines (got %r for header %r)" % (val, name)
            )
        try:
            val.encode("ascii")
        except UnicodeEncodeError:
            if name.lower() in ADDRESS_HEADERS:
                val = ", ".join(
                    sanitize_address(addr, encoding) for addr in getaddresses((val,))
                )
            else:
                val = Header(val, encoding).encode()
        else:
            if name.lower() == "subject":
                val = Header(val).encode()
        return name, val

    def sanitize_address(addr, encoding):
        """
        Format a pair of (name, address) or an email address string.
        """
        address = None
        if not isinstance(addr, tuple):
            addr = force_str(addr)
            try:
                token, rest = parser.get_mailbox(addr)
            except (HeaderParseError, ValueError, IndexError):
                raise ValueError('Invalid address "%s"' % addr)
            else:
                if rest:
                    # The entire email address must be parsed.
                    raise ValueError(
                        'Invalid address; only %s could be parsed from "%s"' % (token, addr)
                    )
                nm = token.display_name or ""
                localpart = token.local_part
                domain = token.domain or ""
        else:
            nm, address = addr
            if "@" not in address:
                raise ValueError(f'Invalid address "{address}"')
            localpart, domain = address.rsplit("@", 1)

        address_parts = nm + localpart + domain
        if "\n" in address_parts or "\r" in address_parts:
            raise ValueError("Invalid address; address parts cannot contain newlines.")

        # Avoid UTF-8 encode, if it's possible.
        try:
            nm.encode("ascii")
            nm = Header(nm).encode()
        except UnicodeEncodeError:
            nm = Header(nm, encoding).encode()
        try:
            localpart.encode("ascii")
        except UnicodeEncodeError:
            localpart = Header(localpart, encoding).encode()
        domain = punycode(domain)

        parsed_address = Address(username=localpart, domain=domain)
        return formataddr((nm, parsed_address.addr_spec))

    class MIMEMixin:
        def as_string(self, unixfrom=False, linesep="\n"):
            """Return the entire formatted message as a string.
            Optional `unixfrom' when True, means include the Unix From_ envelope
            header.

            This overrides the default as_string() implementation to not mangle
            lines that begin with 'From '. See bug #13433 for details.
            """
            fp = StringIO()
            g = generator.Generator(fp, mangle_from_=False)
            g.flatten(self, unixfrom=unixfrom, linesep=linesep)
            return fp.getvalue()

        def as_bytes(self, unixfrom=False, linesep="\n"):
            """Return the entire formatted message as bytes.
            Optional `unixfrom' when True, means include the Unix From_ envelope
            header.

            This overrides the default as_bytes() implementation to not mangle
            lines that begin with 'From '. See bug #13433 for details.
            """
            fp = BytesIO()
            g = generator.BytesGenerator(fp, mangle_from_=False)
            g.flatten(self, unixfrom=unixfrom, linesep=linesep)
            return fp.getvalue()

    class SafeMIMEMessage(MIMEMixin, MIMEMessage):
        def __setitem__(self, name, val):
            # Per RFC 2046 Section 5.2.1, message/rfc822 attachment headers must be ASCII.
            name, val = forbid_multi_line_headers(name, val, "ascii")
            MIMEMessage.__setitem__(self, name, val)

    class SafeMIMEText(MIMEMixin, MIMEText):
        def __init__(self, _text, _subtype="plain", _charset=None):
            self.encoding = _charset
            MIMEText.__init__(self, _text, _subtype=_subtype, _charset=_charset)

        def __setitem__(self, name, val):
            name, val = forbid_multi_line_headers(name, val, self.encoding)
            MIMEText.__setitem__(self, name, val)

        def set_payload(self, payload, charset=None):
            if charset == "utf-8" and not isinstance(charset, Charset.Charset):
                has_long_lines = any(
                    len(line.encode(errors="surrogateescape"))
                    > RFC5322_EMAIL_LINE_LENGTH_LIMIT
                    for line in payload.splitlines()
                )
                # Quoted-Printable encoding has the side effect of shortening long
                # lines, if any (#22561).
                charset = utf8_charset_qp if has_long_lines else utf8_charset
            MIMEText.set_payload(self, payload, charset=charset)

    class SafeMIMEMultipart(MIMEMixin, MIMEMultipart):
        def __init__(
            self, _subtype="mixed", boundary=None, _subparts=None, encoding=None, **_params
        ):
            self.encoding = encoding
            MIMEMultipart.__init__(self, _subtype, boundary, _subparts, **_params)

        def __setitem__(self, name, val):
            name, val = forbid_multi_line_headers(name, val, self.encoding)
            MIMEMultipart.__setitem__(self, name, val)

    EmailAlternative = namedtuple("Alternative", ["content", "mimetype"])
    EmailAttachment = namedtuple("Attachment", ["filename", "content", "mimetype"])

    class EmailMessage:
        """A container for email information."""

        content_subtype = "plain"
        mixed_subtype = "mixed"
        encoding = None  # None => use settings default

        def __init__(
            self,
            subject="",
            body="",
            from_email=None,
            to=None,
            bcc=None,
            connection=None,
            attachments=None,
            headers=None,
            cc=None,
            reply_to=None,
        ):
            """
            Initialize a single email message (which can be sent to multiple
            recipients).
            """
            if to:
                if isinstance(to, str):
                    raise TypeError('"to" argument must be a list or tuple')
                self.to = list(to)
            else:
                self.to = []
            if cc:
                if isinstance(cc, str):
                    raise TypeError('"cc" argument must be a list or tuple')
                self.cc = list(cc)
            else:
                self.cc = []
            if bcc:
                if isinstance(bcc, str):
                    raise TypeError('"bcc" argument must be a list or tuple')
                self.bcc = list(bcc)
            else:
                self.bcc = []
            if reply_to:
                if isinstance(reply_to, str):
                    raise TypeError('"reply_to" argument must be a list or tuple')
                self.reply_to = list(reply_to)
            else:
                self.reply_to = []
            self.from_email = from_email or settings.DEFAULT_FROM_EMAIL
            self.subject = subject
            self.body = body or ""
            self.attachments = []
            if attachments:
                for attachment in attachments:
                    if isinstance(attachment, MIMEBase):
                        self.attach(attachment)
                    else:
                        self.attach(*attachment)
            self.extra_headers = headers or {}
            self.connection = connection

        def get_connection(self, fail_silently=False):
            from django.core.mail import get_connection

            if not self.connection:
                self.connection = get_connection(fail_silently=fail_silently)
            return self.connection

        def message(self):
            encoding = self.encoding or settings.DEFAULT_CHARSET
            msg = SafeMIMEText(self.body, self.content_subtype, encoding)
            msg = self._create_message(msg)
            msg["Subject"] = self.subject
            msg["From"] = self.extra_headers.get("From", self.from_email)
            self._set_list_header_if_not_empty(msg, "To", self.to)
            self._set_list_header_if_not_empty(msg, "Cc", self.cc)
            self._set_list_header_if_not_empty(msg, "Reply-To", self.reply_to)

            # Email header names are case-insensitive (RFC 2045), so we have to
            # accommodate that when doing comparisons.
            header_names = [key.lower() for key in self.extra_headers]
            if "date" not in header_names:
                # formatdate() uses stdlib methods to format the date, which use
                # the stdlib/OS concept of a timezone, however, Django sets the
                # TZ environment variable based on the TIME_ZONE setting which
                # will get picked up by formatdate().
                msg["Date"] = formatdate(localtime=settings.EMAIL_USE_LOCALTIME)
            if "message-id" not in header_names:
                # Use cached DNS_NAME for performance
                msg["Message-ID"] = make_msgid(domain=DNS_NAME)
            for name, value in self.extra_headers.items():
                # Avoid headers handled above.
                if name.lower() not in {"from", "to", "cc", "reply-to"}:
                    msg[name] = value
            return msg

        def recipients(self):
            """
            Return a list of all recipients of the email (includes direct
            addressees as well as Cc and Bcc entries).
            """
            return [email for email in (self.to + self.cc + self.bcc) if email]

        def send(self, fail_silently=False):
            """Send the email message."""
            if not self.recipients():
                # Don't bother creating the network connection if there's nobody to
                # send to.
                return 0
            return self.get_connection(fail_silently).send_messages([self])

        def attach(self, filename=None, content=None, mimetype=None):
            """
            Attach a file with the given filename and content. The filename can
            be omitted and the mimetype is guessed, if not provided.

            If the first parameter is a MIMEBase subclass, insert it directly
            into the resulting message attachments.

            For a text/* mimetype (guessed or specified), when a bytes object is
            specified as content, decode it as UTF-8. If that fails, set the
            mimetype to DEFAULT_ATTACHMENT_MIME_TYPE and don't decode the content.
            """
            if isinstance(filename, MIMEBase):
                if content is not None or mimetype is not None:
                    raise ValueError(
                        "content and mimetype must not be given when a MIMEBase "
                        "instance is provided."
                    )
                self.attachments.append(filename)
            elif content is None:
                raise ValueError("content must be provided.")
            else:
                mimetype = (
                    mimetype
                    or mimetypes.guess_type(filename)[0]
                    or DEFAULT_ATTACHMENT_MIME_TYPE
                )
                basetype, subtype = mimetype.split("/", 1)

                if basetype == "text":
                    if isinstance(content, bytes):
                        try:
                            content = content.decode()
                        except UnicodeDecodeError:
                            # If mimetype suggests the file is text but it's
                            # actually binary, read() raises a UnicodeDecodeError.
                            mimetype = DEFAULT_ATTACHMENT_MIME_TYPE

                self.attachments.append(EmailAttachment(filename, content, mimetype))

        def attach_file(self, path, mimetype=None):
            """
            Attach a file from the filesystem.

            Set the mimetype to DEFAULT_ATTACHMENT_MIME_TYPE if it isn't specified
            and cannot be guessed.

            For a text/* mimetype (guessed or specified), decode the file's content
            as UTF-8. If that fails, set the mimetype to
            DEFAULT_ATTACHMENT_MIME_TYPE and don't decode the content.
            """
            path = Path(path)
            with path.open("rb") as file:
                content = file.read()
                self.attach(path.name, content, mimetype)

        def _create_message(self, msg):
            return self._create_attachments(msg)

        def _create_attachments(self, msg):
            if self.attachments:
                encoding = self.encoding or settings.DEFAULT_CHARSET
                body_msg = msg
                msg = SafeMIMEMultipart(_subtype=self.mixed_subtype, encoding=encoding)
                if self.body or body_msg.is_multipart():
                    msg.attach(body_msg)
                for attachment in self.attachments:
                    if isinstance(attachment, MIMEBase):
                        msg.attach(attachment)
                    else:
                        msg.attach(self._create_attachment(*attachment))
            return msg

        def _create_mime_attachment(self, content, mimetype):
            """
            Convert the content, mimetype pair into a MIME attachment object.

            If the mimetype is message/rfc822, content may be an
            email.Message or EmailMessage object, as well as a str.
            """
            basetype, subtype = mimetype.split("/", 1)
            if basetype == "text":
                encoding = self.encoding or settings.DEFAULT_CHARSET
                attachment = SafeMIMEText(content, subtype, encoding)
            elif basetype == "message" and subtype == "rfc822":
                # Bug #18967: Per RFC 2046 Section 5.2.1, message/rfc822
                # attachments must not be base64 encoded.
                if isinstance(content, EmailMessage):
                    # convert content into an email.Message first
                    content = content.message()
                elif not isinstance(content, Message):
                    # For compatibility with existing code, parse the message
                    # into an email.Message object if it is not one already.
                    content = message_from_bytes(force_bytes(content))

                attachment = SafeMIMEMessage(content, subtype)
            else:
                # Encode non-text attachments with base64.
                attachment = MIMEBase(basetype, subtype)
                attachment.set_payload(content)
                Encoders.encode_base64(attachment)
            return attachment

        def _create_attachment(self, filename, content, mimetype=None):
            """
            Convert the filename, content, mimetype triple into a MIME attachment
            object.
            """
            attachment = self._create_mime_attachment(content, mimetype)
            if filename:
                try:
                    filename.encode("ascii")
                except UnicodeEncodeError:
                    filename = ("utf-8", "", filename)
                attachment.add_header(
                    "Content-Disposition", "attachment", filename=filename
                )
            return attachment

        def _set_list_header_if_not_empty(self, msg, header, values):
            """
            Set msg's header, either from self.extra_headers, if present, or from
            the values argument if not empty.
            """
            try:
                msg[header] = self.extra_headers[header]
            except KeyError:
                if values:
                    msg[header] = ", ".join(str(v) for v in values)

    class EmailMultiAlternatives(EmailMessage):
        """
        A version of EmailMessage that makes it easy to send multipart/alternative
        messages. For example, including text and HTML versions of the text is
        made easier.
        """

        alternative_subtype = "alternative"

        def __init__(
            self,
            subject="",
            body="",
            from_email=None,
            to=None,
            bcc=None,
            connection=None,
            attachments=None,
            headers=None,
            alternatives=None,
            cc=None,
            reply_to=None,
        ):
            """
            Initialize a single email message (which can be sent to multiple
            recipients).
            """
            super().__init__(
                subject,
                body,
                from_email,
                to,
                bcc,
                connection,
                attachments,
                headers,
                cc,
                reply_to,
            )
            self.alternatives = [
                EmailAlternative(*alternative) for alternative in (alternatives or [])
            ]

        def attach_alternative(self, content, mimetype):
            """Attach an alternative content representation."""
            if content is None or mimetype is None:
                raise ValueError("Both content and mimetype must be provided.")
            self.alternatives.append(EmailAlternative(content, mimetype))

        def _create_message(self, msg):
            return self._create_attachments(self._create_alternatives(msg))

        def _create_alternatives(self, msg):
            encoding = self.encoding or settings.DEFAULT_CHARSET
            if self.alternatives:
                body_msg = msg
                msg = SafeMIMEMultipart(
                    _subtype=self.alternative_subtype, encoding=encoding
                )
                if self.body:
                    msg.attach(body_msg)
                for alternative in self.alternatives:
                    msg.attach(
                        self._create_mime_attachment(
                            alternative.content, alternative.mimetype
                        )
                    )
            return msg

        def body_contains(self, text):
            """
            Checks that ``text`` occurs in the email body and in all attached MIME
            type text/* alternatives.
            """
            if text not in self.body:
                return False

            for content, mimetype in self.alternatives:
                if mimetype.startswith("text/") and text not in content:
                    return False
            return True