-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
119 lines (86 loc) · 3.17 KB
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
# -*- coding: utf-8 -*-
from flask import Flask, jsonify, request, url_for, g, redirect, render_template, flash, session, Response
import json, urllib2, os
from utils import jsonp, safe_url_for
from models import is_safe, send_push, geocode_ip
from cartodb_sql import location_intersects, check_login, get_log
from flask.ext.cache import Cache
from flask.ext.mandrill import Mandrill
from flask.ext.cors import CORS, cross_origin
app = Flask(__name__)
app.config.from_object('settings')
if os.environ.get('REDISTOGO_URL'):
cache = Cache(app,config={'CACHE_TYPE': 'redis', 'CACHE_REDIS_URL': os.environ.get('REDISTOGO_URL')})
else:
cache = Cache(app,config={'CACHE_TYPE': 'simple'})
mandrill = Mandrill(app)
cors = CORS(app)
__VERSION__ = 0.1
@app.before_request
def before():
g.CLIENT_IP = request.access_route[0]
@app.route("/test/push")
def token_push():
send_push(os.environ.get('PUSHBULLET_TOKEN'), 'foo')
return "push sent"
@app.route("/test/ip")
def ip_locate():
ip = g.CLIENT_IP
geocode = cached_geocode_ip(ip)
return jsonify(ip=geocode,
route=request.access_route,
location_intersects=location_intersects(geocode['lat'], geocode['lon']) if 'lon' in geocode.keys() else False)
@app.route('/')
def hello():
data = []
for rule in app.url_map.iter_rules():
if safe_url_for(rule):
data.append([list(rule.methods), rule.endpoint, url_for(rule.endpoint)])
return jsonify(routes=data)
@app.route('/auth', methods=['POST'])
@cross_origin()
@jsonp
def auth():
geocode = cached_geocode_ip(g.CLIENT_IP)
return jsonify(response=is_safe(request.form, g.CLIENT_IP, geocode, mandrill)) # ugh
@app.route('/log', methods=['GET'])
def log():
log = get_log('cyberbank')
return Response(json.dumps(log), mimetype='application/json')
### NOT REALLY USED FROM HERE
@app.route('/config')
def config():
if request.method == 'GET':
return jsonify(data=None)
return jsonify(status='ok')
@app.route('/<company_name>/stats')
def stats(company_name):
log = get_log(company_name)
data = {'c': company_name, 'log': log}
if request.method == 'GET':
return render_template('stats.html', data=data)
@app.route('/login', methods=['GET', 'POST'])
def login():
error = None
client_name = request.form['client']
if request.method == 'POST':
params = check_login(request.form['client'], request.form['username'], request.form['password'])
if not params:
error = 'Invalid credentials'
else:
session['logged_in'] = True
flash('You were logged in')
return redirect(url_for('stats', company_name=request.form['client']))
return render_template('login.html', error=error)
@app.route('/logout')
def logout():
session.pop('logged_in', None)
flash('You were logged out')
return redirect(url_for('show_entries'))
##### END OF REQUESTS NOT USED
@cache.memoize(timeout=36000)
def cached_geocode_ip(ip_address):
return geocode_ip(ip_address)
app.config['DEBUG'] = os.environ.get('DEBUG', False)
if __name__ == '__main__':
app.run(debug=True)