API Key Authentication

import os
import requests
from typing import Optional

class OpenWearablesClient:
    """Client for Open Wearables API with key management."""
    
    def __init__(self, jwt_token: Optional[str] = None, api_key: Optional[str] = None):
        self.base_url = "https://api.openwearables.com/api/v1"
        self.jwt_token = jwt_token
        self.api_key = api_key or os.getenv("OPEN_WEARABLES_API_KEY")
    
    def _get_headers(self, use_jwt: bool = False) -> dict:
        """Get authentication headers."""
        if use_jwt and self.jwt_token:
            return {"Authorization": f"Bearer {self.jwt_token}"}
        elif self.api_key:
            return {"X-Open-Wearables-API-Key": self.api_key}
        else:
            raise ValueError("No authentication credentials provided")
    
    # API Key Management (requires JWT)
    
    def create_api_key(self, name: str) -> dict:
        """Create a new API key."""
        response = requests.post(
            f"{self.base_url}/api-keys",
            headers={**self._get_headers(use_jwt=True), "Content-Type": "application/json"},
            json={"name": name}
        )
        response.raise_for_status()
        return response.json()
    
    def list_api_keys(self) -> list[dict]:
        """List all API keys."""
        response = requests.get(
            f"{self.base_url}/api-keys",
            headers=self._get_headers(use_jwt=True)
        )
        response.raise_for_status()
        return response.json()
    
    def rotate_api_key(self, key_id: str) -> dict:
        """Rotate an API key."""
        response = requests.post(
            f"{self.base_url}/api-keys/{key_id}/rotate",
            headers=self._get_headers(use_jwt=True)
        )
        response.raise_for_status()
        return response.json()
    
    def delete_api_key(self, key_id: str) -> dict:
        """Delete an API key."""
        response = requests.delete(
            f"{self.base_url}/api-keys/{key_id}",
            headers=self._get_headers(use_jwt=True)
        )
        response.raise_for_status()
        return response.json()
    
    # Data Access (uses API key)
    
    def get_users(self) -> list[dict]:
        """Get all users using API key authentication."""
        response = requests.get(
            f"{self.base_url}/users",
            headers=self._get_headers()
        )
        response.raise_for_status()
        return response.json()

# Usage example
if __name__ == "__main__":
    # Initialize with JWT token for key management
    jwt_token = "your-jwt-token-here"
    client = OpenWearablesClient(jwt_token=jwt_token)
    
    # Create a new API key
    new_key = client.create_api_key("Production Server")
    print(f"Created API key: {new_key['id']}")
    
    # Save the key securely (e.g., to environment or secret manager)
    api_key = new_key['id']
    
    # Use the API key for data access
    data_client = OpenWearablesClient(api_key=api_key)
    users = data_client.get_users()
    print(f"Found {len(users)} users")
    
    # Rotate key periodically
    rotated_key = client.rotate_api_key(api_key)
    print(f"Rotated to new key: {rotated_key['id']}")