phzhik/gault-millau
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

update setup cookie for tokens

Browse Source
This commit is contained in:
Anatoly 2019-09-03 16:25:44 +03:00
parent 1b3eececf2
commit b20fe5e6fb
1 changed files with 2 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
apps/utils/views.py
View File

@ -38,26 +38,18 @@ class JWTGenericViewMixin(generics.GenericAPIView):
""" """
COOKIES = list() COOKIES = list()
# Set max_age for tokens
if permanent:
access_token_max_age = settings.SIMPLE_JWT.get('ACCESS_TOKEN_LIFETIME_SECONDS')
refresh_token_max_age = settings.SIMPLE_JWT.get('REFRESH_TOKEN_LIFETIME_SECONDS')
else:
access_token_max_age = settings.COOKIES_MAX_AGE
refresh_token_max_age = settings.COOKIES_MAX_AGE
# Write to cookie access and refresh token with secure flag # Write to cookie access and refresh token with secure flag
if access_token and refresh_token: if access_token and refresh_token:
_access_token = self.COOKIE(key='access_token', _access_token = self.COOKIE(key='access_token',
value=access_token, value=access_token,
http_only=self.ACCESS_TOKEN_HTTP_ONLY, http_only=self.ACCESS_TOKEN_HTTP_ONLY,
secure=self.ACCESS_TOKEN_SECURE, secure=self.ACCESS_TOKEN_SECURE,
max_age=access_token_max_age) max_age=settings.COOKIES_MAX_AGE if permanent else None)
_refresh_token = self.COOKIE(key='refresh_token', _refresh_token = self.COOKIE(key='refresh_token',
value=refresh_token, value=refresh_token,
http_only=self.REFRESH_TOKEN_HTTP_ONLY, http_only=self.REFRESH_TOKEN_HTTP_ONLY,
secure=self.REFRESH_TOKEN_SECURE, secure=self.REFRESH_TOKEN_SECURE,
max_age=refresh_token_max_age) max_age=settings.COOKIES_MAX_AGE if permanent else None)
COOKIES.extend((_access_token, _refresh_token)) COOKIES.extend((_access_token, _refresh_token))
return COOKIES return COOKIES