request_early_challenge.py

Enable keyboard shortcuts

import json

import importlib

session_manager = importlib.import_module('device-bound-session-credentials.session_manager')

def main(request, response):

    test_session_manager = session_manager.find_for_request(request)

    use_single_header = json.loads(request.body.decode('utf-8')).get("useSingleHeader")

    if use_single_header is None:

        return (400, response.headers, "")

    challenges = []

    for session_id in session_manager.find_for_request(request).get_session_ids():

        early_challenge = test_session_manager.get_early_challenge(session_id)

        if early_challenge is not None:

            challenges.append(("Sec-Session-Challenge", f'"{early_challenge}";id="{session_id}"'))

    if use_single_header:

        combined_challenges = [("Sec-Session-Challenge", ", ".join([challenge[1] for challenge in challenges]))]

        return (200, combined_challenges, "")

    else:

        return (200, challenges, "")