-
Notifications
You must be signed in to change notification settings - Fork 27
/
demo.py
84 lines (65 loc) · 2.29 KB
/
demo.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
import sys
import random
from flask import Flask, render_template, jsonify
from flask_sqlalchemy import SQLAlchemy
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///test.db'
db = SQLAlchemy(app)
BASECOORDS = [-13.9626, 33.7741]
class Point(db.Model):
id = db.Column(db.Integer, primary_key=True)
latitude_off = db.Column(db.Float)
longitude_off = db.Column(db.Float)
district_id = db.Column(db.Integer, db.ForeignKey('district.id'))
district = db.relationship("District")
def __init__(self, id, district, lat, lng):
self.id = id
self.district = district
self.latitude_off = lat
self.longitude_off = lng
def __repr__(self):
return "<Point %d: Lat %s Lng %s>" % (self.id, self.latitude_off, self.longitude_off)
@property
def latitude(self):
return self.latitude_off + self.district.latitude
@property
def longitude(self):
return self.longitude_off + self.district.longitude
class District(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(80))
latitude = db.Column(db.Float)
longitude = db.Column(db.Float)
def __init__(self, id, name, lat, lng):
self.id = id
self.name = name
self.latitude = lat
self.longitude = lng
@app.route('/')
def index():
districts = District.query.all()
return render_template('index.html', districts=districts)
@app.route('/district/<int:district_id>')
def district(district_id):
points = Point.query.filter_by(district_id=district_id).all()
coords = [[point.latitude, point.longitude] for point in points]
return jsonify({"data": coords})
def make_random_data(db):
NDISTRICTS = 5
NPOINTS = 10
for did in range(NDISTRICTS):
district = District(did, "District %d" % did, BASECOORDS[0], BASECOORDS[1])
db.session.add(district)
for pid in range(NPOINTS):
lat = random.random() - 0.5
lng = random.random() - 0.5
row = Point(pid + NPOINTS * did, district, lat, lng)
db.session.add(row)
db.session.commit()
if __name__ == '__main__':
if len(sys.argv) > 1:
if sys.argv[1] == 'mkdb':
db.create_all()
make_random_data(db)
else:
app.run(debug=True)