API Interface (#617)
dgtlmoon
3 years ago
committed by
GitHub
parent
2c834cfe37
commit
07e279b38d
@ -0,0 +1,125 @@
|
|||||||
|
from flask_restful import abort, Resource
|
||||||
|
from flask import request, make_response
|
||||||
|
import validators
|
||||||
|
from . import auth
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
# https://www.w3.org/Protocols/rfc2616/rfc2616-sec10.html
|
||||||
|
|
||||||
|
class Watch(Resource):
|
||||||
|
def __init__(self, **kwargs):
|
||||||
|
# datastore is a black box dependency
|
||||||
|
self.datastore = kwargs['datastore']
|
||||||
|
self.update_q = kwargs['update_q']
|
||||||
|
|
||||||
|
# Get information about a single watch, excluding the history list (can be large)
|
||||||
|
# curl http://localhost:4000/api/v1/watch/<string:uuid>
|
||||||
|
# ?recheck=true
|
||||||
|
@auth.check_token
|
||||||
|
def get(self, uuid):
|
||||||
|
from copy import deepcopy
|
||||||
|
watch = deepcopy(self.datastore.data['watching'].get(uuid))
|
||||||
|
if not watch:
|
||||||
|
abort(404, message='No watch exists with the UUID of {}'.format(uuid))
|
||||||
|
|
||||||
|
if request.args.get('recheck'):
|
||||||
|
self.update_q.put(uuid)
|
||||||
|
return "OK", 200
|
||||||
|
|
||||||
|
# Return without history, get that via another API call
|
||||||
|
watch['history_n'] = len(watch['history'])
|
||||||
|
del (watch['history'])
|
||||||
|
return watch
|
||||||
|
|
||||||
|
@auth.check_token
|
||||||
|
def delete(self, uuid):
|
||||||
|
if not self.datastore.data['watching'].get(uuid):
|
||||||
|
abort(400, message='No watch exists with the UUID of {}'.format(uuid))
|
||||||
|
|
||||||
|
self.datastore.delete(uuid)
|
||||||
|
return 'OK', 204
|
||||||
|
|
||||||
|
|
||||||
|
class WatchHistory(Resource):
|
||||||
|
def __init__(self, **kwargs):
|
||||||
|
# datastore is a black box dependency
|
||||||
|
self.datastore = kwargs['datastore']
|
||||||
|
|
||||||
|
# Get a list of available history for a watch by UUID
|
||||||
|
# curl http://localhost:4000/api/v1/watch/<string:uuid>/history
|
||||||
|
def get(self, uuid):
|
||||||
|
watch = self.datastore.data['watching'].get(uuid)
|
||||||
|
if not watch:
|
||||||
|
abort(404, message='No watch exists with the UUID of {}'.format(uuid))
|
||||||
|
return watch['history'], 200
|
||||||
|
|
||||||
|
|
||||||
|
class WatchSingleHistory(Resource):
|
||||||
|
def __init__(self, **kwargs):
|
||||||
|
# datastore is a black box dependency
|
||||||
|
self.datastore = kwargs['datastore']
|
||||||
|
|
||||||
|
# Read a given history snapshot and return its content
|
||||||
|
# <string:timestamp> or "latest"
|
||||||
|
# curl http://localhost:4000/api/v1/watch/<string:uuid>/history/<int:timestamp>
|
||||||
|
@auth.check_token
|
||||||
|
def get(self, uuid, timestamp):
|
||||||
|
watch = self.datastore.data['watching'].get(uuid)
|
||||||
|
if not watch:
|
||||||
|
abort(404, message='No watch exists with the UUID of {}'.format(uuid))
|
||||||
|
|
||||||
|
if not len(watch['history']):
|
||||||
|
abort(404, message='Watch found but no history exists for the UUID {}'.format(uuid))
|
||||||
|
|
||||||
|
if timestamp == 'latest':
|
||||||
|
timestamp = list(watch['history'].keys())[-1]
|
||||||
|
|
||||||
|
with open(watch['history'][timestamp], 'r') as f:
|
||||||
|
content = f.read()
|
||||||
|
|
||||||
|
response = make_response(content, 200)
|
||||||
|
response.mimetype = "text/plain"
|
||||||
|
return response
|
||||||
|
|
||||||
|
|
||||||
|
class CreateWatch(Resource):
|
||||||
|
def __init__(self, **kwargs):
|
||||||
|
# datastore is a black box dependency
|
||||||
|
self.datastore = kwargs['datastore']
|
||||||
|
self.update_q = kwargs['update_q']
|
||||||
|
|
||||||
|
@auth.check_token
|
||||||
|
def post(self):
|
||||||
|
# curl http://localhost:4000/api/v1/watch -H "Content-Type: application/json" -d '{"url": "https://my-nice.com", "tag": "one, two" }'
|
||||||
|
json_data = request.get_json()
|
||||||
|
tag = json_data['tag'].strip() if json_data.get('tag') else ''
|
||||||
|
|
||||||
|
if not validators.url(json_data['url'].strip()):
|
||||||
|
return "Invalid or unsupported URL", 400
|
||||||
|
|
||||||
|
extras = {'title': json_data['title'].strip()} if json_data.get('title') else {}
|
||||||
|
|
||||||
|
new_uuid = self.datastore.add_watch(url=json_data['url'].strip(), tag=tag, extras=extras)
|
||||||
|
self.update_q.put(new_uuid)
|
||||||
|
return {'uuid': new_uuid}, 201
|
||||||
|
|
||||||
|
# Return concise list of available watches and some very basic info
|
||||||
|
# curl http://localhost:4000/api/v1/watch|python -mjson.tool
|
||||||
|
# ?recheck_all=1 to recheck all
|
||||||
|
@auth.check_token
|
||||||
|
def get(self):
|
||||||
|
list = {}
|
||||||
|
for k, v in self.datastore.data['watching'].items():
|
||||||
|
list[k] = {'url': v['url'],
|
||||||
|
'title': v['title'],
|
||||||
|
'last_checked': v['last_checked'],
|
||||||
|
'last_changed': v['last_changed'],
|
||||||
|
'last_error': v['last_error']}
|
||||||
|
|
||||||
|
if request.args.get('recheck_all'):
|
||||||
|
for uuid in self.datastore.data['watching'].keys():
|
||||||
|
self.update_q.put(uuid)
|
||||||
|
return {'status': "OK"}, 200
|
||||||
|
|
||||||
|
return list, 200
|
||||||
@ -0,0 +1,33 @@
|
|||||||
|
from flask import request, make_response, jsonify
|
||||||
|
from functools import wraps
|
||||||
|
|
||||||
|
|
||||||
|
# Simple API auth key comparison
|
||||||
|
# @todo - Maybe short lived token in the future?
|
||||||
|
|
||||||
|
def check_token(f):
|
||||||
|
@wraps(f)
|
||||||
|
def decorated(*args, **kwargs):
|
||||||
|
datastore = args[0].datastore
|
||||||
|
|
||||||
|
config_api_token_enabled = datastore.data['settings']['application'].get('api_access_token_enabled')
|
||||||
|
if not config_api_token_enabled:
|
||||||
|
return
|
||||||
|
|
||||||
|
try:
|
||||||
|
api_key_header = request.headers['x-api-key']
|
||||||
|
except KeyError:
|
||||||
|
return make_response(
|
||||||
|
jsonify("No authorization x-api-key header."), 403
|
||||||
|
)
|
||||||
|
|
||||||
|
config_api_token = datastore.data['settings']['application'].get('api_access_token')
|
||||||
|
|
||||||
|
if api_key_header != config_api_token:
|
||||||
|
return make_response(
|
||||||
|
jsonify("Invalid access - API key invalid."), 403
|
||||||
|
)
|
||||||
|
|
||||||
|
return f(*args, **kwargs)
|
||||||
|
|
||||||
|
return decorated
|
||||||
Loading…
Reference in new issue