Handle errors better, use the plaintext output

pull/1/head
Leigh Morresi 4 years ago
parent 2f018ac04c
commit 646a54945a

@ -1,5 +1,8 @@
#!/usr/bin/python3 #!/usr/bin/python3
# @todo logging
import json import json
import eventlet import eventlet
import eventlet.wsgi import eventlet.wsgi
@ -22,7 +25,7 @@ ticker_thread = None
datastore = store.ChangeDetectionStore() datastore = store.ChangeDetectionStore()
messages = [] messages = []
running_update_threads={} running_update_threads = {}
app = Flask(__name__, static_url_path='/static') app = Flask(__name__, static_url_path='/static')
app.config['STATIC_RESOURCES'] = "/app/static" app.config['STATIC_RESOURCES'] = "/app/static"
@ -32,11 +35,11 @@ app.config['STATIC_RESOURCES'] = "/app/static"
# Disables caching of the templates # Disables caching of the templates
app.config['TEMPLATES_AUTO_RELOAD'] = True 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 # 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. # running or something similar.
@app.template_filter('format_last_checked_time') @app.template_filter('format_last_checked_time')
def _jinja2_filter_datetime(watch_obj, format="%Y-%m-%d %H:%M:%S"): def _jinja2_filter_datetime(watch_obj, format="%Y-%m-%d %H:%M:%S"):
global running_update_threads global running_update_threads
if watch_obj['uuid'] in running_update_threads: if watch_obj['uuid'] in running_update_threads:
if running_update_threads[watch_obj['uuid']].is_alive(): if running_update_threads[watch_obj['uuid']].is_alive():
@ -47,17 +50,20 @@ def _jinja2_filter_datetime(watch_obj, format="%Y-%m-%d %H:%M:%S"):
return datetime.datetime.utcfromtimestamp(int(watch_obj['last_checked'])).strftime(format) return datetime.datetime.utcfromtimestamp(int(watch_obj['last_checked'])).strftime(format)
@app.template_filter('format_timestamp') @app.template_filter('format_timestamp')
def _jinja2_filter_datetimestamp(timestamp, format="%Y-%m-%d %H:%M:%S"): def _jinja2_filter_datetimestamp(timestamp, format="%Y-%m-%d %H:%M:%S"):
if timestamp == 0: if timestamp == 0:
return 'Never' return 'Never'
return datetime.datetime.utcfromtimestamp(timestamp).strftime(format) return datetime.datetime.utcfromtimestamp(timestamp).strftime(format)
@app.route("/", methods=['GET']) @app.route("/", methods=['GET'])
def main_page(): def main_page():
global messages global messages
# Show messages but once. # Show messages but once.
# maybe if the change happened more than a few days ago.. add a class
output = render_template("watch-overview.html", watches=datastore.data['watching'], messages=messages) output = render_template("watch-overview.html", watches=datastore.data['watching'], messages=messages)
messages = [] messages = []
return output return output
@ -75,9 +81,9 @@ def static_content(group, filename):
def api_watch_add(): def api_watch_add():
global messages global messages
#@todo add_watch should throw a custom Exception for validation etc # @todo add_watch should throw a custom Exception for validation etc
datastore.add_watch(url=request.form.get('url'), tag=request.form.get('tag')) datastore.add_watch(url=request.form.get('url').strip(), tag=request.form.get('tag').strip())
messages.append({'class':'ok', 'message': 'Saved'}) messages.append({'class': 'ok', 'message': 'Saved'})
launch_checks() launch_checks()
return redirect(url_for('main_page')) return redirect(url_for('main_page'))
@ -86,14 +92,14 @@ def api_watch_add():
def api_watch_checknow(): def api_watch_checknow():
global messages global messages
uuid=request.args.get('uuid') uuid = request.args.get('uuid')
# dict would be better, this is a simple safety catch. # dict would be better, this is a simple safety catch.
for watch in datastore.data['watching']: for watch in datastore.data['watching']:
if watch['uuid'] == uuid: if watch['uuid'] == uuid:
# @todo cancel if already running? # @todo cancel if already running?
running_update_threads[uuid] = fetch_site_status.perform_site_check(uuid=uuid, running_update_threads[uuid] = fetch_site_status.perform_site_check(uuid=uuid,
datastore=datastore) datastore=datastore)
running_update_threads[uuid].start() running_update_threads[uuid].start()
return redirect(url_for('main_page')) return redirect(url_for('main_page'))
@ -105,14 +111,15 @@ def launch_checks():
global running_update_threads global running_update_threads
for watch in datastore.data['watching']: for watch in datastore.data['watching']:
if watch['last_checked'] <= time.time() - 20: if watch['last_checked'] <= time.time() - 86400:
running_update_threads[watch['uuid']] = fetch_site_status.perform_site_check(uuid = watch['uuid'], datastore=datastore) running_update_threads[watch['uuid']] = fetch_site_status.perform_site_check(uuid=watch['uuid'],
datastore=datastore)
running_update_threads[watch['uuid']].start() running_update_threads[watch['uuid']].start()
def ticker_thread_check_time_launch_checks():
# Thread runner to check every minute
def ticker_thread_check_time_launch_checks():
while True: while True:
print ("lanching")
launch_checks() launch_checks()
time.sleep(60) time.sleep(60)
@ -121,10 +128,9 @@ def main(argv):
ssl_mode = False ssl_mode = False
port = 5000 port = 5000
#@todo handle ctrl break # @todo handle ctrl break
ticker_thread = threading.Thread(target=ticker_thread_check_time_launch_checks).start() ticker_thread = threading.Thread(target=ticker_thread_check_time_launch_checks).start()
try: try:
opts, args = getopt.getopt(argv, "sp:") opts, args = getopt.getopt(argv, "sp:")
except getopt.GetoptError: except getopt.GetoptError:

@ -1,21 +1,21 @@
from threading import Thread from threading import Thread
import time import time
import requests import requests
import hashlib import hashlib
# Hmm Polymorphism datastore, thread, etc # Hmm Polymorphism datastore, thread, etc
class perform_site_check(Thread): class perform_site_check(Thread):
def __init__(self, *args, uuid=False, datastore, **kwargs): def __init__(self, *args, uuid=False, datastore, **kwargs):
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
self.timestamp = int(time.time()) # used for storage etc too self.timestamp = int(time.time()) # used for storage etc too
self.uuid = uuid self.uuid = uuid
self.datastore = datastore self.datastore = datastore
self.url = datastore.get_val(uuid, 'url') self.url = datastore.get_val(uuid, 'url')
self.current_md5 = datastore.get_val(uuid, 'previous_md5') self.current_md5 = datastore.get_val(uuid, 'previous_md5')
def save_firefox_screenshot(self, uuid, output): def save_firefox_screenshot(self, uuid, output):
#@todo call selenium or whatever # @todo call selenium or whatever
return return
def save_response_output(self, output): def save_response_output(self, output):
@ -31,19 +31,45 @@ class perform_site_check(Thread):
f.write(output) f.write(output)
f.close() f.close()
def run(self): def run(self):
headers = {
'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.66 Safari/537.36',
'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.9',
'Accept-Encoding': 'gzip, deflate, br',
'Accept-Language': 'en-GB,en-US;q=0.9,en;q=0.8,cs;q=0.7'
}
print("Checking", self.url)
import html2text
try: try:
r = requests.get(self.url) r = requests.get(self.url, headers=headers, timeout=15)
except requests.exceptions.ConnectionError as e:
stripped_text_from_html = html2text.html2text(r.content.decode('utf-8'))
# Usually from networkIO/requests level
except (requests.exceptions.ConnectionError,requests.exceptions.ReadTimeout) as e:
self.datastore.update_watch(self.uuid, 'last_error', str(e)) self.datastore.update_watch(self.uuid, 'last_error', str(e))
print(str(e))
# Usually from html2text level
except UnicodeDecodeError as e:
self.datastore.update_watch(self.uuid, 'last_error', str(e))
print(str(e))
print (str(e))
else: else:
# We rely on the actual text in the html output.. many sites have random script vars etc
self.datastore.update_watch(self.uuid, 'last_error', False) self.datastore.update_watch(self.uuid, 'last_error', False)
self.datastore.update_watch(self.uuid, 'last_check_status', r.status_code) self.datastore.update_watch(self.uuid, 'last_check_status', r.status_code)
fetched_md5=hashlib.md5(r.text.encode('utf-8')).hexdigest() fetched_md5 = hashlib.md5(stripped_text_from_html.encode('utf-8')).hexdigest()
if self.current_md5 != fetched_md5: if self.current_md5 != fetched_md5:
self.datastore.update_watch(self.uuid, 'previous_md5', fetched_md5) self.datastore.update_watch(self.uuid, 'previous_md5', fetched_md5)

@ -14,5 +14,7 @@ validators
bleach==3.2.1 bleach==3.2.1
html5lib==0.9999999 # via bleach html5lib==0.9999999 # via bleach
html2text
# @notes # @notes
# - Dont install socketio, it interferes with flask_socketio # - Dont install socketio, it interferes with flask_socketio

@ -24,3 +24,7 @@ flex-direction: column;
align-items: center; align-items: center;
justify-content: center; justify-content: center;
} }
.watch-table .error {
color: #aa0000;
}

@ -2,6 +2,8 @@ import json
import uuid import uuid
import validators import validators
# @TODO Have a var which is the base value, this is referred to even in the templating.. merge and append,not just append
# Is there an existing library to ensure some data store (JSON etc) is in sync with CRUD methods? # Is there an existing library to ensure some data store (JSON etc) is in sync with CRUD methods?
# Open a github issue if you know something :) # Open a github issue if you know something :)
# https://stackoverflow.com/questions/6190468/how-to-trigger-function-on-value-change # https://stackoverflow.com/questions/6190468/how-to-trigger-function-on-value-change

@ -16,7 +16,7 @@
<!-- make a nice list of tags here to click on --> <!-- make a nice list of tags here to click on -->
<table class="pure-table pure-table-striped"> <table class="pure-table pure-table-striped watch-table">
<thead> <thead>
<tr> <tr>
<th>#</th> <th>#</th>
@ -30,10 +30,12 @@
{% for watch in watches %} {% for watch in watches %}
<tr class="{{ loop.cycle('pure-table-odd', 'pure-table-even') }}"> <tr id="{{ watch.uuid }}" class="{{ loop.cycle('pure-table-odd', 'pure-table-even') }} {% if watch.last_error is defined and watch.last_error != False %}error{% endif %}">
<td>{{ loop.index }}</td> <td>{{ loop.index }}</td>
<td>{{ watch.url }}</td> <td>{{ watch.url }}</td>
<td>{{watch|format_last_checked_time}}</td> <td>{{watch|format_last_checked_time}}
{% if watch.last_error is defined and watch.last_error != False %} !! {% endif %}
</td>
<td>{{watch.last_changed|format_timestamp}}</td> <td>{{watch.last_changed|format_timestamp}}</td>
<td><a href="/api/checknow?uuid={{ watch.uuid}}" class="pure-button pure-button-primary">Recheck</a> <button type="submit" class="pure-button pure-button-primary">Delete</button></td> <td><a href="/api/checknow?uuid={{ watch.uuid}}" class="pure-button pure-button-primary">Recheck</a> <button type="submit" class="pure-button pure-button-primary">Delete</button></td>
</tr> </tr>

Loading…
Cancel
Save