def register(request: Request, **kwargs):
"""
管理员注册
"""
if request.is_post():
params = RegisterAdminReqSerializer(request.data).params_data
if request.session.get('verify', '') != params.verify:
raise RspError('验证码错误')
if params.password1 != params.password2:
raise RspError({'password1': [_('两次密码不一致')]})
if User.objects.filter(username=params.username).exists():
raise RspError(_('%s 已被注册') % params.username)
user_model = User()
user_model.username = params.username
user_model.alias = params.alias
user_model.set_password(params.password1)
user_model.reg_ip = request.real_ip
user_model.clean_fields()
user_model.save()
user_info_model, _created = UserInfo.objects.get_or_create(user=user_model)
user_info_model.email = params.email
user_info_model.qq = params.qq
user_info_model.phone = params.phone
user_info_model.save()
if settings.ALLOW_REGISTER_ROLE_CHOICE and params.default_group:
role = Role.objects.filter(type=Role.RoleType.GROUP, name=params.default_group).first()
if role:
user_model.role.add(role)
User.login_user(request, user_model)
return Response(msg=_('注册成功'))
return render_to_response('myadmin/register.html', locals())
def list(
self,
request,
):
"""查询 列表"""
return render_to_response('analysis/query/list.html',
super().list(request))
def create_svnrepo(self, request):
"""
创建 SVN 仓库
:param request:
:return:
"""
svnrepo_name = request.POST.get('svnrepo_name', '').strip()
if svnrepo_name:
SvnPath.create_svnrepo(svnrepo_name)
return render_to_response('svn_admin/svn_path/create_svnrepo.html', super().edit(request))
def edit(self, request, *args, **kwargs):
"""角色编辑"""
model_instance: Role = self.get_model_instance(EditParams)
params = self.RoleEditParmas(request.query_params).params_data
if not model_instance.id:
if params.type:
model_instance.type = params.type
if params.parent and int(params.parent) != model_instance.id:
model_instance.parent = Role.objects.get(id=params.parent)
serializer = self.get_serializer(instance=model_instance)
return render_to_response("myadmin/role/edit.html",
self.response(serializer.data))
def login(request: Request):
"""登录
"""
msg = ''
now = datetime.datetime.now()
if request.is_post:
try:
params = LoginSerializer(request.data).params_data
request.COOKIES["username"] = params.username
username = params.username
password = params.password
if not password:
raise LoginError('密码错误 !')
check_login_status(request)
if not username or not password:
raise LoginErrors.USERNAME_OR_PASSWORD_EMPTY
is_pass = False
if settings.USE_LDAP_AUTH:
the_user = ldap_login(request, username, password)
is_pass = the_user is not None
if not is_pass:
the_user: User = User.objects.filter(username=username).first()
if not the_user and '@' in username:
user_info = UserInfo.objects.filter(email=username).first()
if user_info:
the_user = user_info.user
if not the_user:
raise LoginErrors.ACCOUNT_NOT_EXIST
is_pass = the_user.check_password(password)
if the_user.status != User.Status.NORMAL:
raise LoginErrors.ACCOUNT_STATUS_ERROR(_('账户已 %s') % the_user.get_status_display())
if is_pass:
User.login_user(request, the_user)
redirect_url = request.query_params.get('from_url', settings.INDEX_URL)
if request.is_ajax():
return Response(LoginRspSer(dict(url=redirect_url)))
else:
return HttpResponseRedirect(redirect_url)
else:
raise LoginErrors.USERNAME_OR_PASSWORD_FAIL
except RspError as error:
request.session['err_count'] = request.session.get('err_count', 0) + 1
raise error
return render_to_response('myadmin/login.html', locals())
def edit(self, request, *args, **kwargs):
"""查询 编辑"""
data = super().edit(request).data.data
data.update({'mark_map': SqlMarkManager.mark_map})
return render_to_response('analysis/query/edit.html', data)
def list(request: Request):
"""动态配置"""
data = settings_manager.group_options_map
return render_to_response('dynamic_settimgs/list.html',
data,
request=request)
def list(self, request, *args, **kwargs):
"""角色列表"""
return render_to_response("myadmin/role/list.html",
super().list(request, *args, **kwargs))
def js(self, request):
"""Celery任务js库"""
return render_to_response('celery_task/celery_task.js',
AssociatedTaskResult,
content_type='application/javascript')
def edit(self, request):
""" 编辑"""
return render_to_response('svn_admin/svn_path/edit.html', super().edit(request))
def list(self, request):
""" 列表"""
SvnPath.init_svn_projects()
return render_to_response('svn_admin/svn_path/list.html', super().list(request))
def index(self, request):
""" Svn 项目管理"""
return render_to_response('svn_admin/svn_path/index.html',locals())
def edit(self, request):
"""用户编辑"""
return render_to_response("myadmin/user/edit.html",
super().edit(request))
def list(self, request):
"""用户列表"""
return render_to_response("myadmin/user/list.html",
super().list(request))
def js(request):
return render_to_response('ws_gateway/ws.js',
locals(),
content_type='application/javascript')
def test(request):
messages.info(request, 'test messages')
#users = list(User.objects.all())
#print(users)
return render_to_response('ws_gateway/test.html', locals())