Generate token after autorization

[Kanellaman]

Generate token after autorization

[Kanellaman]

I resolved it by expanding the validator

from oauth2_provider.oauth2_validators import OAuth2Validator
from .models import AccessToken, RefreshToken
from store.models import Subscription
from django.http import JsonResponse

class CustomOAuth2Validator(OAuth2Validator):
    def save_bearer_token(self, token, request, *args, **kwargs):
        super().save_bearer_token(token, request, *args, **kwargs)
        # Assuming the header for subscription is 'X-Subscription'
        subscription_id = request.headers.get('HTTP_SUBSCRIPTION')
        if subscription_id:
            try:
                subscription = Subscription.objects.get(uuid=subscription_id)
                # Update the AccessToken with the subscription
                access_token = AccessToken.objects.get(token=token['access_token'])
                access_token.subscription = subscription
                access_token.save()
            except Subscription.DoesNotExist:
                return JsonResponse({"error": "Wrong subscription ID"}, status=400)