mirror of
https://github.com/hequan2017/seal.git
synced 2026-04-25 19:20:47 +08:00
123 lines
3.3 KiB
Python
123 lines
3.3 KiB
Python
import logging
|
|
import json
|
|
from django.shortcuts import render, HttpResponse
|
|
from django.contrib.auth import authenticate, login
|
|
from django.shortcuts import redirect, render
|
|
from system.models import Users
|
|
from django.urls import reverse_lazy
|
|
from django.contrib.auth.backends import ModelBackend
|
|
from django.db.models import Q
|
|
from system.form import UserPasswordForm
|
|
from django.contrib.auth.mixins import LoginRequiredMixin, PermissionRequiredMixin
|
|
from django.contrib.auth.decorators import login_required
|
|
from django.views.generic import ListView, View, DetailView, CreateView, UpdateView
|
|
from django.contrib.auth import logout
|
|
|
|
logger = logging.getLogger('system')
|
|
|
|
|
|
class GetInfo(View):
|
|
"""
|
|
为了 配合 seal-vue 项目 临时使用
|
|
"""
|
|
def get(self, request):
|
|
admin = {
|
|
'name': 'super_admin',
|
|
'user_id': '1',
|
|
'access': ['super_admin', 'admin'],
|
|
'token': 'super_admin',
|
|
'avator': 'https://file.iviewui.com/dist/a0e88e83800f138b94d2414621bd9704.png'
|
|
}
|
|
return HttpResponse(json.dumps(admin))
|
|
|
|
|
|
class CustomBackend(ModelBackend):
|
|
"""
|
|
用户名字/邮箱名字 登录
|
|
:param request:
|
|
:return:
|
|
"""
|
|
|
|
def authenticate(self, request, username=None, password=None, **kwargs):
|
|
try:
|
|
user = Users.objects.get(Q(username=username) | Q(email=username))
|
|
if user.check_password(password):
|
|
return user
|
|
except Exception as e:
|
|
logger.error(e)
|
|
return None
|
|
|
|
|
|
def login_view(request):
|
|
"""
|
|
登录
|
|
:param request: username,password
|
|
:return:
|
|
"""
|
|
|
|
error_msg = "用户名或密码错误,或者被禁用,请重试"
|
|
|
|
if request.method == "GET":
|
|
return render(request, 'system/login.html', {'error_msg': error_msg, })
|
|
|
|
if request.method == "POST":
|
|
u = request.POST.get("username")
|
|
p = request.POST.get("password")
|
|
user = authenticate(request, username=u, password=p)
|
|
if user is not None:
|
|
if user.is_active:
|
|
login(request, user)
|
|
request.session['is_login'] = True
|
|
login_ip = request.META['REMOTE_ADDR']
|
|
return redirect('/index')
|
|
else:
|
|
return render(request, 'system/login.html', {'error_msg': error_msg, })
|
|
else:
|
|
return render(request, 'system/login.html', {'error_msg': error_msg, })
|
|
|
|
|
|
@login_required(login_url="/system/login")
|
|
def index(request):
|
|
"""
|
|
首页
|
|
:param request:
|
|
:return:
|
|
"""
|
|
return render(request, 'system/index.html')
|
|
|
|
|
|
class UserPasswordUpdateView(LoginRequiredMixin, UpdateView):
|
|
"""
|
|
修改密码
|
|
:param request:
|
|
:return:
|
|
"""
|
|
template_name = 'system/password.html'
|
|
model = Users
|
|
form_class = UserPasswordForm
|
|
success_url = reverse_lazy('system:logout')
|
|
|
|
def get_object(self, queryset=None):
|
|
return self.request.user
|
|
|
|
def get_context_data(self, **kwargs):
|
|
return super().get_context_data(**kwargs)
|
|
|
|
def get_success_url(self):
|
|
return super().get_success_url()
|
|
|
|
|
|
def logout_view(request):
|
|
"""
|
|
注销
|
|
:param request:
|
|
:return:
|
|
"""
|
|
logout(request)
|
|
return redirect('system:login')
|
|
|
|
|
|
class DisableCSRFCheck(object):
|
|
def process_request(self, request):
|
|
setattr(request, '_dont_enforce_csrf_checks', True)
|