Bearer handling

grai-io · Nov 3, 2023 · 57c3cca · 57c3cca
1 parent 95ec232
commit 57c3cca
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 13 deletions.
diff --git a/grai-server/app/auth/validation.py b/grai-server/app/auth/validation.py
@@ -14,8 +14,6 @@ def _generate_token(user: User) -> str:
 def send_validation_email(user: User):
     token = _generate_token(user)
 
-    print(token)
-
     c = {
         "base_url": config("FRONTEND_URL", "http://localhost:3000"),
         "token": token,

diff --git a/grai-server/app/scripts/test.py b/grai-server/app/scripts/test.py
diff --git a/grai-server/app/workspaces/permissions.py b/grai-server/app/workspaces/permissions.py
@@ -1,7 +1,17 @@
-from rest_framework_api_key.permissions import BaseHasAPIKey
-
+from rest_framework_api_key.permissions import BaseHasAPIKey, KeyParser, APIKey
+from django.http import HttpRequest
+import re
 from .models import WorkspaceAPIKey
 
 
+class BearerKeyParser(KeyParser):
+    keyword = "Bearer"
+
+
+class BearerApiKey(BaseHasAPIKey):
+    model = APIKey
+    key_parser = BearerKeyParser()
+
+
 class HasWorkspaceAPIKey(BaseHasAPIKey):
     model = WorkspaceAPIKey