add two examples to the defense folder

jgarman · jgarman · commit 00e8d3fa2a81 · 2017-06-14T14:06:08.000-04:00
diff --git a/examples/defense/move_sensor.py b/examples/defense/move_sensor.py
@@ -0,0 +1,39 @@
+#!/usr/bin/env python
+
+import sys
+from cbapi.defense.models import Device
+from cbapi.example_helpers import build_cli_parser, get_cb_defense_object
+
+
+def main():
+    parser = build_cli_parser("Move a device into a new security policy")
+    device_options = parser.add_mutually_exclusive_group(required=True)
+    device_options.add_argument("-i", "--id", type=int, help="Device ID of sensor to move")
+    device_options.add_argument("-n", "--hostname", help="Hostname to move")
+
+    policy_options = parser.add_mutually_exclusive_group(required=True)
+    policy_options.add_argument("--policyid", type=int, help="Policy ID")
+    policy_options.add_argument("--policyname", help="Policy name")
+
+    args = parser.parse_args()
+    cb = get_cb_defense_object(args)
+
+    if args.id:
+        devices = [cb.select(Device, args.id)]
+    else:
+        devices = list(cb.select(Device).where("hostNameExact:{0}".format(args.hostname)))
+
+    for device in devices:
+        if args.policyid:
+            destpolicy = int(args.policyid)
+            device.policyId = int(args.policyid)
+        else:
+            destpolicy = args.policyname
+            device.policyName = args.policyname
+
+        device.save()
+        print("Moved device id {0} (hostname {1}) into policy {2}".format(device.deviceId, device.name, destpolicy))
+
+
+if __name__ == "__main__":
+    sys.exit(main())
diff --git a/examples/defense/notifications.py b/examples/defense/notifications.py
@@ -0,0 +1,21 @@
+#!/usr/bin/env python
+
+import sys
+from cbapi.example_helpers import build_cli_parser, get_cb_defense_object
+import json
+
+
+def main():
+    parser = build_cli_parser("Listen to real-time notifications")
+    parser.add_argument("-s", type=int, help="# of seconds to sleep between polls", default=30)
+
+    args = parser.parse_args()
+    cb = get_cb_defense_object(args)
+
+    while True:
+        for notification in cb.notification_listener(args.s):
+            print(json.dumps(notification, indent=2))
+
+
+if __name__ == "__main__":
+    sys.exit(main())
diff --git a/src/cbapi/example_helpers.py b/src/cbapi/example_helpers.py
@@ -1,6 +1,7 @@
 import argparse
 from cbapi.response import CbEnterpriseResponseAPI
 from cbapi.protection import CbEnterpriseProtectionAPI
+from cbapi.defense import CbDefenseAPI
 import codecs
 import sys
 
@@ -57,6 +58,21 @@ def get_cb_protection_object(args):
     return cb
 
 
+def get_cb_defense_object(args):
+    if args.verbose:
+        import logging
+        logging.basicConfig()
+        logging.getLogger("cbapi").setLevel(logging.DEBUG)
+        logging.getLogger("__main__").setLevel(logging.DEBUG)
+
+    if args.cburl and args.apitoken:
+        cb = CbDefenseAPI(url=args.cburl, token=args.apitoken, ssl_verify=(not args.no_ssl_verify))
+    else:
+        cb = CbDefenseAPI(profile=args.profile)
+
+    return cb
+
+
 def get_object_by_name_or_id(cb, cls, name_field="name", id=None, name=None, force_init=True):
     clsname = cls.__name__
     try:
