You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
124 lines
4.0 KiB
124 lines
4.0 KiB
import json
|
|
import uuid as uuid_builder
|
|
import validators
|
|
|
|
|
|
# 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 :)
|
|
# https://stackoverflow.com/questions/6190468/how-to-trigger-function-on-value-change
|
|
class ChangeDetectionStore:
|
|
|
|
def __init__(self):
|
|
self.data = {
|
|
'watching': {},
|
|
'settings': {
|
|
'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;'
|
|
},
|
|
'requests': {
|
|
'timeout': 15, # Default 15 seconds
|
|
'max_seconds_from_last_check': 3 * 60 * 60 # Default 3 hours
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
# Base definition for all watchers
|
|
self.generic_definition = {
|
|
'url': None,
|
|
'tag': None,
|
|
'last_checked': 0,
|
|
'last_changed': 0,
|
|
'title': None,
|
|
'uuid': str(uuid_builder.uuid4()),
|
|
'headers' : {}, # Extra headers to send
|
|
'history' : {} # Dict of timestamp and output stripped filename
|
|
}
|
|
|
|
|
|
try:
|
|
with open('/datastore/url-watches.json') as json_file:
|
|
from_disk = json.load(json_file)
|
|
|
|
self.data.update(from_disk)
|
|
|
|
# Reinitialise each `watching` with our generic_definition in the case that we add a new var in the future.
|
|
# @todo pretty sure theres a python we todo this with an abstracted(?) object!
|
|
i = 0
|
|
for uuid, watch in self.data['watching'].items():
|
|
_blank = self.generic_definition.copy()
|
|
_blank.update(watch)
|
|
self.data['watching'].update({uuid: _blank})
|
|
print("Watching:", uuid, _blank['url'])
|
|
|
|
# First time ran, doesnt exist.
|
|
except (FileNotFoundError, json.decoder.JSONDecodeError):
|
|
print("Creating JSON store")
|
|
|
|
self.add_watch(url='https://changedetection.io', tag='general')
|
|
self.add_watch(url='http://www.quotationspage.com/random.php', tag='test')
|
|
|
|
def update_watch(self, uuid, val, var):
|
|
|
|
self.data['watching'][uuid].update({val: var})
|
|
self.sync_to_json()
|
|
|
|
def get_all_tags(self):
|
|
tags=[]
|
|
for uuid, watch in self.data['watching'].items():
|
|
|
|
# Support for comma separated list of tags.
|
|
for tag in watch['tag'].split(','):
|
|
tag = tag.strip()
|
|
if not tag in tags:
|
|
tags.append(tag)
|
|
|
|
tags.sort()
|
|
return tags
|
|
|
|
def delete(self, uuid):
|
|
# Probably their should be dict...
|
|
del(self.data['watching'][uuid])
|
|
self.sync_to_json()
|
|
|
|
|
|
def url_exists(self, url):
|
|
|
|
# Probably their should be dict...
|
|
for watch in self.data['watching']:
|
|
if watch['url'] == url:
|
|
return True
|
|
|
|
return False
|
|
|
|
def get_val(self, uuid, val):
|
|
# Probably their should be dict...
|
|
return self.data['watching'][uuid].get(val)
|
|
|
|
def add_watch(self, url, tag):
|
|
|
|
# @todo deal with exception
|
|
validators.url(url)
|
|
|
|
# @todo use a common generic version of this
|
|
|
|
_blank = self.generic_definition.copy()
|
|
_blank.update({
|
|
'url': url,
|
|
'tag': tag,
|
|
'uuid': str(uuid_builder.uuid4())
|
|
})
|
|
|
|
self.data['watching'].update({_blank['uuid']: _blank})
|
|
|
|
self.sync_to_json()
|
|
|
|
def sync_to_json(self):
|
|
with open('/datastore/url-watches.json', 'w') as json_file:
|
|
json.dump(self.data, json_file, indent=4)
|
|
|
|
# body of the constructor
|