forked from niryariv/opentaba-server
/
app.py
187 lines (148 loc) · 6.02 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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
# -*- coding: utf-8 -*-
#!/usr/bin/python
import os
import datetime
import json
from bson import json_util
from werkzeug.contrib.atom import AtomFeed
from werkzeug.urls import url_encode
from flask import Flask
from flask import abort, make_response, request
from lib.conn import *
from lib.cache import cached, _setup_cache
import lib.helpers as helpers
app = Flask(__name__)
app.debug = RUNNING_LOCAL # if we're local, keep debug on
#### Cache Helper ####
@app.before_first_request
def setup_cache():
"""
We initialize the cache here for the first time because __main__ is not run when we launch
within a WSGI container.
"""
_setup_cache(app)
#### ROUTES ####
@app.route('/gushim.json')
@cached(app, timeout=3600)
def get_gushim():
"""
get gush_id metadata
"""
detailed = request.args.get('detailed', '') == 'true'
gushim = helpers._get_gushim(fields={'gush_id': True, 'last_checked_at': True, '_id': False})
if detailed:
# Flatten list of gushim into a dict
g_flat = dict((g['gush_id'], {"gush_id": g['gush_id'],
"last_checked_at": g['last_checked_at'],
"plan_stats": {}}) for g in gushim)
# Get plan statistics from DB
stats = helpers._get_plan_statistics()
# Merge stats into gushim dict
for g in stats['result']:
try:
gush_id = g['_id']['gush_id']
status = g['_id']['status']
g_flat[gush_id]['plan_stats'][status] = g['count']
except KeyError, e:
# Gush has plans but is missing from list of gushim?
app.logger.warn("Gush #%d has plans but is not listed in the Gushim list", gush_id)
app.log_exception(e)
# De-flatten our dict
gushim = g_flat.values()
return helpers._create_response_json(gushim)
@app.route('/gush/<gush_id>.json')
@cached(app, timeout=3600)
def get_gush(gush_id):
"""
get gush_id metadata
"""
gush = helpers._get_gushim(query={"gush_id": gush_id})
if gush is None or len(gush) == 0:
abort(404)
return helpers._create_response_json(gush[0])
@app.route('/gush/<gushim>/plans.json')
@cached(app, timeout=3600)
def get_plans(gushim):
"""
Get JSON plan data for one or more gush IDs.
The URL format for multiple gushim is something like /gush/12340,12350,12360/plans.json
"""
gushim = gushim.split(',')
if len(gushim) > 1:
gushim_query = {'gushim': {'$in': gushim}, 'year': {'$gt': datetime.datetime.now().year - 10}}
else:
gushim_query = {'gushim': gushim[0]}
return helpers._create_response_json(helpers._get_plans(query=gushim_query))
@app.route('/recent.json')
@cached(app, timeout=3600)
def get_recent_plans():
"""
Get the 10 most recent plans to show on the site's home page
"""
return helpers._create_response_json(helpers._get_plans(count=10))
@app.route('/plans.atom')
@cached(app, timeout=3600)
def atom_feed():
muni_name = os.environ.get('MUNICIPALITY_NAME', '').decode('utf-8')
title = u'תב"ע פתוחה - %s' % muni_name if muni_name else u'תב"ע פתוחה'
return helpers._create_response_atom_feed(request, helpers._get_plans(count=20), feed_title=title).get_response()
@app.route('/gush/<gushim>/plans.atom')
@cached(app, timeout=3600)
def atom_feed_gush(gushim):
"""
Create a feed for one or more gush IDs.
The URL format for multiple gushim is something like /gush/12340,12350,12360/plans.atom
"""
gushim = gushim.split(',')
if len(gushim) > 1:
gushim_query = {'gushim': {'$in': gushim}}
else:
gushim_query = {'gushim': gushim[0]}
muni_name = os.environ.get('MUNICIPALITY_NAME', '').decode('utf-8')
title = u'תב"ע פתוחה - %s - גוש %s' % (muni_name, ', '.join(gushim)) if muni_name else u'תב"ע פתוחה - גוש %s' % ', '.join(gushim)
return helpers._create_response_atom_feed(request, helpers._get_plans(query=gushim_query), feed_title=title).get_response()
@app.route('/plans/search/<path:plan_name>')
@cached(app, timeout=3600)
def find_plan(plan_name):
"""
Find plans that contain the search query and return a json array of their plan and gush ids
"""
return helpers._create_response_json(helpers._get_plans(count=3, query={'number': {'$regex': '.*%s.*' % plan_name}}))
@app.route('/plan/<plan_id>/mavat')
@cached(app, timeout=3600)
def redirect_to_mavat(plan_id):
"""
If we have a mavat code for the given plan redirect the client to the plan's page on the
mavat website using an auto-sending form
"""
try:
plans = helpers._get_plans(count=1, query={'plan_id': int(plan_id)})
except ValueError: # plan_id is not an int
abort(400)
except: # DB error
abort(500)
if plans is None or len(plans) == 0 or plans[0]['mavat_code'] == '':
abort(404)
return make_response('<html><body>'
'<form action="http://mavat.moin.gov.il/MavatPS/Forms/SV4.aspx?tid=4" method="post" name="redirect_form">'
'<input type="hidden" name="PL_ID" value="' + plans[0]['mavat_code'] + '">'
'</form><script language="javascript">document.redirect_form.submit();</script></body></html>')
# TODO add some text on the project
@app.route('/')
def hello():
return ('<html><body style="font-size: 3em; margin: 100px; text-align:center">'
'<p>Hi. You\'ve reached the server side of <a href="http://opentaba.info">opentaba.info</a></p>'
'<p><a href="https://github.com/niryariv/opentaba-server">Source</a></p>'
'</body></html>')
@app.route('/wakeup')
def wakeup():
"""
wake up Heroku dyno from idle. perhaps can if >1 dynos
used as endpoint for a "wakeup" request when the client inits
"""
return helpers._create_response_json({'morning': 'good'})
#### MAIN ####
if __name__ == '__main__':
# Bind to PORT if defined, otherwise default to 5000.
port = int(os.environ.get('PORT', 5000))
app.run(host='0.0.0.0', port=port)