Marmelator/pyinfra
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 36 Wiki Activity

endpoint suffixes passed to stream processor

Browse Source
This commit is contained in:
Matthias Bisping 2022-05-08 17:28:55 +02:00
parent 8c1ad64464
commit 1d09337378
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
pyinfra/server/server.py
View File

@ -1,5 +1,6 @@
import logging
import flask
from flask import Flask, jsonify, request
from funcy import compose, flatten
@ -33,12 +34,14 @@ class RestOndDemandProcessor(OnDemandProcessor):
class RestStreamProcessor(RestOndDemandProcessor):
"""Wraps an on-demand-processor. Combine with a webserver that provides the endpoints 'submit' and 'pickup'."""
def __init__(self, fn):
def __init__(self, fn, submit_suffix="submit", pickup_suffix="pickup"):
super(RestStreamProcessor, self).__init__(fn=fn)
self.submit_suffix = submit_suffix
self.pickup_suffix = pickup_suffix
def submit(self, request, **kwargs):
super(RestStreamProcessor, self).submit(request)
return jsonify(f"{request.base_url.replace('/submit', '')}/pickup")
return jsonify(request.base_url.replace(self.submit_suffix, self.pickup_suffix))
def pickup(self):
result = self.compute_next()
@ -64,7 +67,7 @@ def valid(result):
def set_up_processing_server(process_fn):
app = Flask(__name__)
stream = RestStreamProcessor(process_fn)
stream = RestStreamProcessor(process_fn, submit_suffix="submit", pickup_suffix="pickup")
@app.route("/ready", methods=["GET"])
def ready():