def host_del(request, pk):
origin_url = memory_reverse_url(request, 'cmdb:host_list')
if request.method == 'GET':
return render(request, 'rbac/delete.html', {'cancel': origin_url})
models.Host.objects.filter(id=pk).delete()
return redirect(origin_url)
def menu_del(request, pk):
# 用户点击取消删除角色按钮以后就回到role_list 界面
origin_url = memory_reverse_url(request, 'rbac:menu_list')
if request.method == "GET":
return render(request, 'rbac/delete.html', {"cancel": origin_url})
models.Menu.objects.filter(id=pk).delete()
return redirect(origin_url)
def host_add(request):
if request.method == 'GET':
form = HostModelForm()
return render(request, 'rbac/change.html', {'form': form})
form = HostModelForm(data=request.POST)
if form.is_valid():
form.save()
return redirect(memory_reverse_url(request, 'cmdb:host_list'))
return render(request, 'rbac/change.html', {'form': form})
def permission_del(request, pk):
"""
删除
:param request:
:param pk:
:return:
"""
origin_url = memory_reverse_url(request, 'rbac:menu_list')
if request.method == "GET":
return render(request, 'rbac/delete.html', {"cancel": origin_url})
models.Permission.objects.filter(id=pk).delete()
return redirect(origin_url)
def user_del(request, pk):
"""
删除用户
:param request:
:param pk:
:return:
"""
origin_url = memory_reverse_url(request, 'cmdb:user_list')
if request.method == 'GET':
return render(request, 'rbac/delete.html', {'cancel': origin_url})
models.UserInfo.objects.filter(id=pk).delete()
return redirect(origin_url)
def host_edit(request, pk):
obj = models.Host.objects.filter(id=pk).first()
if not obj:
return HttpResponse('主机不存在')
if request.method == 'GET':
form = HostModelForm(instance=obj)
return render(request, 'rbac/change.html', {'form': form})
form = HostModelForm(instance=obj, data=request.POST)
if form.is_valid():
form.save()
return redirect(memory_reverse_url(request, 'cmdb:host_list'))
return render(request, 'rbac/change.html', {'form': form})
def menu_edit(request, pk):
obj = models.Menu.objects.filter(id=pk).first()
if not obj:
return HttpResponse("菜单不存在")
if request.method == "GET":
form = MenuModelForm(instance=obj)
return render(request, 'rbac/change.html', {"form": form})
form = MenuModelForm(instance=obj, data=request.POST)
if form.is_valid():
form.save()
return redirect(memory_reverse_url(request, 'rbac:menu_list'))
return render(request, 'rbac/change.html', {"form": form})
def second_menu_del(request, pk):
"""
删除二级菜单
:param request:
:param pk:
:return:
"""
# 用户点击取消删除角色按钮以后就回到role_list 界面
origin_url = memory_reverse_url(request, 'rbac:menu_list')
if request.method == "GET":
return render(request, 'rbac/delete.html', {"cancel": origin_url})
models.Permission.objects.filter(id=pk).delete()
return redirect(origin_url)
def user_add(request):
"""
添加角色
:param request:
:return:
"""
if request.method == 'GET':
form = UserModelForm()
return render(request, 'rbac/change.html', {'form': form})
form = UserModelForm(data=request.POST)
if form.is_valid():
form.save()
return redirect(memory_reverse_url(request, 'cmdb:user_list'))
return render(request, 'rbac/change.html', {'form': form})
def menu_add(request):
"""
:param request:
:return:
"""
if request.method == "GET":
form = MenuModelForm()
return render(request, 'rbac/change.html', {'form':form})
form = MenuModelForm(data=request.POST)
if form.is_valid():
form.save()
return redirect(memory_reverse_url(request,'rbac:menu_list'))
return render(request, 'rbac/change.html', {'form': form}) # 如果输入为空，则在form 中有errors的错误信息
def second_menu_add(request, menu_id):
"""
添加二级菜单
:param request:
:param menu_id: 已选择的一级菜单ID(用于设置默认值)
:return:
"""
menu_obj = models.Menu.objects.filter(pk=menu_id).first()
if request.method == "GET":
form = SecondMenuModelForm(initial={'menu': menu_obj})
return render(request, 'rbac/change.html', {'form': form})
form = SecondMenuModelForm(data=request.POST)
if form.is_valid():
form.save()
return redirect(memory_reverse_url(request, 'rbac:menu_list'))
return render(request, 'rbac/change.html', {'form': form})
def permission_edit(request, pk):
"""
编辑
:param request:
:param pk:
:return:
"""
obj = models.Permission.objects.filter(id=pk).first()
if not obj:
return HttpResponse("菜单不存在")
if request.method == "GET":
form = PermissionModelForm(instance=obj)
return render(request, 'rbac/change.html', {"form": form})
form = PermissionModelForm(instance=obj, data=request.POST)
if form.is_valid():
form.save()
return redirect(memory_reverse_url(request, 'rbac:menu_list'))
return render(request, 'rbac/change.html', {"form": form})
def user_reset_pwd(request, pk):
"""
重置密码
:param request:
:param pk:
:return:
"""
obj = models.UserInfo.objects.filter(id=pk).first()
if not obj:
return HttpResponse('用户不存在')
if request.method == 'GET':
form = ResetPasswordUserModelForm()
return render(request, 'rbac/change.html', {'form': form})
form = ResetPasswordUserModelForm(instance=obj, data=request.POST)
if form.is_valid():
form.save()
return redirect(memory_reverse_url(request, 'cmdb:user_list'))
return render(request, 'rbac/change.html', {'form': form})
def permission_add(request,second_menu_id):
"""
增加权限信息
:param request:
:param second_menu_id: 关联的父级菜单ID
:return:
"""
if request.method == "GET":
form = PermissionModelForm()
return render(request, 'rbac/change.html', {'form':form})
form = PermissionModelForm(data=request.POST)
if form.is_valid():
second_menu_object = models.Permission.objects.filter(id=second_menu_id).first()
if not second_menu_object:
return HttpResponse("二级菜单不存在，请重新选择！")
# form.instance 中包含了用户提交的所有值
# 这里相当于是增加了一个pid的值
form.instance.pid = second_menu_object
form.save()
return redirect(memory_reverse_url(request,'rbac:menu_list'))
return render(request, 'rbac/change.html', {'form': form})
