22 lines
741 B
Python
22 lines
741 B
Python
"""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"""
|
|
assert isinstance(request, (HttpRequest, Request))
|
|
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]
|