alexis2/web.py

#!/usr/bin/env python3

from flask import Flask, abort, jsonify, request
from db import db
from linklist import LinkList

app = Flask(__name__)
d = db()


BASEPATH = "/pymnts"


@app.route(BASEPATH + '/source/add/', methods=['POST'])
def add_source():
    selector = None
    rss = False
    json = request.get_json()
    if "selector" in json:
        selector = json["selector"]
    if "rss" in json:
        rss = json["rss"]

    if "category" not in json:
        json["category"] = 1

    json["category"] = int(json["category"])
    d.add_source(json["source"], rss, selector, json["name"], json["category"])
    return jsonify({'ok': "ok"})


@app.route(BASEPATH + '/source/test/', methods=['POST'])
def test_source():
    selector = None
    rss = False
    json = request.get_json()
    if "selector" in json and len(json["selector"]) > 0:
        selector = json["selector"]
    if "rss" in json:
        rss = json["rss"]

    l = LinkList(json["source"], selector, rss=rss)
    return jsonify({'links': l.links})


@app.route(BASEPATH + '/articles/', methods=['POST'])
@app.route(BASEPATH + '/articles', methods=['POST'])
def filter_articles():
    return jsonify(d.filter_articles(request.get_json(), limit=15))


@app.route(BASEPATH + '/keywords/', methods=['GET'])
@app.route(BASEPATH + '/keywords', methods=['GET'])
def get_keywords():
    return jsonify({'keywords': d.get_keywords_list()})  # limit


@app.route(BASEPATH + '/keywords/', methods=['POST'])
@app.route(BASEPATH + '/keywords',  methods=['POST'])
def update_keywords(kw, _type):
    if _type == 'ADD':
        res = db.add_keyword(kw)
        return jsonify({})
    elif _type == 'DEL':
        res = db.delete_keyword(kw)
        return jsonify({'keywords': d.get_keywords_list()})
    elif _type == 'UPD':
        res = db.update_keyword(kw)
        return jsonify({'keywords': d.get_keywords_list()})


@app.route(BASEPATH + '/articles/', methods=['GET'])
@app.route(BASEPATH + '/articles', methods=['GET'])
def get_articles():
    return jsonify(d.filter_articles({}, limit=15))


@app.route(BASEPATH + '/article/delete/<string:art_id>', methods=['GET'])
def delete_article(art_id):
    if not d.delete_article(art_id):
        abort(401)
    return jsonify({'ok': "ok"})


@app.route(BASEPATH + '/article/<string:art_id>', methods=['GET'])
def get_article(art_id):
    art = d.article(art_id)
    if art is None:
        abort(401)
    return jsonify({'articles': art})


@app.route(BASEPATH + '/sources/', methods=['GET'])
@app.route(BASEPATH + '/sources', methods=['GET'])
def get_sources():
    s = sorted(d.sources(), key=lambda k: k['name'].lower())
    return jsonify({'sources': s})


@app.route(BASEPATH + '/sources/<string:source_id>', methods=['GET'])
def get_source(source_id):
    source = d.sources(source_id)
    if source is None:
        abort(401)
    return jsonify({'source': source})


@app.route(BASEPATH + '/add_source', methods=['POST'])
def create_task():
    if not request.json or 'title' not in request.json:
        abort(400)
    task = {
        'id': tasks[-1]['id'] + 1,
        'title': request.json['title'],
        'description': request.json.get('description', ""),
        'done': False
    }
    tasks.append(task)
    return jsonify({'task': task}), 201


if __name__ == '__main__':
    app.run(host="0.0.0.0", debug=False)