"""Utils app method."""
import random
from django.http.request import HttpRequest
from rest_framework.request import Request


def generate_code(digits=6, string_output=True):
    """Generate random int."""
    max_value = 10 ** digits - 1
    min_value = 10 ** (digits - 1)
    value = random.randint(min_value, max_value)
    return str(value) if string_output else value


def get_token_from_request(request):
    """Get access token from request"""
    if 'Authorization' in request.headers:
        if isinstance(request, HttpRequest):
            return request.headers.get('Authorization').split(' ')[::-1][0]
        elif isinstance(request, Request):
            return request._request.headers.get('Authorization').split(' ')[::-1][0]