Add Azure Service Bus message queueing

Send a smaller payload (name, asset_id and location) to Azure Service Bus
queue.

ShadowCat/FlaskCat/FlaskCat/config.py

@@ -23,3 +23,11 @@
 DB_COLL_DEVICES = DB_DATABASE.httpDevices
 DB_COLL_PINGS = DB_DATABASE.pings
 DB_COLL_HISTORY = DB_DATABASE.history
+
+# Azure Service Bus configurations
+SVC_BUS_NAMESPACE = 'pyzure'
+SVC_BUS_ACCESS_KEY_NAME = 'RootManageSharedAccessKey'
+SVC_BUS_ACCESS_KEY_VALUE = 'BhSjDnj6xsHC5tmd73qRlZKdHYWdM6RM3YyF028hPH4='
+QUEUE_NAME = 'taskqueue'
+QUEUE_MAX_SIZE = '5120'
+QUEUE_MSG_TTL = 'PT1M'

ShadowCat/FlaskCat/FlaskCat/ping_service.py

@@ -3,6 +3,7 @@
 from models import User
 from bson.json_util import dumps
 from datetime import datetime
+from servicebus_queue import ServiceBusQueue
 import pymongo
 import utilities
 import logging
@@ -11,6 +12,14 @@
 CORS(app)
 app.config.from_object('config')
 
+# azure servicebus queue
+svc = ServiceBusQueue(
+    app.config['SVC_BUS_NAMESPACE'],
+    app.config['SVC_BUS_ACCESS_KEY_NAME'],
+    app.config['SVC_BUS_ACCESS_KEY_VALUE'],
+    app.config['QUEUE_NAME']
+)
+
 
 @app.route('/api/ping', methods=['POST'])
 def ping_location():
@@ -23,6 +32,8 @@ def ping_location():
     data = get_payload(json_data)
     add_ping(data)
     add_history(data)
+    payload = get_azure_payload(data)
+    svc.send(payload)
 
     return dumps(''), 201, {'Content-Type': 'application/json'}
 
@@ -40,6 +51,14 @@ def get_payload(raw_data):
     )
 
 
+def get_azure_payload(data):
+    return {
+        'name': data.name,
+        'asset_id': data.asset_id,
+        'location': data.point
+    }
+
+
 def add_ping(data):
     try:
         app.config['DB_COLL_PINGS'].update_one(

ShadowCat/FlaskCat/FlaskCat/servicebus_queue.py

@@ -0,0 +1,33 @@
+from azure.servicebus import ServiceBusService, Message, Queue
+import ast
+
+
+class ServiceBusQueue(object):
+    def __init__(self,
+                 namespace,
+                 access_key_name,
+                 access_key_value,
+                 q_name,
+                 q_max_size='5120',
+                 msg_ttl='PT1M'):
+        self.bus_svc = ServiceBusService(
+            service_namespace=namespace,
+            shared_access_key_name=access_key_name,
+            shared_access_key_value=access_key_value
+        )
+        self.queue_options = Queue()
+        self.queue_options.max_size_in_megabytes = q_max_size
+        self.queue_options.default_message_time_to_live = msg_ttl
+
+        self.queue_name = q_name
+        self.bus_svc.create_queue(self.queue_name, self.queue_options)
+
+    def send(self, msg):
+        message = bytes(msg)
+        message = Message(message)
+        self.bus_svc.send_queue_message(self.queue_name, message)
+
+    def receive(self):
+        msg = self.bus_svc.receive_queue_message(self.queue_name, peek_lock=False)
+        data = ast.literal_eval(msg.body)
+        return data