app.py

from imports import *
from prof import *
from calendarfunc import *
import json_log_formatter
import logging

app = Flask(__name__)
app.config['SEND_FILE_MAX_AGE_DEFAULT'] = 0
port = int(os.environ.get("PORT", 5000))

script_dir = os.path.dirname(__file__)
minorFileName = 'course_list.json'
minorFileName = os.path.join(script_dir, minorFileName)

formatter = json_log_formatter.JSONFormatter()

json_handler = logging.FileHandler(filename='my-log.log')
json_handler.setFormatter(formatter)

logger = logging.getLogger('my_json')
logger.addHandler(json_handler)
logger.setLevel(logging.INFO)

@app.route('/', methods=['GET', 'POST'])
def home():
	if request.method == 'POST':
		data = request.json
		logger.info('Downloaded ICS for Students')
		return download_ics_file(data)
	logger.info('GET Home')
	return render_template('home.html')

@app.route('/ajax/', methods=['POST'])
def getCourse():
	query = request.form.get('query')
	results = searchData(query)
	if results:
		logger.info('query success')
		return json.dumps( results[0] )
	else:
		return json.dumps( {} )

@app.route('/minor/')
def minor():
	with app.open_resource(minorFileName, 'r') as minorFile:
		data = json.load(minorFile)
		logger.info('loaded minor data')
	return json.dumps( data )

@app.route('/professor', methods=['POST'])
def result():
	prof = request.form['prof']
	tb, times, dept, website, prof = fetch_results(prof)
	logger.info('prof')
	return render_template('main.html', name=prof, website=website, data=tb, times=times, profs=profs, dept=dept, error=False)

my_events = []

@app.route('/professor', methods=['GET'])
def main():
	global my_events
	prof = request.args.get('prof')
	if prof:
		tb, times, dept, website, prof = fetch_results(prof)
		my_events = format_data(tb)
		logger.info('Get Prof Info')
		return render_template('main.html', name=prof, website=website, data=tb, times=times, profs=profs, dept=dept, error=False)
	else:
		tb_predefined, times_predefined = get_predefined()
		return render_template('main.html', profs=profs, times=times_predefined, data=tb_predefined)

@app.route('/ics_helper')
def ics_helper():
	return download_ics_file(my_events)

@app.route("/download_helper")
def func():
	return download_ics_file(my_events)

@app.route('/landing', methods=['GET'])
def landing():
	return render_template('landing.html')


if __name__ == '__main__':
	port = int(os.environ.get("PORT", 5000))
	app.run(host='0.0.0.0', port=port, debug=True)