sendAwsSms.py

# ##############
# sendAwsSms.py
#
# This will send an sms message using the AWS SNS services
#
# Arguments
#  -p   - (required) This is the phone number that will receive the sms
#  -s   - (required) This is the subject that is sent on the message
#  -m   - (required) This is the message to be sent.  Messages longer than AWS max will automatically 
#                        be split to more messages
#  -t   - (optional) For Transactional -- Critical Messages (** Otherwise the message is sent as promotional)
#  -r   - (optional) us-west-2 (default)

import sys
import getopt
import boto3
import os
import json
from botocore.exceptions import ClientError

# ########### 
# GLOBAL SETTINGS
HOMECU_SCRIPT_HELP = 'sendAwsSms.py -p <phone number> -s <subject> -m <message> [-t] [-r <region>]'

# SENDER ID -- NOT SUPPORTED IN US

# Always use us-west-2.. This should work for local development, alpha stack(us-east-2) and prod
AWS_REGION = 'us-west-2'


def main(argv):

    main_return = {
        'status': '000',
        'error': '',
        'response': {}
    }

    # ### PARSE ARGUMENTS
    try:
        opts, args = getopt.getopt(argv, "p:s:m:r:t")
    except getopt.GetoptError:
        main_return['status'] = '999'
        main_return['error'] = HOMECU_SCRIPT_HELP
        print(json.dumps(main_return))
        sys.exit(2)

    missing_phone = True
    missing_subject = True
    missing_message = True
    arg_transaction = False

    for opt, arg in opts:
        if opt == '-p':
            arg_phone = arg
            missing_phone = False
        elif opt == '-s':
            sms_subject = arg
            missing_subject = False
        elif opt == '-m':
            sms_message = arg
            missing_message = False
        elif opt == '-t':
            arg_transaction = True
        elif opt == '-r':
            arg_region = arg


    if missing_phone or missing_message or missing_subject:
        main_return['status'] = '999'
        main_return['error'] = HOMECU_SCRIPT_HELP
        print (json.dumps(main_return))
        sys.exit(2)


    # ########
    # SANITIZE

    # PHONE NUMBER
    if len(arg_phone) != 11 or not arg_phone.isdigit():
        main_return['status'] = '999'
        main_return['error'] = "Unsupported phone number"
        print (json.dumps(main_return))
        sys.exit(2)
    else:
        sms_phone_nbr = arg_phone


    # CONTINUE

    # SET SMS TYPE
    if arg_transaction:
        homecu_sms_type = 'Transactional'
    else:
        homecu_sms_type = 'Promotional'

    # SETTINGS
    # ###########

    try:
        # OPEN boto3 client
        client = boto3.client('sns', region_name=AWS_REGION)

        # DefaultSMSType 
          # Transactional (Critical for MFA)
          # Promotional (Default)
        response = client.set_sms_attributes(
            attributes={
                'DefaultSMSType': homecu_sms_type
            }
        )
    
        # ###############
        # PUBLISH MESSAGE 
        response = client.publish(
            PhoneNumber = sms_phone_nbr,
            Message = sms_message,
            Subject = sms_subject
         )
    except Exception as e:
        main_return['status'] = '999'
        main_return['error'] = e
        print (json.dumps(main_return))
        sys.exit(2)

    # SUCCESS
    main_return['response'] = response
    print (json.dumps(main_return));

if __name__ == "__main__":
    main(sys.argv[1:])