github
/
spug
mirror of https://github.com/openspug/spug
1
0
Fork 0
Code Issues Releases Wiki Activity

U improve websocket security

pull/220/head
vapao 2020-10-10 18:15:44 +08:00
parent a5f0ef299f
commit eb5e3849f8
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
spug_api/libs/middleware.py
View File

@ -35,6 +35,7 @@ class AuthenticationMiddleware(MiddlewareMixin):
user = User.objects.filter(access_token=access_token).first() user = User.objects.filter(access_token=access_token).first()
if user and x_real_ip == user.last_ip and user.token_expired >= time.time() and user.is_active: if user and x_real_ip == user.last_ip and user.token_expired >= time.time() and user.is_active:
request.user = user request.user = user
if request.path != '/notify/':
user.token_expired = time.time() + 8 * 60 * 60 user.token_expired = time.time() + 8 * 60 * 60
user.save() user.save()
return None return None