Skip to content
Permalink
Branch: master
Find file Copy path
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
217 lines (168 sloc) 8.7 KB
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
main.py: Serves up the Strava AeroLabs app! Filled with error handlers, page specific calculations, and more!
Initial Implementation: 09/06/2016 [William Golembieski]
Last Modification: 10/17/2016 [William Golembieski]
"""
from flask import Flask, render_template, redirect, request, flash, session, url_for
from get_wind_data import get_wind
from get_strava_segment import get_strava_data
from config import MY_STRAVA_PUBLIC_ACCESS_TOKEN, GOOGLE_API_KEY
from calculate import bearing, wind_angle, wind_assist, crosswind
from polyline import decode
from dbconnect import connection
import wtforms
from passlib.hash import sha256_crypt
from MySQLdb import escape_string as inject_attk_check
import gc
__author__ = "William Golembieski"
__copyright__ = "Copyright 2016, Strava AeroLabs"
__credits__ = ["William Golembieski"]
__license__ = "GNU General Public License v3.0"
__email__ = "BillGolembieski@projectu23.com"
__status__ = "Alpha"
app = Flask(__name__)
# Error handling
# Page not found
@app.errorhandler(404)
def page_not_found(e):
return render_template('404.html', error=e)
# Method not found
@app.errorhandler(405)
def page_not_found(e):
return render_template('405.html', error=e)
# Internal Server Error
@app.errorhandler(500)
def method_not_found(e):
return render_template('500.html', error=e)
# Routing for homepage
@app.route('/')
def index():
return render_template('index.html')
# Routing for strava segment windstat page
@app.route('/windstat', methods=['POST'])
def windstat():
# Receive data from form on windstat. Getting users segment ID
input_segment_url = request.form['segment_url']
# Get strava data from user input in form of JSON. See get_strava_data.py for example JSON response.
strava_data = get_strava_data(input_segment_url, MY_STRAVA_PUBLIC_ACCESS_TOKEN)
# Start / End latitude longitude 'import' and cast as float
start_lat = float(strava_data['start_latitude'])
start_lng = float(strava_data['start_longitude'])
end_lat = float(strava_data['end_latitude'])
end_lng = float(strava_data['end_longitude'])
# UNUSED: Calculate center of segment. Usage intended for map centering.
# center_lat = start_lat - ((start_lat - end_lat)/2)
# center_lng = start_lng - ((start_lng - end_lng)/2)
# Build map from google poly line
google_poly_line = strava_data['map']['polyline']
dict_lat_lng = decode(strava_data['map']['polyline'])
google_img_url = \
'http://maps.googleapis.com/maps/api/staticmap?sensor=false&key=' + GOOGLE_API_KEY + \
'&size=300x300&path=weight:3|color:red|enc:' + google_poly_line
# get wind data for start point. // wind_dict = {'wind mph': wind_val, 'wind direction': wind_dir}
wind_dict = get_wind(start_lat, start_lng)
wind_speed_mph = wind_dict['wind mph']
wind_dir = float(wind_dict['wind direction']) # Account for arrow in /static/images @ 45 deg
# get angle difference
course_bearing = bearing(start_lat, start_lng, end_lat, end_lng)
angle_diff = wind_angle(course_bearing, wind_dir)
# calculate headwind / tailwind component. Negative = headwind
calculated_wind_assist = wind_assist(float(wind_speed_mph), angle_diff)
if calculated_wind_assist > 0:
wind_class = "Tailwind"
elif calculated_wind_assist < 0:
wind_class = "Headwind"
else:
wind_class = "No Wind Assist Detected"
# enumerate intensity of headwind / tailwind
if calculated_wind_assist > 5.0:
# Strong Tailwind
assist_enum = 1
elif 5.0 > calculated_wind_assist >= 0:
# Moderate Tailwind
assist_enum = 2
elif 0 > calculated_wind_assist > -5.0:
# Moderate Headwind
assist_enum = 3
else:
# Strong Headwind
assist_enum = 4
# calculate cross wind component (figure out what negative / positive mean)
calculated_xwind = crosswind(float(wind_speed_mph), angle_diff)
# pass in data to be displayed on /windstat/result ('/windstat/result', data1, data2,....dataN)
return render_template('/windstat.html/', wind_dict=wind_dict,
calculated_wind_assist=abs(calculated_wind_assist),
calculated_xwind=abs(calculated_xwind), google_poly_line=google_poly_line,
strava_data=strava_data, input_segment_id=input_segment_url, angle_diff=angle_diff,
course_bearing=int(course_bearing), dict_lat_lng=dict_lat_lng,
google_img_url=google_img_url,
wind_dir_deg=int(wind_dir), wind_class=wind_class, wind_speed_mph=wind_speed_mph,
assist_enum=assist_enum)
class RegistrationForm(wtforms.Form):
username = wtforms.StringField('Username',
[wtforms.validators.Length(min=4, max=20,
message=
"Please enter a username between 4 and 20 characters long"
)])
email = wtforms.StringField('Email', [wtforms.validators.Length(min=6, max=50),
wtforms.validators.Email(message="Please enter a valid email address")])
password = wtforms.PasswordField('Password', [wtforms.validators.InputRequired(),
wtforms.validators.EqualTo('confirm',
message="Passwords must match")])
confirm = wtforms.PasswordField('Repeat Password')
accept_tos = wtforms.BooleanField('I accept the <a href ="/tos/">Terms of Service</a> and '
'<a href="/privacy/">Privacy Notice</a> Last Updated 12/11/2016',
[wtforms.validators.InputRequired()])
@app.route('/tos')
def terms_of_service():
return render_template('tos.html')
@app.route('/privacy')
def privacy_page():
return render_template('privacy.html')
@app.route('/wundererror')
def wundererror(e):
return render_template('wundererror.html', e)
@app.route('/register')
def register_page():
try:
form = RegistrationForm(request.form)
if request.method == "POST" and form.validate():
# Pull data from html form
username = form.username.data
email = form.email.data
# Immediately encrypt via sha256
password = sha256_crypt.encrypt((str(form.password.data)))
# Connect to database
cursor, conn = connection()
# Using cursor, select a username in database. inject_attk_check() protects against sql injection.
un_attempt = cursor.execute("SELECT * FROM users WHERE username = (%s)",
(inject_attk_check(username)))
# Check to see if username is taken by searching for username in db first.
# If returned value is longer than 0 then the username is already taken.
if len(int(un_attempt)) > 0:
# Call to flask.flash()
flash("That username is already taken, please try another")
render_template('register.html', form=form)
else:
cursor.execute("INSERT INTO users(username, password, email) VALUES (%s, %s, %s)",
inject_attk_check(username), inject_attk_check(password), inject_attk_check(email))
# Commit changes to database
conn.commit()
flash("Thanks for registering")
# Close cursor and connection
cursor.close()
conn.close()
# Garbage collect after closing database connections. This is to ensure we don't have any leaks.
gc.collect()
session["logged_in"] = True
session['username'] = username
return redirect(url_for('dashboard'))
return render_template("register.html", form=form)
# fix this after debugging
except Exception as e:
return str(e)
# Check to make sure we only run the web server when this file is run directly
if __name__ == "__main__":
app.run(debug=True)
You can’t perform that action at this time.