feed_action_add.py

__author__ = 'bwolfson'

import sys
import optparse

# in the github repo, cbapi is not in the example directory
sys.path.append('../src/cbapi')

import cbapi

def build_cli_parser():
    parser = optparse.OptionParser(usage="%prog [options]", description="Add an action to a feed on the CB server")

    # for each supported output type, add an option
    #
    parser.add_option("-c", "--cburl", action="store", default=None, dest="server_url",
                      help="CB server's URL.  e.g., http://127.0.0.1 ")
    parser.add_option("-a", "--apitoken", action="store", default=None, dest="token",
                      help="API Token for Carbon Black server")
    parser.add_option("-n", "--no-ssl-verify", action="store_false", default=True, dest="ssl_verify",
                      help="Do not verify server SSL certificate.")
    parser.add_option("-i", "--id", action = "store", default = None, dest = "id",
                      help = "id of the feed")
    parser.add_option("-t", "--action_type", action = "store", default = None, dest = "action_type_id",
                      help = "id for the type of action to enable: Type 0 for email, 1 to write to "
                             "syslog, and 3 to create alert")
    parser.add_option("-e", "--email_recipients", action = "store", default = None, dest = "email_recipients",
                      help = "list of user ids of email recipients")
    return parser

def main(argv):
    parser = build_cli_parser()
    opts, args = parser.parse_args(argv)
    if not opts.server_url or not opts.token or not opts.id or not opts.action_type_id or not opts.email_recipients:
      print "Missing required param; run with --help for usage"
      sys.exit(-1)

    # build a cbapi object
    #
    cb = cbapi.CbApi(opts.server_url, token=opts.token, ssl_verify=opts.ssl_verify)

    #Check to make sure the user supplies a correct action_type_id
    type = opts.action_type_id
    if not (int(type) == int(0) or int(type) == int(1) or int(type) == int(3)):
        print "action_type_id must be either 0,1,or 3"
        sys.exit(-1)

    #Check to make sure the action isn't already enabled
    curr_actions = cb.feed_action_enum(opts.id)
    for action in curr_actions:
        if int(action['action_type']) == int(opts.action_type_id):
            print "action already enabled"
            sys.exit(-1)
        else:
            continue

    action = cb.feed_action_add(opts.id, opts.action_type_id, opts.email_recipients)

    for key in action.keys():
        print "%-20s : %s" % (key, action[key])

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