42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
# Databricks notebook source
|
||
import base64
|
||
import time
|
||
import urllib.parse
|
||
import requests
|
||
from Crypto.Cipher import AES
|
||
from Crypto.Util.Padding import pad
|
||
|
||
# === 配置区 ===
|
||
# Spring Boot 服务端的密钥（Base64编码）
|
||
SECRET_KEY_BASE64 = "qkLkLHPnqNfJ5d3X6DgLpw=="
|
||
# 接口地址
|
||
# https://mdcuat.astrazeneca.cn/earth/app/mail/task/ma/check
|
||
# https://mdc.astrazeneca.cn/earth/app/mail/task/ma/check
|
||
API_URL = "https://mdc.astrazeneca.cn/earth/app/mail/task/ma/check"
|
||
|
||
# === AES 加密函数 ===
|
||
def generate_aes_token(secret_key_base64: str) -> str:
|
||
# Hutool 的 SecureUtil.aes() 默认是 AES/ECB/PKCS5Padding
|
||
secret_key = base64.b64decode(secret_key_base64)
|
||
cipher = AES.new(secret_key, AES.MODE_ECB)
|
||
|
||
# 加密当前时间戳（毫秒）
|
||
timestamp = str(int(time.time() * 1000))
|
||
encrypted = cipher.encrypt(pad(timestamp.encode("utf-8"), AES.block_size))
|
||
|
||
# 输出 Base64 编码的 token
|
||
return base64.b64encode(encrypted).decode("utf-8")
|
||
|
||
# === 调用接口 ===
|
||
def call_check_api():
|
||
token = generate_aes_token(SECRET_KEY_BASE64)
|
||
encoded_token = urllib.parse.quote(token)
|
||
url = f"{API_URL}?token={encoded_token}"
|
||
|
||
print(f"请求URL: {url}")
|
||
response = requests.get(url)
|
||
print("响应状态码:", response.status_code)
|
||
print("响应内容:", response.text)
|
||
|
||
if __name__ == "__main__":
|
||
call_check_api() |