Function FetchRequestToken

tools/python/oauth2-flow/oauth2_flow.py:81–126

(cookies)

def FetchRequestToken(cookies):
    logging.info("Loading OAuth2 auth page ...")

    query = urllib.parse.urlencode({
        "client_id": OAuthClientId,
        "redirect_uri": OAuthRedirectUri,
        "response_type": OAuthResponseType,
        "scope": OAuthScope
    })

    headers = {
        "Cookie": encodeCookies(cookies),
    }

    conn = http.client.HTTPSConnection(osmHost)
    conn.request("GET", f"/oauth2/authorize?{query}", headers=headers)

    res = conn.getresponse()

    if res.status >= 400:
        print(res.status)
        print(res.headers)
        print(res.read().decode('utf-8'))
        raise Exception("Can't load OAuth2 page")

    elif res.status == 302:
        logging.info("User already accepted OAuth2 request!")
        redirectUri = res.headers['Location']
        logging.info(f"Accepted OAuth2: {redirectUri}")
        oauthCode = FindOauthCode(redirectUri)

        if not oauthCode:
            raise Exception(f"Can't find 'code' in redirect URI: {redirectUri}")

        return oauthCode
    else:
        respBody = res.read().decode("utf-8")
        authToken = FindAuthenticityToken("/oauth2/authorize", respBody)
        if not authToken:
            print(res.status)
            print(res.headers)
            print(respBody)
            raise Exception("Invalid authToken '{authToken}'")
        logging.info(f"Parsed authToken = {authToken}")

        return SendAuthRequest(authToken, cookies)

def SendAuthRequest(authToken, cookies):
    logging.info("Accepting OAuth2 ...")