57 lines
1.2 KiB
Python
57 lines
1.2 KiB
Python
from flask import Flask, render_template, url_for, jsonify, send_file, request
|
|
from datetime import datetime
|
|
from pathlib import Path
|
|
|
|
from src.iris_api import Timetable
|
|
|
|
def compile_javascript(base_path = "static"):
|
|
js_file_ptrs = list(Path(base_path).glob("*.js"))
|
|
js_files = []
|
|
|
|
print(js_file_ptrs)
|
|
|
|
for file in js_file_ptrs:
|
|
print(file.name)
|
|
fname = f"{file.name}"
|
|
js_files.append(url_for('static', filename=fname))
|
|
|
|
return js_files
|
|
|
|
eva = 8002377
|
|
# eva = 8004158
|
|
|
|
|
|
app = Flask(__name__)
|
|
|
|
@app.route("/favicon.ico")
|
|
def favicon():
|
|
return send_file("static/images/favicon.ico")
|
|
|
|
@app.route("/")
|
|
def main_page():
|
|
# print(compile_javascript())
|
|
tt = Timetable(eva)
|
|
tt.get_changes()
|
|
|
|
return render_template(
|
|
'base.html',
|
|
station=tt.station,
|
|
current_time_str=datetime.now().strftime(r"%H:%M:%S"),
|
|
js_files=compile_javascript()
|
|
)
|
|
|
|
|
|
@app.route("/update", methods=['POST'])
|
|
def update():
|
|
tt = Timetable(eva)
|
|
|
|
tt.get_stops()
|
|
|
|
table_data = tt.str_stops()
|
|
# print(table_data[0])
|
|
|
|
|
|
result = {'status': 'success'}
|
|
# result.update(data)
|
|
result["table_data"] = table_data
|
|
return jsonify(result) |