run_workflow.py

import uuid
from typing import Optional
import os

from flask import Flask, render_template, request
from temporalio.client import Client, TLSConfig

from activities import BookVacationInput

# Import the workflow from the previous code
from book_workflow import BookWorkflow

app = Flask(__name__)

# Set to more than 1 to show activity retries due to service down
ATTEMPTS = 1


@app.route("/")
async def display_form():
    return render_template("book_vacation.html")


@app.route("/book", methods=["POST"])
async def book_vacation():
    user_id = f'{request.form.get("name").replace(" ", "-").lower()}-{str(uuid.uuid4().int)[:6]}'
    attempts = request.form.get("attempts")
    car = request.form.get("car")
    hotel = request.form.get("hotel")
    flight = request.form.get("flight")

    input = BookVacationInput(
        attempts=int(attempts),
        book_user_id=user_id,
        book_car_id=car,
        book_hotel_id=hotel,
        book_flight_id=flight,
    )

    if (
        os.getenv("TEMPORAL_MTLS_TLS_CERT")
        and os.getenv("TEMPORAL_MTLS_TLS_KEY") is not None
    ):
        server_root_ca_cert: Optional[bytes] = None
        f = open(os.getenv("TEMPORAL_MTLS_TLS_CERT"), "rb")
        client_cert = f.read()

        f = open(os.getenv("TEMPORAL_MTLS_TLS_KEY"), "rb")
        client_key = f.read()

        # Start client
        client = await Client.connect(
            os.getenv("TEMPORAL_HOST_URL"),
            namespace=os.getenv("TEMPORAL_NAMESPACE"),
            tls=TLSConfig(
                server_root_ca_cert=server_root_ca_cert,
                client_cert=client_cert,
                client_private_key=client_key,
            ),
        )
    else:
        client = await Client.connect("localhost:7233")

    result = await client.execute_workflow(
        BookWorkflow.run,
        input,
        id=user_id,
        task_queue="saga-task-queue",
    )
    if result == "Voyage cancelled":
        return render_template("book_vacation.html", cancelled=True)

    else:
        print(result)
        result_list = result.split("Booked ")
        car = result_list[1].split(" Booking ")[0].title()
        hotel = result_list[2].split(" Booking ")[0].title()
        flight = result_list[2].split(" Booking ")[1].title()
        print(user_id)
        return render_template(
            "book_vacation.html",
            result=result,
            cancelled=False,
            car=car,
            hotel=hotel,
            flight=flight,
            user_id=user_id,
        )


if __name__ == "__main__":
    app.run(host="0.0.0.0", debug=True)