mirror of
https://github.com/aljazceru/python-teos.git
synced 2025-12-17 14:14:22 +01:00
142 lines
4.7 KiB
Python
142 lines
4.7 KiB
Python
from pisa import *
|
|
from pisa.watcher import Watcher
|
|
from pisa.inspector import Inspector
|
|
from pisa.appointment import Appointment
|
|
from flask import Flask, request, Response, abort, jsonify
|
|
import json
|
|
|
|
|
|
# FIXME: HERE FOR TESTING (get_block_count). REMOVE WHEN REMOVING THE FUNCTION
|
|
from pisa.utils.authproxy import AuthServiceProxy
|
|
from pisa.conf import BTC_RPC_USER, BTC_RPC_PASSWD, BTC_RPC_HOST, BTC_RPC_PORT
|
|
|
|
# ToDo: #5-add-async-to-api
|
|
app = Flask(__name__)
|
|
HTTP_OK = 200
|
|
HTTP_BAD_REQUEST = 400
|
|
HTTP_SERVICE_UNAVAILABLE = 503
|
|
|
|
|
|
@app.route('/', methods=['POST'])
|
|
def add_appointment():
|
|
remote_addr = request.environ.get('REMOTE_ADDR')
|
|
remote_port = request.environ.get('REMOTE_PORT')
|
|
|
|
if debug:
|
|
logging.info('[API] connection accepted from {}:{}'.format(remote_addr, remote_port))
|
|
|
|
# Check content type once if properly defined
|
|
request_data = json.loads(request.get_json())
|
|
appointment = inspector.inspect(request_data)
|
|
|
|
if type(appointment) == Appointment:
|
|
appointment_added = watcher.add_appointment(appointment, debug, logging)
|
|
rcode = HTTP_OK
|
|
|
|
# ToDo: #13-create-server-side-signature-receipt
|
|
if appointment_added:
|
|
response = "appointment accepted. locator: {}".format(appointment.locator)
|
|
else:
|
|
response = "appointment rejected"
|
|
|
|
elif type(appointment) == tuple:
|
|
rcode = HTTP_BAD_REQUEST
|
|
response = "appointment rejected. Error {}: {}".format(appointment[0], appointment[1])
|
|
|
|
else:
|
|
# We should never end up here, since inspect only returns appointments or tuples. Just in case.
|
|
rcode = HTTP_BAD_REQUEST
|
|
response = "appointment rejected. Request does not match the standard"
|
|
|
|
if debug:
|
|
logging.info('[API] sending response and disconnecting: {} --> {}:{}'.format(response, remote_addr,
|
|
remote_port))
|
|
|
|
return Response(response, status=rcode, mimetype='text/plain')
|
|
|
|
|
|
# FIXME: THE NEXT THREE API ENDPOINTS ARE FOR TESTING AND SHOULD BE REMOVED / PROPERLY MANAGED BEFORE PRODUCTION!
|
|
# ToDo: #17-add-api-keys
|
|
@app.route('/get_appointment', methods=['GET'])
|
|
def get_appointment():
|
|
locator = request.args.get('locator')
|
|
response = []
|
|
|
|
# ToDo: #15-add-system-monitor
|
|
|
|
appointment_in_watcher = watcher.appointments.get(locator)
|
|
|
|
if appointment_in_watcher:
|
|
for appointment in appointment_in_watcher:
|
|
appointment_data = appointment.to_json()
|
|
appointment_data['status'] = "being_watched"
|
|
response.append(appointment_data)
|
|
|
|
if watcher.responder:
|
|
responder_jobs = watcher.responder.jobs
|
|
|
|
for job_id, job in responder_jobs.items():
|
|
if job.locator == locator:
|
|
job_data = job.to_json()
|
|
job_data['status'] = "dispute_responded"
|
|
job_data['confirmations'] = watcher.responder.confirmation_counter.get(job_id)
|
|
response.append(job_data)
|
|
|
|
if not response:
|
|
response.append({"locator": locator, "status": "not found"})
|
|
|
|
response = jsonify(response)
|
|
|
|
return response
|
|
|
|
|
|
@app.route('/get_all_appointments', methods=['GET'])
|
|
def get_all_appointments():
|
|
watcher_appointments = []
|
|
responder_jobs = []
|
|
|
|
# ToDo: #15-add-system-monitor
|
|
|
|
if request.remote_addr in request.host or request.remote_addr == '127.0.0.1':
|
|
for app_id, appointment in watcher.appointments.items():
|
|
jobs_data = [job.to_json() for job in appointment]
|
|
|
|
watcher_appointments.append({app_id: jobs_data})
|
|
|
|
if watcher.responder:
|
|
for job_id, job in watcher.responder.jobs.items():
|
|
job_data = job.to_json()
|
|
job_data['confirmations'] = watcher.responder.confirmation_counter.get(job_id)
|
|
responder_jobs.append({job_id: job_data})
|
|
|
|
response = jsonify({"watcher_appointments": watcher_appointments, "responder_jobs": responder_jobs})
|
|
|
|
else:
|
|
abort(404)
|
|
|
|
return response
|
|
|
|
|
|
@app.route('/get_block_count', methods=['GET'])
|
|
def get_block_count():
|
|
bitcoin_cli = AuthServiceProxy("http://%s:%s@%s:%d" % (BTC_RPC_USER, BTC_RPC_PASSWD, BTC_RPC_HOST,
|
|
BTC_RPC_PORT))
|
|
|
|
return jsonify({"block_count": bitcoin_cli.getblockcount()})
|
|
|
|
|
|
def start_api(d, l):
|
|
# FIXME: Pretty ugly but I haven't found a proper way to pass it to add_appointment
|
|
global debug, logging, watcher, inspector
|
|
debug = d
|
|
logging = l
|
|
|
|
# ToDo: #18-separate-api-from-watcher
|
|
watcher = Watcher()
|
|
inspector = Inspector(debug, logging)
|
|
|
|
# Setting Flask log t ERROR only so it does not mess with out logging
|
|
logging.getLogger('werkzeug').setLevel(logging.ERROR)
|
|
|
|
app.run(host=HOST, port=PORT)
|