server_ojp2.py

#!/usr/bin/env python3

import logging

from fastapi import FastAPI, Request, Response

import app_logging
from api.ojp2 import FareServiceOjp2
from configuration import (
    HTTP_HOST,
    HTTP_PORT,
    HTTPS,
    SSL_CERTFILE,
    SSL_KEYFILE,
    HTTP_SLUG,
)

app_logging.setup_logging()
logger = logging.getLogger(__name__)
app = FastAPI(title="ojp-fare API for OJP2")

fare_service = FareServiceOjp2()

@app.get("/health/liveness", tags=["Health"])
async def liveness(fastapi_req: Request):
    """
    Implements liveness probe.
    """
    logger.debug("Received liveness probe request: " + str(fastapi_req))
    return Response("Liveness: OK", media_type="text/plain; charset=utf-8")

# implements basic readiness probe
@app.get("/health/readiness", tags=["Health"])
async def readiness(fastapi_req: Request):
    """
    Implements readiness probe.
    """
    logger.debug("Received readiness probe request: " + str(fastapi_req))
    return Response("Readiness: OK", media_type="text/plain; charset=utf-8")

@app.post("/" + HTTP_SLUG, tags=["Open Journey Planner"])
async def post_request(fastapi_req: Request) ->Response:
    """
    Handles OjpFare requests: for ojp2 only.
    """
    body = await fastapi_req.body()
    logger.debug("Received request: " + str(body))
    return fare_service.handle_request(body)

if __name__ == "__main__":
    import uvicorn

    if HTTPS:
        uvicorn.run(
            app,
            host=HTTP_HOST,
            port=HTTP_PORT,
            ssl_keyfile=SSL_KEYFILE,
            ssl_certfile=SSL_CERTFILE,
        )
    else:
        uvicorn.run(app, host=HTTP_HOST, port=HTTP_PORT)