与azure斗智斗勇。

import os
from urllib.parse import urlparse, parse_qs
from dotenv import load_dotenv
import requests

load_dotenv()

# 设置基本 URL
base_url = f"https://login.microsoftonline.com/{os.getenv('TENANT_ID')}/oauth2/v2.0/authorize"

# 构建请求参数
params = {
    "client_id": os.getenv("CLIENT_ID"),
    "scope": os.getenv("SCOPE"),
    "redirect_uri": os.getenv("REDIRECT_URI"),
    "response_type": "code",
    "code_challenge": os.getenv("CODE_CHALLENGE"),
    "code_challenge_method": os.getenv("CODE_CHALLENGE_METHOD")
}

# 创建请求
request = requests.Request('GET', base_url, params=params)

# 准备请求并打印完整的 URL
prepared_request = request.prepare()

print("访问这个网址登录")
print(prepared_request.url)
# 示例 URL
url = input("登陆过后的redirect url")

# 解析 URL
parsed_url = urlparse(url)

# 获取查询参数
query_params = parse_qs(parsed_url.query)

# 输出解析结果
for key, value in query_params.items():
    print(f"{key}: {value[0]}")  # value 是一个列表，取第一个元素
    if key == "code":
        with open("getAToken", "w") as f:
            f.write(f"code={value[0]}")

import os
import requests
from dotenv import load_dotenv

# 加载 .env 文件
load_dotenv()
load_dotenv("getAToken")

# 从环境变量中获取参数
tenant_id = os.getenv("TENANT_ID")  # 请在 .env 文件中定义此参数
client_id = os.getenv("CLIENT_ID")  # 你的 Client ID
redirect_uri = os.getenv("REDIRECT_URI")  # 重定向 URI
scope = os.getenv("SCOPE")  # 请求的权限范围
code = os.getenv("code")  # 授权码
code_verifier = os.getenv("CODE_VERIFIER")  # 代码验证器
client_secret = os.getenv("CLIENT_SECRET")  # 客户端密钥（仅对 Web 应用需要）

# 构建请求的 URL
token_url = f"https://login.microsoftonline.com/{tenant_id}/oauth2/v2.0/token"

# 设置请求参数
data = {
    "client_id": client_id,
    "scope": scope,
    "code": code,
    "redirect_uri": redirect_uri,
    "grant_type": "authorization_code",
    "code_verifier": code_verifier,
    "client_secret": client_secret  # 仅对 Web 应用需要
}

print(data)

# 发送 POST 请求
response = requests.post(token_url, data=data, headers={
    "Content-Type": "application/x-www-form-urlencoded",
})

# 检查响应
if response.status_code == 200:
    # 成功，解析访问令牌
    token_response = response.json()
    print(token_response)
    with open("refresh_token", 'w') as env_file:
        env_file.write(f"refresh_token={token_response.get("refresh_token")}")
    access_token = token_response.get("access_token")
    print("Access Token:", access_token)
else:
    # 处理错误
    print("Error:", response.status_code, response.text)