app.py

# noinspection PyPackageRequirements
import os
from os.path import join, dirname

from dotenv import load_dotenv
from flask import Flask, render_template, request

from yoti_python_sdk import Client
from yoti_python_sdk.dynamic_sharing_service.policy import (
    DynamicPolicyBuilder,
    SourceConstraintBuilder,
)
from yoti_python_sdk.dynamic_sharing_service import DynamicScenarioBuilder
from yoti_python_sdk.dynamic_sharing_service import create_share_url

dotenv_path = join(dirname(__file__), ".env")
load_dotenv(dotenv_path)

from settings import YOTI_SCENARIO_ID, YOTI_CLIENT_SDK_ID, YOTI_KEY_FILE_PATH  # noqa

app = Flask(__name__)


def save_image(selfie_data):
    upload_path = os.path.join(app.root_path, "static", "YotiSelfie.jpg")
    fd = open(upload_path, "wb")
    fd.write(selfie_data)
    fd.close()


@app.route("/")
def index():
    return render_template(
        "index.html", scenario_id=YOTI_SCENARIO_ID, client_sdk_id=YOTI_CLIENT_SDK_ID
    )


@app.route("/dynamic-share")
def dynamic_share():
    client = Client(YOTI_CLIENT_SDK_ID, YOTI_KEY_FILE_PATH)
    policy = (
        DynamicPolicyBuilder().with_full_name().with_age_over(18).with_email().build()
    )
    scenario = (
        DynamicScenarioBuilder()
        .with_policy(policy)
        .with_callback_endpoint("/yoti/auth")
        .build()
    )
    share = create_share_url(client, scenario)
    return render_template(
        "dynamic-share.html",
        yoti_client_sdk_id=YOTI_CLIENT_SDK_ID,
        yoti_share_url=share.share_url,
    )


@app.route("/source-constraints")
def source_constraints():
    client = Client(YOTI_CLIENT_SDK_ID, YOTI_KEY_FILE_PATH)
    constraint = (
        SourceConstraintBuilder().with_driving_licence().with_passport().build()
    )
    policy = (
        DynamicPolicyBuilder()
        .with_full_name(constraints=constraint)
        .with_structured_postal_address(constraints=constraint)
        .build()
    )
    scenario = (
        DynamicScenarioBuilder()
        .with_policy(policy)
        .with_callback_endpoint("/yoti/auth")
        .build()
    )
    share = create_share_url(client, scenario)
    return render_template(
        "dynamic-share.html",
        yoti_client_sdk_id=YOTI_CLIENT_SDK_ID,
        yoti_share_url=share.share_url,
    )


@app.route("/yoti/auth")
def auth():
    client = Client(YOTI_CLIENT_SDK_ID, YOTI_KEY_FILE_PATH)
    activity_details = client.get_activity_details(request.args["token"])
    profile = activity_details.profile
    profile_dict = vars(profile)

    context = profile_dict.get("attributes")
    context["base64_selfie_uri"] = getattr(activity_details, "base64_selfie_uri")
    context["remember_me_id"] = getattr(activity_details, "remember_me_id")
    context["parent_remember_me_id"] = getattr(
        activity_details, "parent_remember_me_id"
    )
    context["receipt_id"] = getattr(activity_details, "receipt_id")
    context["timestamp"] = getattr(activity_details, "timestamp")

    # change this number according to the age condition defined in Yoti Hub
    age_verified = profile.find_age_over_verification(18)

    # Age verification objects don't have the same properties as an attribute,
    # so for this example we had to mock an object with the same properties
    if age_verified is not None:
        context["age_verified"] = {
            "name": "age_verified",
            "value": age_verified,
            "sources": age_verified.attribute.sources,
            "verifiers": age_verified.attribute.verifiers,
        }

    selfie = context.get("selfie")
    if selfie is not None:
        save_image(selfie.value)
    return render_template("profile.html", **context)


if __name__ == "__main__":
    app.run(host="0.0.0.0", ssl_context="adhoc")