custom_alert.py

#!/usr/bin/python
'''
Creates a custom Securitybot alert for a specified user.
'''
import argparse
from securitybot.sql import SQLEngine
from securitybot.util import create_new_alert

from typing import Any

def main(args):
    # type: (Any) -> None
    SQLEngine('localhost', 'root', '', 'securitybot')

    create_new_alert('custom_alert', args.name[0], args.title[0], args.reason[0])

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Send a custom Securitybot alert')

    parser.add_argument('-n', '--name', dest='name', nargs=1, required=True,
                        help='Username to send alert to')
    parser.add_argument('-t', '--title', dest='title', nargs=1, required=True,
                        help='User-visible alert title')
    parser.add_argument('-r', '--reason', dest='reason', nargs=1,
                        help='Long-form reason for the alert to provided context to user.')

    args = parser.parse_args()
    main(args)