updates

Backend/venv/lib/python3.12/site-packages/safety/auth/models.py

@@ -0,0 +1,105 @@
+from dataclasses import dataclass
+import os
+from typing import Any, Optional, Dict
+
+from authlib.integrations.base_client import BaseOAuth
+
+from safety_schemas.models import Stage
+
+
+@dataclass
+class Organization:
+    id: str
+    name: str
+
+    def to_dict(self) -> Dict:
+        """
+        Convert the Organization instance to a dictionary.
+
+        Returns:
+            dict: The dictionary representation of the organization.
+        """
+        return {"id": self.id, "name": self.name}
+
+
+@dataclass
+class Auth:
+    org: Optional[Organization]
+    keys: Any
+    client: Any
+    code_verifier: str
+    client_id: str
+    stage: Optional[Stage] = Stage.development
+    email: Optional[str] = None
+    name: Optional[str] = None
+    email_verified: bool = False
+
+    def is_valid(self) -> bool:
+        """
+        Check if the authentication information is valid.
+
+        Returns:
+            bool: True if valid, False otherwise.
+        """
+        if os.getenv("SAFETY_DB_DIR"):
+            return True
+
+        if not self.client:
+            return False
+
+        if self.client.api_key:
+            return True
+
+        return bool(self.client.token and self.email_verified)
+
+    def refresh_from(self, info: Dict) -> None:
+        """
+        Refresh the authentication information from the provided info.
+
+        Args:
+            info (dict): The information to refresh from.
+        """
+        from safety.auth.utils import is_email_verified
+
+        self.name = info.get("name")
+        self.email = info.get("email")
+        self.email_verified = is_email_verified(info)  # type: ignore
+
+    def get_auth_method(self) -> str:
+        """
+        Get the authentication method.
+
+        Returns:
+            str: The authentication method.
+        """
+        if self.client.api_key:
+            return "API Key"
+
+        if self.client.token:
+            return "Token"
+
+        return "None"
+
+
+class XAPIKeyAuth(BaseOAuth):
+    def __init__(self, api_key: str) -> None:
+        """
+        Initialize the XAPIKeyAuth instance.
+
+        Args:
+            api_key (str): The API key to use for authentication.
+        """
+        self.api_key = api_key
+
+    def __call__(self, r: Any) -> Any:
+        """
+        Add the API key to the request headers.
+
+        Args:
+            r (Any): The request object.
+
+        Returns:
+            Any: The modified request object.
+        """
+        r.headers["X-API-Key"] = self.api_key
+        return r