"""
Django management command to send a welcome email to an account user.
This module defines the ``send_welcome_email`` management command for the Smarter platform.
It allows administrators to send the HTML welcome email to a user
recorded in the AccountContact table, based on various identifying inputs.
Features
--------
- Sends an HTML welcome email to a target user of an account.
- User can be located via username, email, account number, or company name.
- Will create the AccountContact record if it does not already exist.
Command-line Options
-------------------
- ``--account_number``: The Smarter account number to which the user belongs.
- ``--company_name``: The company name to which the user belongs.
- ``--username``: The username of the user.
- ``--email``: The email address of the user.
Typical Usage
-------------
Used by administrators via ``manage.py send_welcome_email`` to send or re-send
the welcome email to new or existing users for onboarding or support purposes.
"""
from typing import Optional
from smarter.apps.account.models import Account, AccountContact, UserProfile
from smarter.common.exceptions import SmarterValueError
from smarter.lib.django.management.base import SmarterCommand
# pylint: disable=E1101
[docs]
class Command(SmarterCommand):
"""Send the html welcome email to a user in the AccountContact table."""
[docs]
def add_arguments(self, parser):
"""Add arguments to the command."""
parser.add_argument("--account_number", type=str, help="The Smarter account number to which the user belongs")
parser.add_argument("--company_name", type=str, help="The company name to which the user belongs")
parser.add_argument("--username", type=str, help="The username")
parser.add_argument("--email", type=str, help="The email address")
[docs]
def handle(self, *args, **options):
"""Create the superuser account."""
self.handle_begin()
account_number = options["account_number"]
company_name = options["company_name"]
username = options["username"]
email = options["email"]
account: Optional[Account] = None
if username:
user_profile = UserProfile.objects.get(user__username=username)
account = user_profile.cached_account
email = email or user_profile.user.email
elif email:
user_profile = UserProfile.objects.get(user__email=email)
account = user_profile.cached_account
else:
if options["account_number"]:
try:
account = Account.get_cached_object(account_number=account_number)
except Account.DoesNotExist as e:
self.handle_completed_failure(e, msg=f"Account {account_number} not found.")
return
elif options["company_name"]:
try:
account = Account.get_cached_object(company_name=company_name)
except Account.DoesNotExist as e:
self.handle_completed_failure(e, msg=f"Account {company_name} not found.")
return
else:
raise SmarterValueError("You must provide either an account number or a company name.")
raise SmarterValueError(
"You must provide either a username or an email address and an account number or company name."
)
account_contact, _ = AccountContact.objects.get_or_create(
account=account,
email=email,
)
account_contact.send_welcome_email()
self.handle_completed_success()