45 lines
990 B
Python
45 lines
990 B
Python
|
# snakeapi_service/snakeapi_server.py
|
||
|
|
import os
|
||
|
|
import importlib.util
|
||
|
|
from flask import Flask, request
|
||
|
|
from flask_cors import CORS
|
||
|
|
|
||
|
|
# load notebook code as module
|
||
|
|
spec = importlib.util.spec_from_file_location(
|
||
|
|
"nb_module", os.path.join("notebooks", "notebook.py")
|
||
|
|
)
|
||
|
|
nb = importlib.util.module_from_spec(spec)
|
||
|
|
spec.loader.exec_module(nb)
|
||
|
|
|
||
|
|
handlers = {
|
||
|
|
"info": nb.info,
|
||
|
|
"start": nb.start,
|
||
|
|
"move": nb.move,
|
||
|
|
"end": nb.end
|
||
|
|
}
|
||
|
|
|
||
|
|
app = Flask(__name__)
|
||
|
|
CORS(app)
|
||
|
|
|
||
|
|
@app.route("/", methods=["GET"])
|
||
|
|
def on_info():
|
||
|
|
return handlers["info"]()
|
||
|
|
|
||
|
|
@app.route("/start", methods=["POST"])
|
||
|
|
def on_start():
|
||
|
|
handlers["start"](request.get_json())
|
||
|
|
return "ok"
|
||
|
|
|
||
|
|
@app.route("/move", methods=["POST"])
|
||
|
|
def on_move():
|
||
|
|
return handlers["move"](request.get_json())
|
||
|
|
|
||
|
|
@app.route("/end", methods=["POST"])
|
||
|
|
def on_end():
|
||
|
|
handlers["end"](request.get_json())
|
||
|
|
return "ok"
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
port = int(os.environ.get("PORT", "3006"))
|
||
|
|
app.run(host="0.0.0.0", port=port)
|