1
from flask import Flask
2
from flask import request
3
from flask import render_template
4
from flask import make_response
5
from flask import jsonify
6
import logging
7
from logging import FileHandler
8
import pymongo
9
import os
10
import lxml.html
11
import urllib2
12
import StringIO
13
import json
14
15
16
app = Flask(__name__)
17
18
19
@app.route('/', methods=['GET'])
20
def index():
21
    if request.args.has_key('url'):
22
        return render_template('index.html', url=request.args['url'])
23
    else:
24
        return render_template('index.html')
25
26
27
@app.route('/fetch', methods=['GET'])
28
def fetch():
29
    connection = pymongo.Connection()
30
    db = connection['mural']
31
    collection = db['data']
32
    ret = {}
33
    x = 0
34
    resource = "default"
35
    if request.args.has_key('uri'):
36
        resource = request.args['uri']
37
    for i in collection.find({'uri':resource}):
38
        del(i['_id'])
39
        ret[x] = i
40
        x = x + 1
41
    if len(ret) == 0:
42
        ret['error'] = "Sorry! No re-treats for you."
43
    return jsonify(ret)
44
45
46
@app.route('/search', methods=['GET'])
47
def search():
48
    connection = pymongo.Connection()
49
    db = connection['mural']
50
    collection = db['data']
51
    y = 0
52
    ret = {}
53
    keywords_dict = json.loads(request.args['data'])
54
    keywords = json.loads(keywords_dict)['data']
55
    for i in collection.find():
56
        for keyword in keywords:
57
            print keyword
58
            try:
59
                if keyword in i['nodes']:
60
                    del(i['_id'])
61
                    ret[y] = i
62
                    y = y + 1
63
            except:
64
                pass
65
    return render_template('blank.html', content = ret)
66
67
68
@app.route('/submit', methods=['POST'])
69
def submit():
70
    c = pymongo.Connection()
71
    db = c['mural']
72
    coll = db['data']
73
    requestData = json.loads(request.form['data'])
74
    try:
75
        for i in requestData:
76
            coll.insert(i)
77
        response = make_response()
78
        response.status = '200 OK'
79
        response.status_code = 200
80
        return response
81
    except:
82
        response = make_response()
83
        response.status = "500"
84
        response.data = "Your post could not be saved. Try posting again."
85
        return response
86
87
88
@app.route('/SWeeText', methods=['GET'])
89
def SWeeText():
90
    if request.args.has_key('url'):
91
        myhandler1 = urllib2.Request(request.args['url'], headers={'User-Agent': "Mozilla/5.0(X11; U; Linux i686) Gecko/20071127 Firefox/2.0.0.11"})
92
        a = urllib2.urlopen(myhandler1)
93
        page = a.read()
94
        a.close()
95
        try:
96
            page = unicode(page, 'utf-8')
97
        except UnicodeDecodeError:
98
            pass
99
        root = lxml.html.parse(StringIO.StringIO(page)).getroot()
100
        root.make_links_absolute(request.args['url'], resolve_base_href = True)
101
102
        # inject the JS toolbar to annotate text
103
        script = root.makeelement('script')
104
        script.set('src', 'static/text-annotation.js')
105
        link = root.makeelement('link')
106
        link.set('href', 'static/text-annotation.css')
107
        link.set('type', 'text/css')
108
        link.set('rel', 'stylesheet')
109
110
        root.body.append(script)
111
        root.head.append(link)
112
113
        return lxml.html.tostring(root)
114
115
116
#Log the errors, don't depend on apache to log it for you.
117
    fil = FileHandler(os.path.join(os.path.dirname(__file__), 'logme'),mode='a')
118
    fil.setLevel(logging.ERROR)
119
    app.logger.addHandler(fil)
120
121
122
if __name__ == "__main__":
123
    app.run(debug=True, host='0.0.0.0')