import requests

# 1. 定義登錄接口和憑證
login_url = "https://api.datahub.com/v1/auth/login"
headers = {
    "Content-Type": "application/json",
    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
}
payload = {
    "username": "your_username",
    "password": "your_password"
}

# 2. 發(fā)送登錄請求
try:
    response = requests.post(login_url, json=payload, headers=headers)
    
    # 檢查響應狀態(tài)
    if response.status_code == 200:
        # 3. 解析響應數據
        data = response.json()
        
        # 假設服務器返回的數據結構如下：
        # {"code": 0, "msg": "success", "data": {"access_token": "eyJhbGciOiJIUz...", "refresh_token": "..."}}
        
        access_token = data.get("data", {}).get("access_token")
        
        if access_token:
            print(f"成功獲取 Token: {access_token[:20]}...")
        else:
            print("響應中未找到 access_token 字段")
    else:
        print(f"登錄失敗，狀態(tài)碼: {response.status_code}, 原因: {response.text}")
        
except Exception as e:
    print(f"發(fā)生錯誤: {e}")

import requests

# 假設我們已經獲取到了 token
access_token = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9..."

# 1. 構造請求頭
headers = {
    "Authorization": f"Bearer {access_token}",
    "Content-Type": "application/json",
    "Accept": "application/json"
}

# 2. 訪問需要認證的接口（例如：提交數據或獲取用戶信息）
target_url = "https://api.datahub.com/v1/user/profile"
# 或者是一個 POST 提交接口
# target_url = "https://api.datahub.com/v1/data/submit"

try:
    # 發(fā)起 GET 請求獲取數據
    resp = requests.get(target_url, headers=headers)
    
    if resp.status_code == 200:
        print("數據獲取成功:", resp.json())
    elif resp.status_code == 401:
        print("錯誤：Token 失效或未提供，請重新登錄！")
    elif resp.status_code == 403:
        print("錯誤：Token 有效，但權限不足（Scope 不夠）！")
    else:
        print(f"未知錯誤: {resp.status_code}")

except Exception as e:
    print(f"請求異常: {e}")

class JWTCrawler:
    def __init__(self, username, password):
        self.username = username
        self.password = password
        self.base_url = "https://api.datahub.com"
        self.token = None
        self.session = requests.Session() # 使用 Session 保持連接

    def login(self):
        """登錄并獲取 Token"""
        url = f"{self.base_url}/v1/auth/login"
        payload = {"username": self.username, "password": self.password}
        try:
            resp = self.session.post(url, json=payload)
            self.token = resp.json().get("data", {}).get("access_token")
            return True if self.token else False
        except Exception as e:
            print(f"登錄失敗: {e}")
            return False

    def request(self, method, endpoint, **kwargs):
        """通用請求方法，自動注入 Token"""
        if not self.token:
            if not self.login():
                raise Exception("無法獲取有效 Token")
        
        url = f"{self.base_url}{endpoint}"
        headers = kwargs.get("headers", {})
        headers["Authorization"] = f"Bearer {self.token}"
        kwargs["headers"] = headers
        
        response = self.session.request(method, url, **kwargs)
        
        # 處理 Token 過期的情況（通常狀態(tài)碼為 401）
        if response.status_code == 401:
            print("Token 已過期，嘗試重新登錄...")
            if self.login():
                # 重新發(fā)起請求
                headers["Authorization"] = f"Bearer {self.token}"
                kwargs["headers"] = headers
                response = self.session.request(method, url, **kwargs)
            else:
                raise Exception("刷新 Token 失敗")
                
        return response

# 使用示例
# crawler = JWTCrawler("user", "pass")
# resp = crawler.request("GET", "/v1/data/list")
# print(resp.json())