alexis2/web.py

#!/usr/bin/env python3
from flask import Flask, abort, jsonify, request
from json import loads

from db import db
from linklist import LinkList

app = Flask(__name__)
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 + '/keywordsw/', methods=['GET'])
@app.route(BASEPATH + '/keywordsw', methods=['GET'])
def get_keywordsw():
    return jsonify({'keywords': d.get_keywords_weight_list()})  # limit


@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 add__keyword():
    json = request.get_json()
    if 'kw' not in json or 'weight' not in json:
        print(json)
        abort(400)
    if not d.add_keyword(json['kw'], json['weight']):
        print("cant add")
        abort(400)
    return jsonify({'keywords': d.get_keywords_weight_list()})


@app.route(BASEPATH + '/keywords/', methods=['PUT'])
@app.route(BASEPATH + '/keywords',  methods=['PUT'])
def update_keyword():
    json = request.get_json()
    if 'kw' not in json:
        abort(400)
    try:
        if type(json['kw']) == dict:
            kw = json['kw']
        else:
            kw = loads(json['kw'])
    except Exception as e:
        print(json['kw'])
        print("cant parse json")
        print(e)
        abort(400)

    print(kw)
    if not d.update_keyword(kw):
        abort(400)
    return jsonify({'keywords': d.get_keywords_weight_list()})


@app.route(BASEPATH + '/keywords/', methods=['DELETE'])
@app.route(BASEPATH + '/keywords',  methods=['DELETE'])
def delete_keyword():
    json = request.get_json()
    if json is None or 'kw' not in json:
        abort(400)
    kw = json['kw']
    if not d.delete_keyword(kw):
        abort(400)
    return jsonify({'keywords': d.get_keywords_weight_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__':
    d = db()
    #app.run(host="0.0.0.0", debug=True)
    app.run(host="0.0.0.0", debug=False)