codeflash-admin/codeflash-internal
1
0
Fork 0
mirror of https://github.com/codeflash-ai/codeflash-internal.git synced 2026-05-04 18:25:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge branch 'main' into ashraf/cf-1045-add-on-commit-trigger-optimize-git-diffs

Browse source
This commit is contained in:
Kevin Turcios 2026-01-29 22:42:57 +00:00 committed by GitHub
commit 84503fbaac
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 16 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
django/aiservice/authapp/auth.py
View file

@ -6,7 +6,7 @@ from ninja.errors import HttpError
from ninja.security import HttpBearer
from authapp.auth_utils import hash_api_key, instance_for_api_key
from authapp.models import CFAPIKeys, Subscriptions
from authapp.models import CFAPIKeys, Organizations, Subscriptions
class AuthenticatedRequest(Protocol):
@ -21,15 +21,16 @@ class AuthenticatedRequest(Protocol):
should_log_features: bool # whether to log optimization features
async def check_subscription_status(user_id, tier) -> bool:
async def check_subscription_status(user_id, tier, organization_id=None) -> bool:
"""Check if a user has a premium subscription that doesn't require feature logging.
Args:
user_id: The ID of the user to check
tier: The user's tier if already available
organization_id: The ID of the user's organization if available
Returns:
bool: False if features should not be logged (premium user), True otherwise
bool: False if features should not be logged (premium user or paid org), True otherwise
"""
# If tier is already set, no need to check subscription
@ -37,6 +38,15 @@ async def check_subscription_status(user_id, tier) -> bool:
return False
try:
# Check if user belongs to a paid organization
if organization_id:
org = await Organizations.objects.filter(id=organization_id).afirst()
if org and org.name == "codeflash-ai":
return True
if org and org.subscription:
# Paid organization - don't log features
return False
subscription = await Subscriptions.objects.filter(user_id=user_id).afirst()
if subscription and subscription.plan_type.lower() in ["pro", "enterprise"]:
# Premium users for CF- don't log features
@ -65,7 +75,9 @@ class AuthBearer(HttpBearer):
request.tier = api_key_instance.tier
request.api_key_id = api_key_instance.id
request.organization_id = api_key_instance.organization_id
request.should_log_features = await check_subscription_status(user_id=request.user, tier=request.tier)
request.should_log_features = await check_subscription_status(
user_id=request.user, tier=request.tier, organization_id=request.organization_id
)
return token
print("THIS SHOULD NOT HAPPEN! More than one users found in the db with the same api key!")