def update(self, request, *args, **kwargs):
""" 修改资产的数据 """
id = self.kwargs.get("pk", None)
print(request.data)
asset = Assets.objects.filter(id=id).first()
if asset:
try:
asset.ip = request.data.get("ip")
asset.asset_type = request.data.get("asset_type")
asset.status = request.data.get("status")
asset.system = request.data.get("system")
asset.remark = request.data.get("remark")
asset.sshuser = request.data.get("sshuser")
asset.verify_type = int(request.data.get("verify_type"))
asset.sshpwd = request.data.get("sshpwd")
asset.key = request.data.get("sshkey")
asset.sshport = request.data.get("sshport")
asset.mysql_user = request.data.get("mysql_user")
asset.mysql_pwd = request.data.get("mysql_pwd")
asset.mysql_port = request.data.get("mysql_port")
asset.ftp_port = request.data.get("ftp_port")
# 关联
id = request.data.get("host_machine")
if id:
asset.host_machine = Assets.objects.filter(id=id).first()
else:
asset.host_machine = None
opsid = request.data.get("opsuser")
if opsid:
asset.opsuser = User.objects.filter(id=opsid).first()
else:
asset.opsuser = None
userid = request.data.get("user")
if userid:
asset.user = User.objects.filter(id=userid).first()
else:
asset.user = None
idc_id = request.data.get("idc")
if idc_id:
asset.idc = IDC.objects.filter(id=idc_id).first()
else:
asset.idc = None
asset.save()
except IntegrityError as e:
raise CodeError(str(e))
else:
# 注册 tag
tags = request.data.get("tags")
if tags:
tags = tags.split(",")
asset.tags.set(tags)
asset.save()
code = 0
msg = "success"
else:
raise CodeError("未找到!")
return Response("添加成功")
def update(self, request, *args, **kwargs):
id = self.kwargs.get("id", None)
role = get_object_or_404(Role, id=id)
title = request.data.get("title", None)
status = request.data.get("status", True)
if title:
if title != role.title:
role.title = title
elif status != role.status:
role.status = status
else:
raise CodeError("没有变化！")
role.save()
return Response({"message": "成功"})
raise CodeError("缺少参数！")
def create(self, request, *args, **kwargs):
title = request.data.get("title", None)
status = request.data.get("status", True)
url = request.data.get("url", None)
method = request.data.get("method", None)
if title and url and method:
try:
p = UrlPermission(title=title,
status=status,
method=method,
url=url)
p.save()
except Exception as e:
raise CodeError(str(e))
return Response({"message": "成功"})
raise CodeError("验证失败")
def update(self, request, *args, **kwargs):
id = self.kwargs.get("id", None)
m_obj = get_object_or_404(Menu, id=id)
title = request.data.get("title", None)
url = request.data.get("url", None)
path = request.data.get("path", None)
name = request.data.get("name", None)
icon = request.data.get("icon", None)
status = request.data.get("status", True)
is_parent = request.data.get("is_parent", True)
if status == "true":
status = True
else:
status = False
if is_parent == "true":
is_parent = True
else:
is_parent = False
parent = request.data.get("parent", None)
p = Menu.objects.filter(id=parent).first()
if title and url and path and name:
try:
m_obj.title = title
m_obj.url = url
m_obj.path = path
m_obj.status = status
m_obj.icon = icon
m_obj.is_parent = is_parent
m_obj.parent = p
m_obj.save()
except Exception as e:
raise CodeError(str(e))
return Response({"message": "修改成功"})
def create(self, request, *args, **kwargs):
title = request.data.get("title", None)
status = request.data.get("status", True)
if title:
role = Role(title=title, status=status)
role.save()
else:
raise CodeError("错误的参数")
return Response({"message": "成功"})
def put(self, request, **kwargs):
id = self.kwargs.get("key")
user = get_object_or_404(User, id=id)
selected = request.data.get("selected")
try:
user.roles.set(selected)
except Exception as e:
raise CodeError("修改失败:" + str(e))
return Response({"message": "success!"})
def post(self, request, **kwargs):
id = self.kwargs.get("id", None)
pids = request.data.get("ids", None)
role = get_object_or_404(Role, id=id)
try:
role.permission.set(pids)
role.save()
except Exception as e:
raise CodeError("出错" + str(e))
return Response({"message": "success"})
def get(self, request, **kwargs):
id = self.kwargs.get("id", None)
role = Role.objects.filter(id=id).first()
role_menus = []
if role:
role_id = [p.id for p in role.menu.all()]
else:
raise CodeError("未找到！")
for menu in Menu.objects.all():
d = {
"value": menu.id,
"title": "%s : %s" % (menu.title, menu.path),
}
role_menus.append(d)
return Response({"menus": role_menus, "checked": role_id})
def create(self, request, *args, **kwargs):
"""
创建资产
"""
asset = Assets()
try:
asset.ip = request.data.get("ip")
asset.asset_type = request.data.get("asset_type")
asset.status = request.data.get("status")
asset.system = request.data.get("system")
asset.remark = request.data.get("remark")
asset.sshuser = request.data.get("sshuser")
asset.verify_type = int(request.data.get("verify_type"))
asset.sshpwd = request.data.get("sshpwd")
asset.sshport = request.data.get("sshport")
asset.mysql_user = request.data.get("mysql_user")
asset.mysql_pwd = request.data.get("mysql_pwd")
asset.mysql_port = request.data.get("mysql_port")
asset.ftp_port = request.data.get("ftp_port")
# 关联
id = request.data.get("host_machine")
if id:
asset.host_machine = Assets.objects.filter(id=id).first()
opsid = request.data.get("opsuser")
if opsid:
asset.opsuser = User.objects.filter(id=opsid).first()
userid = request.data.get("user")
if userid:
asset.user = User.objects.filter(id=userid).first()
idc_id = request.data.get("idc")
if idc_id:
asset.idc = IDC.objects.filter(id=idc_id).first()
asset.save()
except IntegrityError as e:
raise CodeError("重复添加")
else:
# 注册 tag
tags = request.data.get("tag")
if tags:
tags = tags.split(",")
asset.tags.set(tags)
asset.save()
return Response("添加成功")
def get(self, request, **kwargs):
""" 返回值 已选权限/全部权限"""
id = self.kwargs.get("id", None)
role = Role.objects.filter(id=id).first()
role_permissions = []
if role:
role_id = [p.id for p in role.permission.all()]
else:
raise CodeError("未找到！")
for permission in UrlPermission.objects.all():
d = {
"value": permission.id,
"title": "%s %s:%s" % (permission.title, permission.url, permission.method),
}
role_permissions.append(d)
return Response({"permissions": role_permissions, "checked": role_id})
def update(self, request, *args, **kwargs):
id = self.kwargs.get("id", None)
p_obj = get_object_or_404(UrlPermission, id=id)
title = request.data.get("title", None)
url = request.data.get("url", None)
method = request.data.get("method", None)
status = request.data.get("status", True)
if p_obj and title and url and method:
try:
p_obj.title = title
p_obj.url = url
p_obj.method = method
p_obj.status = status
p_obj.save()
except Exception as e:
raise CodeError(str(e))
return Response({"message": "修改成功"})