44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
import socket
|
|
|
|
from django.core.mail import mail_admins, mail_managers, send_mail
|
|
from django.core.management.base import BaseCommand
|
|
from django.utils import timezone
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "Sends a test email to the email addresses specified as arguments."
|
|
missing_args_message = "You must specify some email recipients, or pass the --managers or --admin options."
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument(
|
|
"email",
|
|
nargs="*",
|
|
help="One or more email addresses to send a test email to.",
|
|
)
|
|
parser.add_argument(
|
|
"--managers",
|
|
action="store_true",
|
|
help="Send a test email to the addresses specified in settings.MANAGERS.",
|
|
)
|
|
parser.add_argument(
|
|
"--admins",
|
|
action="store_true",
|
|
help="Send a test email to the addresses specified in settings.ADMINS.",
|
|
)
|
|
|
|
def handle(self, *args, **kwargs):
|
|
subject = "Test email from %s on %s" % (socket.gethostname(), timezone.now())
|
|
|
|
send_mail(
|
|
subject=subject,
|
|
message="If you're reading this, it was successful.",
|
|
from_email=None,
|
|
recipient_list=kwargs["email"],
|
|
)
|
|
|
|
if kwargs["managers"]:
|
|
mail_managers(subject, "This email was sent to the site managers.")
|
|
|
|
if kwargs["admins"]:
|
|
mail_admins(subject, "This email was sent to the site admins.")
|