41 lines
1.5 KiB
Python
41 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', dest='managers',
|
|
help='Send a test email to the addresses specified in settings.MANAGERS.',
|
|
)
|
|
parser.add_argument(
|
|
'--admins', action='store_true', dest='admins',
|
|
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.")
|