Skip to content

Commit

Permalink
Merge pull request #1733 from st1020/fix/delete-session-if-user-disable
Browse files Browse the repository at this point in the history
fix: delete session if user disable
  • Loading branch information
Bidaya0 authored Aug 16, 2023
2 parents d3ea337 + fb30573 commit 01e93f2
Showing 1 changed file with 9 additions and 2 deletions.
11 changes: 9 additions & 2 deletions dongtai_common/endpoint/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
from operator import ior
from typing import TYPE_CHECKING

from django.contrib.auth import logout
from django.core.paginator import EmptyPage, Paginator
from django.db.models import Count
from django.http import JsonResponse
Expand All @@ -30,6 +31,7 @@
)
from dongtai_common.utils import const
from dongtai_common.utils.init_schema import VIEW_CLASS_TO_SCHEMA
from dongtai_conf import settings

if TYPE_CHECKING:
from django.core.paginator import _SupportsPagination
Expand Down Expand Up @@ -94,9 +96,14 @@ def dispatch(self, request, *args, **kwargs):
self.request = request
self.headers = self.default_response_headers # deprecate?

if not request.user.is_active:
if not request.user.is_active and not request.user.is_anonymous:
logout(request)
request.session.delete()
return R.failure(msg="用户已经禁用")
response = R.failure(msg="用户已经禁用", status_code=403)
request.session.delete()
response.delete_cookie(key=settings.CSRF_COOKIE_NAME, domain=settings.SESSION_COOKIE_DOMAIN)
response.delete_cookie(key="sessionid", domain=settings.SESSION_COOKIE_DOMAIN)
return response

try:
self.initial(request, *args, **kwargs)
Expand Down

0 comments on commit 01e93f2

Please sign in to comment.