start_service_flask.py

# Starts a SkirRPC service on http://localhost:8787/?myapi
#
# Run with:
#   python start_service_flask.py
#
# Run call_service.py to call this service from another process.

import urllib.parse

import skir
from flask import Flask, Response, request
from werkzeug.datastructures import Headers

from skirout import service_skir, user_skir

app = Flask(__name__)


class ServiceImpl:
    def __init__(self):
        self._id_to_user = {}

    def get_user(
        self, request: service_skir.GetUserRequest
    ) -> service_skir.GetUserResponse:
        user_id = request.user_id
        user = self._id_to_user.get(user_id)
        return service_skir.GetUserResponse(user=user)

    def add_user(
        self,
        request: service_skir.AddUserRequest,
        req_headers: Headers,
    ) -> service_skir.AddUserResponse:
        user = request.user
        if user.user_id == 0:
            raise ValueError("invalid user id")
        print(f"Adding user: {user}")
        self._id_to_user[user.user_id] = user
        return service_skir.AddUserResponse()

    _id_to_user: dict[int, user_skir.User]


service_impl = ServiceImpl()

skir_service = skir.Service[Headers]()
skir_service.add_method(service_skir.AddUser, service_impl.add_user)
skir_service.add_method(service_skir.GetUser, service_impl.get_user)

# Configure the service:
# skir_service.options.can_send_unknown_error_message = True
# skir_service.options.keep_unrecognized_values = True
# ...


@app.route("/")
def hello_world():
    return "Hello, World!"


@app.route("/myapi", methods=["GET", "POST"])
def myapi():
    if request.method == "POST":
        req_body = request.get_data(as_text=True)
    else:
        req_body = urllib.parse.unquote(request.query_string.decode("utf-8"))
    req_headers = request.headers
    raw_response = skir_service.handle_request(req_body, req_headers)
    return Response(
        raw_response.data,
        status=raw_response.status_code,
        content_type=raw_response.content_type,
    )


if __name__ == "__main__":
    app.run(debug=True, host="localhost", port=int("8787"))