How to Mock OAuth 2.0 Device Code Flow in Python

import hashlib
import time
import uuid


class DeviceCodeFlowMock:
    def __init__(self):
        self.device_codes = {}

    def request_device_code(self, client_id, scope="read write"):
        device_code = uuid.uuid4().hex
        user_code = str(uuid.uuid4().int)[:8].upper()
        expires_in = 300
        interval = 5
        self.device_codes[device_code] = {
            "status": "pending",
            "created_at": time.time(),
            "expires_in": expires_in,
            "user_code": user_code,
            "client_id": client_id,
            "scope": scope,
        }
        return {
            "device_code": device_code,
            "user_code": user_code,
            "verification_uri": "https://example.com/device",
            "expires_in": expires_in,
            "interval": interval,
        }

    def verify_user_code(self, user_code):
        for device_code, details in self.device_codes.items():
            if details["user_code"] == user_code and details["status"] == "pending":
                if time.time() - details["created_at"] > details["expires_in"]:
                    details["status"] = "expired"
                    return False
                details["status"] = "approved"
                return True
        return False

    def poll_for_token(self, device_code):
        details = self.device_codes.get(device_code)
        if not details:
            return {"error": "invalid_device_code"}
        if time.time() - details["created_at"] > details["expires_in"]:
            details["status"] = "expired"
            return {"error": "expired_token"}
        if details["status"] == "approved":
            token = hashlib.sha256(device_code.encode()).hexdigest()
            return {"access_token": token, "token_type": "Bearer", "expires_in": 3600}
        if details["status"] == "pending":
            return {"error": "authorization_pending"}
        return {"error": "access_denied"}


if __name__ == "__main__":
    flow = DeviceCodeFlowMock()
    request = flow.request_device_code("client-123")
    print("Device code request:", request)
    print("User approval:", flow.verify_user_code(request["user_code"]))
    print("Token response:", flow.poll_for_token(request["device_code"]))