config.py

Author: Sid Mohan

datafog/config.py

@@ -1,4 +1,66 @@
+import os
 from enum import Enum
+from typing import Optional
+
+from pydantic import HttpUrl
+from pydantic_settings import BaseSettings
+
+
+class DataFogConfig(BaseSettings):
+    """Configuration settings for DataFog SDK"""
+
+    # API Keys and Authentication
+    api_key: str = os.environ.get("DATAFOG_API_KEY", "")
+
+    # Base URLs for different services
+    annotator_base_url: HttpUrl = (
+        ""  # for ex, "https://datafog-api-prod.azurewebsites.net"
+    )
+    anonymizer_base_url: HttpUrl = (
+        ""  # for ex, "https://datafog-api-anonymizer-prod.azurewebsites.net"
+    )
+
+    # Default language
+    default_language: str = "en"
+
+    # Timeouts
+    request_timeout: int = 30  # seconds
+
+    # Retry settings
+    max_retries: int = 3
+    retry_backoff_factor: float = 0.3
+
+    # Rate limiting
+    rate_limit_per_minute: Optional[int] = None
+
+    # Logging
+    log_level: str = "INFO"
+
+    class Config:
+        env_prefix = "DATAFOG_"
+        case_sensitive = False
+
+    def update(self, **kwargs):
+        """Update configuration with new values"""
+        for key, value in kwargs.items():
+            if hasattr(self, key):
+                setattr(self, key, value)
+            else:
+                raise ValueError(f"Invalid configuration key: {key}")
+
+
+# Create a global instance of the configuration
+datafog_config = DataFogConfig()
+
+
+def get_config() -> DataFogConfig:
+    """Get the current configuration"""
+    return datafog_config
+
+
+def configure(**kwargs):
+    """Update the global configuration"""
+    datafog_config.update(**kwargs)
 
 
 class OperationType(str, Enum):