refactor: drop pydantic, support blocking and raise errors

Author: fantix

gel/auth/__init__.py

@@ -15,3 +15,17 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
+
+from . import email_password
+from .common import BaseServerError, TokenData
+from .pkce import PKCE, generate_pkce, AsyncPKCE, generate_async_pkce
+
+__all__ = [
+    "email_password",
+    "BaseServerError",
+    "TokenData",
+    "PKCE",
+    "generate_pkce",
+    "AsyncPKCE",
+    "generate_async_pkce",
+]

gel/auth/token_data.py gel/auth/common.py

@@ -17,15 +17,42 @@
 #
 
 from __future__ import annotations
-from typing import Optional
+from typing import Any, Optional
 
+import dataclasses
 import uuid
 
-from pydantic import BaseModel
+import httpx
 
 
-class TokenData(BaseModel):
+@dataclasses.dataclass
+class TokenData:
     auth_token: str
     identity_id: uuid.UUID
     provider_token: Optional[str]
     provider_refresh_token: Optional[str]
+
+
+@dataclasses.dataclass
+class BaseServerError(Exception):
+    status_code: int
+    message: str
+
+    @classmethod
+    def raise_or_json(cls, response: httpx.Response, **kwargs) -> Any:
+        try:
+            response.raise_for_status()
+        except httpx.HTTPStatusError as e:
+            raise cls(
+                status_code=e.response.status_code,
+                message=e.response.text,
+                **kwargs,
+            )
+        json = response.json()
+        if "error" in json:
+            raise cls(
+                status_code=response.status_code,
+                message=json["error"],
+                **kwargs,
+            )
+        return json