Browse Source

test+add sources

David 10 years atrás
parent
commit
3f6b80f0c2
1 changed files with 27 additions and 0 deletions
  1. 27 0
      web.py

+ 27 - 0
web.py

@@ -3,6 +3,7 @@
 from flask import Flask,abort,jsonify,request
 from db import db
 import json
+from linklist import LinkList
 
 app = Flask(__name__)
 d = db()
@@ -13,6 +14,32 @@ BASEPATH="/pymnts"
 def index():
     return "Hello, World!"
 
+@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"]
+
+    d.add_source(json["source"],rss,selector,json["name"])
+    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():