154 lines
4.4 KiB
Python
154 lines
4.4 KiB
Python
#!/usr/bin/python3
|
|
|
|
import json
|
|
import eventlet
|
|
import eventlet.wsgi
|
|
|
|
import time
|
|
import os
|
|
import getopt
|
|
import sys
|
|
import datetime
|
|
|
|
import threading
|
|
|
|
from flask import Flask, render_template, request, send_file, send_from_directory, safe_join, abort, redirect, url_for
|
|
|
|
# Local
|
|
import store
|
|
import fetch_site_status
|
|
|
|
ticker_thread = None
|
|
|
|
datastore = store.ChangeDetectionStore()
|
|
messages = []
|
|
running_update_threads={}
|
|
|
|
app = Flask(__name__, static_url_path='/static')
|
|
app.config['STATIC_RESOURCES'] = "/app/static"
|
|
|
|
# app.config['SECRET_KEY'] = 'secret!'
|
|
|
|
# Disables caching of the templates
|
|
app.config['TEMPLATES_AUTO_RELOAD'] = True
|
|
|
|
# We use the whole watch object from the store/JSON so we can see if there's some related status in terms of a thread
|
|
# running or something similar.
|
|
@app.template_filter('format_last_checked_time')
|
|
def _jinja2_filter_datetime(watch_obj, format="%Y-%m-%d %H:%M:%S"):
|
|
|
|
global running_update_threads
|
|
if watch_obj['uuid'] in running_update_threads:
|
|
if running_update_threads[watch_obj['uuid']].is_alive():
|
|
return "Checking now.."
|
|
|
|
if watch_obj['last_checked'] == 0:
|
|
return 'Never'
|
|
|
|
return datetime.datetime.utcfromtimestamp(int(watch_obj['last_checked'])).strftime(format)
|
|
|
|
@app.template_filter('format_timestamp')
|
|
def _jinja2_filter_datetimestamp(timestamp, format="%Y-%m-%d %H:%M:%S"):
|
|
if timestamp == 0:
|
|
return 'Never'
|
|
return datetime.datetime.utcfromtimestamp(timestamp).strftime(format)
|
|
|
|
@app.route("/", methods=['GET'])
|
|
def main_page():
|
|
global messages
|
|
|
|
# Show messages but once.
|
|
output = render_template("watch-overview.html", watches=datastore.data['watching'], messages=messages)
|
|
messages = []
|
|
return output
|
|
|
|
|
|
@app.route("/static/<string:group>/<string:filename>", methods=['GET'])
|
|
def static_content(group, filename):
|
|
try:
|
|
return send_from_directory("/app/static/{}".format(group), filename=filename)
|
|
except FileNotFoundError:
|
|
abort(404)
|
|
|
|
|
|
@app.route("/api/add", methods=['POST'])
|
|
def api_watch_add():
|
|
global messages
|
|
|
|
#@todo add_watch should throw a custom Exception for validation etc
|
|
datastore.add_watch(url=request.form.get('url'), tag=request.form.get('tag'))
|
|
messages.append({'class':'ok', 'message': 'Saved'})
|
|
launch_checks()
|
|
return redirect(url_for('main_page'))
|
|
|
|
|
|
@app.route("/api/checknow", methods=['GET'])
|
|
def api_watch_checknow():
|
|
global messages
|
|
|
|
uuid=request.args.get('uuid')
|
|
|
|
# dict would be better, this is a simple safety catch.
|
|
for watch in datastore.data['watching']:
|
|
if watch['uuid'] == uuid:
|
|
# @todo cancel if already running?
|
|
running_update_threads[uuid] = fetch_site_status.perform_site_check(uuid=uuid,
|
|
datastore=datastore)
|
|
running_update_threads[uuid].start()
|
|
|
|
return redirect(url_for('main_page'))
|
|
|
|
|
|
# Can be used whenever, launch threads that need launching to update the stored information
|
|
def launch_checks():
|
|
import fetch_site_status
|
|
global running_update_threads
|
|
|
|
for watch in datastore.data['watching']:
|
|
if watch['last_checked'] <= time.time() - 20:
|
|
running_update_threads[watch['uuid']] = fetch_site_status.perform_site_check(uuid = watch['uuid'], datastore=datastore)
|
|
running_update_threads[watch['uuid']].start()
|
|
|
|
def ticker_thread_check_time_launch_checks():
|
|
|
|
while True:
|
|
print ("lanching")
|
|
launch_checks()
|
|
time.sleep(60)
|
|
|
|
|
|
def main(argv):
|
|
ssl_mode = False
|
|
port = 5000
|
|
|
|
#@todo handle ctrl break
|
|
ticker_thread = threading.Thread(target=ticker_thread_check_time_launch_checks).start()
|
|
|
|
|
|
try:
|
|
opts, args = getopt.getopt(argv, "sp:")
|
|
except getopt.GetoptError:
|
|
print('backend.py -s SSL enable -p [port]')
|
|
sys.exit(2)
|
|
|
|
for opt, arg in opts:
|
|
if opt == '-s':
|
|
ssl_mode = True
|
|
|
|
if opt == '-p':
|
|
port = arg
|
|
|
|
# @todo finalise SSL config, but this should get you in the right direction if you need it.
|
|
if ssl_mode:
|
|
eventlet.wsgi.server(eventlet.wrap_ssl(eventlet.listen(('', port)),
|
|
certfile='cert.pem',
|
|
keyfile='privkey.pem',
|
|
server_side=True), app)
|
|
|
|
else:
|
|
eventlet.wsgi.server(eventlet.listen(('', port)), app)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main(sys.argv[1:])
|