app.py

import ast
import os
from flask_caching import Cache
from flask_pymongo import PyMongo
from flask import Flask, request, jsonify, render_template

app = Flask(__name__)
REDIS_HOST = os.environ.get('REDIS_HOST', 'localhost')
MONGO_HOST = os.environ.get('MONGO_HOST', 'localhost')
app.config["MONGO_URI"] = "mongodb://{mongo_host}:27017/flask".format(mongo_host=MONGO_HOST)
app.config['MONGO_DBNAME'] = 'dashboard'
mongo = PyMongo(app)
cache = Cache(app, config={'CACHE_TYPE': 'redis',
                           'CACHE_REDIS_URL': 'redis://{redis_host}:6379/0'.format(redis_host=REDIS_HOST)})


@app.route('/message', methods=["POST"])
@cache.cached()
def message():
    data = request.args
    if request.method == 'POST':
        if data['text']:
            res = mongo.db.dashboard.insert_one(dict(data))
            # return render_template("index.html", messages='Tag inserted successfully % s ' % res)
            return jsonify({'ok': True, 'message': 'Message created successfully % s ' % res.inserted_id}), 200
        else:
            return jsonify({'ok': False, 'message': 'Text should present', 'data': data['text']}), 400


@app.route('/tag/<ObjectId:message_id>', methods=['POST'])
@cache.cached()
def add_tag_to_message(message_id):
    data = request.args
    if request.method == 'POST':
        if data['text']:
            res = mongo.db.dashboard.update_one({"_id": message_id}, {"$addToSet": {"tags": dict(data)}})
            return jsonify({'ok': True, 'message': 'Tag inserted successfully % s ' % res}), 200
        else:
            return jsonify({'ok': False, 'message': 'Text should present', 'data': data}), 400
    return jsonify({'ok': False, 'message': 'Text should present', 'data': data}), 400


@app.route('/comment/<ObjectId:message_id>', methods=['POST'])
@cache.cached()
def add_comment_to_message(message_id):
    data = request.args
    if request.method == 'POST':
        if data['text']:
            res = mongo.db.dashboard.update_one({"_id": message_id}, {"$addToSet": {"comments": dict(data)}})
            return jsonify({'ok': True, 'message': 'Comment inserted successfully % s ' % res}), 200
        else:
            return jsonify({'ok': False, 'message': 'Text should present'}), 400


@app.route('/message/<ObjectId:message_id>', methods=['GET'])
@cache.cached()
def message_by_id(message_id):
    if request.method == 'GET':
        res = mongo.db.dashboard.find_one_or_404(message_id)
        return jsonify({'ok': True, 'message': 'Message found  % s' % res}), 200


@app.route('/stats/<ObjectId:message_id>', methods=['GET'])
@cache.cached()
def stats_by_id(message_id):
    if request.method == 'GET':
        res = mongo.db.dashboard.find_one_or_404(message_id)
        tags = 0
        comments = 0
        if len(res["comments"]) != 0:
            comments = len(res["comments"])
        if len(res["tags"]) != 0:
            tags = len(res["tags"])
        return jsonify({
            'ok': True,
            'message': 'This message has {tags} tags and {comments} comments'.format(
                tags=tags, comments=comments
            ),
            'comments': res['comments'],
            'tags': res['tags']
        }), 200


if __name__ == '__main__':
    app.run(debug=True, host='0.0.0.0', port=5000)