Initial commit

3 files changed, 200 insertions, 0 deletions

diff --git a/inventory/__init__.py b/inventory/__init__.py
new file mode 100644
index 0000000..7c12b51
--- /dev/null
+++ b/inventory/__init__.py
@@ -0,0 +1,13 @@
+from flask import Flask
+
+app = Flask(__name__)
+app.config.from_pyfile('inventory.cfg')
+
+with app.app_context():
+    from .api import app as api
+    app.register_blueprint(api, url_prefix='/api')
+
+
+@app.route('/')
+def index():
+    return 'foo'
diff --git a/inventory/api.py b/inventory/api.py
new file mode 100644
index 0000000..0fde577
--- /dev/null
+++ b/inventory/api.py
@@ -0,0 +1,147 @@
+import datetime
+
+from flask import Blueprint, jsonify, current_app, request, abort, url_for
+from flask_pymongo import PyMongo
+from marshmallow import ValidationError
+import pymongo
+import pytz
+
+from .schema import NodeSchema
+
+# Setup PyMongo
+mongo = PyMongo(current_app, tz_aware=True)
+mongo.db.nodes.create_index([('fields.value', pymongo.TEXT), ('name', pymongo.TEXT)], name='fields.value_text_name_text')
+mongo.db.nodes.create_index([('parent_id', pymongo.ASCENDING)], name='parent_id')
+
+app = Blueprint('api', __name__)
+
+
+# Error handling
+@app.errorhandler(400)
+def on_invalid_request(e):
+    return jsonify({'message': e.description or 'Invalid request'}), 400
+
+
+@app.errorhandler(404)
+def on_not_found(e):
+    return jsonify({'message': e.description or 'Not found'}), 404
+
+
+@app.errorhandler(500)
+def on_internal_error(e):
+    return jsonify({'message': e.description or 'Invalid request'}), 500
+
+
+@app.errorhandler(ValidationError)
+def on_validation_error(e):
+    return jsonify({'message': 'Validation error', 'fields': e.messages}), 400
+
+
+# Routes
+@app.route('/nodes')
+def root_nodes():
+    schema = NodeSchema(many=True)
+    data = schema.dump(mongo.db.nodes.find({'parent_id': None}))
+    return jsonify(data)
+
+
+@app.route('/nodes', methods=['POST'])
+def add_node():
+    data = request.json
+    if data is None or not isinstance(data, dict):
+        abort(400, 'Payload must be a JSON object')
+    schema = NodeSchema()
+    node = schema.load(data)
+    node['created_at'] = pytz.utc.localize(datetime.datetime.utcnow())
+    result = mongo.db.nodes.insert_one(node)
+    if not result.acknowledged:
+        abort(500, 'Write operation not acknowledged')
+    node_id = result.inserted_id
+    return jsonify({'id': str(node_id), 'url': url_for('.node', node_id=node_id)}), 201
+
+
+@app.route('/nodes/<ObjectId:node_id>')
+def node(node_id):
+    result = mongo.db.nodes.aggregate([
+        {'$match': {'_id': node_id}},
+        {
+            '$graphLookup': {
+                'from': 'nodes',
+                'startWith': '$parent_id',
+                'connectFromField': 'parent_id',
+                'connectToField': '_id',
+                'as': 'parents',
+            },
+        },
+        {
+            '$graphLookup': {
+                'from': 'nodes',
+                'startWith': '$_id',
+                'connectFromField': '_id',
+                'connectToField': 'parent_id',
+                'as': 'children',
+                'maxDepth': 1,
+            },
+        },
+    ])
+    try:
+        node = result.next()
+    except StopIteration:
+        abort(404, 'No node found')
+
+    parents = dict((parent['_id'], parent) for parent in node.pop('parents'))
+    current = node
+    # Recursively assign each node their respective parent
+    while current:
+        # NOTE: Python assigns from left to right
+        current['parent'] = current = parents.get(current.get('parent_id'))
+
+    children = dict((child['_id'], child) for child in node.pop('children'))
+    children_list = node['children'] = []
+    for child in children.values():
+        # Direct children should be assigned to the top-most list
+        if child['parent_id'] == node['_id']:
+            children_list.append(child)
+            continue
+        # Otherwise assign to their respective parent
+        parent = children[child['parent_id']]
+        parent.setdefault('children', []).append(child)
+
+    schema = NodeSchema()
+    return jsonify(schema.dump(node))
+
+
+@app.route('/nodes/<ObjectId:node_id>', methods=['PUT'])
+def update_node(node_id):
+    data = request.json
+    if data is None or not isinstance(data, dict):
+        abort(400, 'Payload must be a JSON object')
+    schema = NodeSchema()
+    node = schema.load(data)
+    node['updated_at'] = pytz.utc.localize(datetime.datetime.utcnow())
+    result = mongo.db.nodes.update_one({'_id': node_id}, {'$set': node})
+    if not result.acknowledged:
+        abort(500, 'Write operation not acknowledged')
+    return '', 204
+
+
+@app.route('/nodes/<ObjectId:node_id>', methods=['DELETE'])
+def delete_node(node_id):
+    result = mongo.db.nodes.delete_one({'_id': node_id})
+    if result.deleted_count == 0:
+        abort(404, 'No node found')
+    return jsonify({}), 204
+
+
+@app.route('/search', methods=['POST'])
+def find_nodes():
+    if 'q' not in request.form:
+        abort(400, 'Missing q argument')
+    schema = NodeSchema(many=True)
+    data = schema.dump(mongo.db.nodes.find({'$text': {'$search': request.form['q']}}))
+    return jsonify(data)
+
+
+@app.route('/')
+def index():
+    return 'api'
diff --git a/inventory/schema.py b/inventory/schema.py
new file mode 100644
index 0000000..4837d58
--- /dev/null
+++ b/inventory/schema.py
@@ -0,0 +1,40 @@
+import bson
+from flask import url_for
+from marshmallow import Schema, fields
+
+# Map ObjectId to String
+Schema.TYPE_MAPPING[bson.ObjectId] = fields.String
+
+
+class ObjectId(fields.Field):
+    def _serialize(self, value, attr, obj, **kwargs):
+        if value is None:
+            return value
+        return str(value)
+
+    def _deserialize(self, value, attr, data, **kwargs):
+        if value is None:
+            return value
+        return bson.ObjectId(value)
+
+
+class FieldSchema(Schema):
+    name = fields.String(required=True)
+    value = fields.String(required=True)
+    # type
+
+
+class NodeSchema(Schema):
+    _id = ObjectId(dump_only=True, data_key='id')
+    name = fields.String(required=True)
+    parent_id = ObjectId(default=None)
+    _fields = fields.List(fields.Nested(FieldSchema()), default=[], attribute='fields', data_key='fields')
+
+    # These are not set by the caller, but by the API endpoint
+    created_at = fields.AwareDateTime(dump_only=True)
+    updated_at = fields.AwareDateTime(dump_only=True)
+
+    # Not actual stored fields
+    parent = fields.Nested(lambda: NodeSchema(), dump_only=True)
+    children = fields.List(fields.Nested(lambda: NodeSchema()), dump_only=True)
+    url = fields.Function(lambda obj: url_for('.node', node_id=obj['_id']), dump_only=True)