klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-05 21:29:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use the state.billing_enabled field to determine whether to use the subscribed scope

Browse Source
This commit is contained in:
sabaimran
2023-11-24 20:41:19 -08:00
parent 69c8f45830
commit 9c868ee10b
1 changed files with 15 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
src/khoj/configure.py
View File

@@ -75,11 +75,14 @@ class UserAuthenticationBackend(AuthenticationBackend):
.afirst() .afirst()
) )
if user: if user:
subscription_state = await aget_user_subscription_state(user) if state.billing_enabled:
subscribed = subscription_state == SubscriptionState.SUBSCRIBED.value subscription_state = await aget_user_subscription_state(user)
if subscribed: subscribed = subscription_state == SubscriptionState.SUBSCRIBED.value
return AuthCredentials(["authenticated", "subscribed"]), AuthenticatedKhojUser(user) if subscribed:
return AuthCredentials(["authenticated", "subscribed"]), AuthenticatedKhojUser(user) return AuthCredentials(["authenticated", "subscribed"]), AuthenticatedKhojUser(
user_with_token.user
)
return AuthCredentials(["authenticated"]), AuthenticatedKhojUser(user)
if len(request.headers.get("Authorization", "").split("Bearer ")) == 2: if len(request.headers.get("Authorization", "").split("Bearer ")) == 2:
# Get bearer token from header # Get bearer token from header
bearer_token = request.headers["Authorization"].split("Bearer ")[1] bearer_token = request.headers["Authorization"].split("Bearer ")[1]
@@ -91,10 +94,13 @@ class UserAuthenticationBackend(AuthenticationBackend):
.afirst() .afirst()
) )
if user_with_token: if user_with_token:
subscription_state = await aget_user_subscription_state(user_with_token.user) if state.billing_enabled:
subscribed = subscription_state == SubscriptionState.SUBSCRIBED.value subscription_state = await aget_user_subscription_state(user_with_token.user)
if subscribed: subscribed = subscription_state == SubscriptionState.SUBSCRIBED.value
return AuthCredentials(["authenticated", "subscribed"]), AuthenticatedKhojUser(user_with_token.user) if subscribed:
return AuthCredentials(["authenticated", "subscribed"]), AuthenticatedKhojUser(
user_with_token.user
)
return AuthCredentials(["authenticated"]), AuthenticatedKhojUser(user_with_token.user) return AuthCredentials(["authenticated"]), AuthenticatedKhojUser(user_with_token.user)
if state.anonymous_mode: if state.anonymous_mode:
user = await self.khojuser_manager.filter(username="default").prefetch_related("subscription").afirst() user = await self.khojuser_manager.filter(username="default").prefetch_related("subscription").afirst()