def asset_detail(request, id, format=None): try: snippet = Assets.objects.get(id=id) except Assets.DoesNotExist: return Response(status=status.HTTP_404_NOT_FOUND) if request.method == 'GET': serializer = AssetsSerializer(snippet) return Response(serializer.data) elif request.method == 'PUT': serializer = AssetsSerializer(snippet, data=request.data) if serializer.is_valid(): serializer.save() recordAssets.delay(user=str(request.user), content="更新资产:{name}".format(name=snippet.name), type="assets", id=id) return Response(serializer.data) return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST) elif request.method == 'DELETE': if not request.user.has_perm('OpsManage.delete_asset_assets'): return Response(status=status.HTTP_403_FORBIDDEN) snippet.delete() recordAssets.delay(user=str(request.user), content="删除资产:{name}".format(name=snippet.name), type="assets", id=id) return Response(status=status.HTTP_204_NO_CONTENT)
def group_detail(request, id, format=None): """ Retrieve, update or delete a server assets instance. """ try: snippet = RoleList.objects.get(id=id) except RoleList.DoesNotExist: return Response(status=status.HTTP_404_NOT_FOUND) if request.method == 'GET': serializer = GroupSerializer(snippet) return Response(serializer.data) elif request.method == 'PUT': serializer = GroupSerializer(snippet, data=request.data) old_name = snippet.name if serializer.is_valid(): serializer.save() recordAssets.delay( user=str(request.user), content="修改部门名称:{old_name} -> {group_name}".format( old_name=old_name, group_name=request.data.get("name")), type="group", id=id) return Response(serializer.data) return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST) elif request.method == 'DELETE': snippet.delete() recordAssets.delay( user=str(request.user), content="删除部门:{group_name}".format(group_name=snippet.name), type="group", id=id) return Response(status=status.HTTP_204_NO_CONTENT)
def permission_detail(request, id, format=None): try: snippet = PermissonList.objects.get(id=id) except PermissonList.DoesNotExist: return Response(status=status.HTTP_404_NOT_FOUND) if request.method == 'GET': serializer = PermissionSerializer(snippet) return Response(serializer.data) elif request.method == 'PUT': serializer = PermissionSerializer(snippet, data=request.data) old_name = snippet.name if serializer.is_valid(): serializer.save() recordAssets.delay( user=str(request.user), content="修改权限名称:{old_name} -> {per_name}".format( old_name=old_name, per_name=request.data.get("name")), type="group", id=id) return Response(serializer.data) return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST) elif request.method == 'DELETE': if not request.user.has_perm('opman.delete_group'): return Response(status=status.HTTP_403_FORBIDDEN) snippet.delete() recordAssets.delay( user=str(request.user), content="删除权限:{per_name}".format(per_name=snippet.name), type="group", id=id) return Response(status=status.HTTP_204_NO_CONTENT)
def asset_server_detail(request, id, format=None): """ Retrieve, update or delete a server assets instance. """ try: snippet = Server_Assets.objects.get(id=id) except Server_Assets.DoesNotExist: return Response(status=status.HTTP_404_NOT_FOUND) if request.method == 'GET': serializer = ServerSerializer(snippet) return Response(serializer.data) elif request.method == 'PUT': '''如果更新字段包含assets则先更新总资产表''' if (request.data.get('data')): data = request.data.get('data') else: data = request.data if (data.get('assets')): assets_data = data.pop('assets') try: assets_snippet = Assets.objects.get(id=snippet.assets.id) assets = AssetsSerializer(assets_snippet, data=assets_data) except Assets.DoesNotExist: return Response(status=status.HTTP_404_NOT_FOUND) if assets.is_valid(): assets.save() recordAssets.delay( user=str(request.user), content="修改服务器资产:{ip}".format(ip=snippet.ip), type="server", id=id) serializer = ServerSerializer(snippet, data=data) if serializer.is_valid(): serializer.save() return Response(serializer.data) return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST) elif request.method == 'DELETE': if not request.user.has_perm('OpsManage.can_delete_server_assets'): return Response(status=status.HTTP_403_FORBIDDEN) snippet.delete() try: assets_snippet = Assets.objects.get(id=snippet.assets.id) assets_snippet.delete() recordAssets.delay(user=str(request.user), content="删除服务器资产:{ip}".format(ip=snippet.ip), type="server", id=id) except Assets.DoesNotExist: pass return Response(status=status.HTTP_204_NO_CONTENT)
def group_list(request, format=None): if request.method == 'GET': snippets = RoleList.objects.all() serializer = GroupSerializer(snippets, many=True) return Response(serializer.data) elif request.method == 'POST': serializer = GroupSerializer(data=request.data) if serializer.is_valid(): serializer.save() recordAssets.delay(user=str(request.user), content="添加部门:{group_name}".format( group_name=request.data.get("name")), type="group", id=serializer.data.get('id')) return Response(serializer.data, status=status.HTTP_201_CREATED) return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
def business_list(request, format=None): if request.method == 'GET': snippets = Business_Assets.objects.all() serializer = BusinessSerializer(snippets, many=True) return Response(serializer.data) elif request.method == 'POST': serializer = BusinessSerializer(data=request.data) if serializer.is_valid(): serializer.save() recordAssets.delay( user=str(request.user), content="添加业务分组名称:{business_name}".format( business_name=request.data.get("business_name")), type="business", id=serializer.data.get('id')) return Response(serializer.data, status=status.HTTP_201_CREATED) return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
def raid_list(request, format=None): """ List all order, or create a server assets order. """ if request.method == 'GET': snippets = Raid_Assets.objects.all() serializer = RaidSerializer(snippets, many=True) return Response(serializer.data) elif request.method == 'POST': serializer = RaidSerializer(data=request.data) if serializer.is_valid(): serializer.save() recordAssets.delay(user=str(request.user), content="添加Raid类型:{raid_name}".format( raid_name=request.data.get("raid_name")), type="raid", id=serializer.data.get('id')) return Response(serializer.data, status=status.HTTP_201_CREATED) return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
def asset_server_list(request, format=None): if request.method == 'GET': snippets = Server_Assets.objects.all() serializer = ServerSerializer(snippets, many=True) return Response(serializer.data) elif request.method == 'POST': if (request.data.get('data')): data = request.data.get('data') else: data = request.data print(data) serializer = ServerSerializer(data=data) if serializer.is_valid(): serializer.save() recordAssets.delay( user=str(request.user), content="添加服务器资产:{ip}".format(ip=data.get("ip")), type="server", id=serializer.data.get('id')) return Response(serializer.data, status=status.HTTP_201_CREATED) return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
def group_list(request, format=None): """ List all order, or create a server assets order. """ if request.method == 'GET': snippets = RoleList.objects.all() serializer = GroupSerializer(snippets, many=True) return Response(serializer.data) elif request.method == 'POST': if not request.user.has_perm('Opsmanage.change_group'): return Response(status=status.HTTP_403_FORBIDDEN) serializer = GroupSerializer(data=request.data) if serializer.is_valid(): serializer.save() recordAssets.delay(user=str(request.user), content="添加用户组:{group_name}".format( group_name=request.data.get("name")), type="group", id=serializer.data.get('id')) return Response(serializer.data, status=status.HTTP_201_CREATED) return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
def service_detail(request, id, format=None): """ Retrieve, update or delete a server assets instance. """ try: snippet = Service_Assets.objects.get(id=id) except Service_Assets.DoesNotExist: return Response(status=status.HTTP_404_NOT_FOUND) if request.method == 'GET': serializer = ServiceSerializer(snippet) return Response(serializer.data) elif request.method == 'PUT': serializer = ServiceSerializer(snippet, data=request.data) old_name = snippet.service_name if serializer.is_valid(): serializer.save() recordAssets.delay( user=str(request.user), content="修改业务类型为:{old_name} -> {service_name}".format( old_name=old_name, service_name=request.data.get("service_name")), type="service", id=id) return Response(serializer.data) return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST) elif request.method == 'DELETE' and request.user.has_perm( 'OpsManage.can_delete_assets'): if not request.user.has_perm('OpsManage.can_delete_service_assets'): return Response(status=status.HTTP_403_FORBIDDEN) snippet.delete() recordAssets.delay(user=str(request.user), content="删除业务类型:{service_name}".format( service_name=snippet.service_name), type="service", id=id) return Response(status=status.HTTP_204_NO_CONTENT)
def asset_net_list(request, format=None): """ List all order, or create a new net assets. """ if request.method == 'GET': snippets = Network_Assets.objects.all() serializer = NetworkSerializer(snippets, many=True) return Response(serializer.data) elif request.method == 'POST': if (request.data.get('data')): data = request.data.get('data') else: data = request.data serializer = NetworkSerializer(data=data) if serializer.is_valid(): serializer.save() recordAssets.delay( user=str(request.user), content="添加网络设备资产:{ip}".format(ip=data.get("ip")), type="net", id=serializer.data.get('id')) return Response(serializer.data, status=status.HTTP_201_CREATED) return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
def assets_facts(request, args=None): if request.method == "POST" and request.user.has_perm('OpsManage.change_server_assets'): server_id = request.POST.get('server_id') genre = request.POST.get('type') if genre == 'setup': try: server_assets = Server_Assets.objects.get(id=request.POST.get('server_id')) if server_assets.keyfile == 1: resource = [{"hostname": server_assets.ip, "port": int(server_assets.port)}] else: resource = [ {"hostname": server_assets.ip, "port": server_assets.port, "username": server_assets.username, "password": server_assets.passwd}] except Exception as e: return JsonResponse({'msg': "数据更新失败-查询不到该主机资料~", "code": 502}) ANS = ANSRunner(resource) ANS.run_model(host_list=[server_assets.ip], module_name='setup', module_args="") data = ANS.handle_cmdb_data(ANS.get_model_result()) if data: for ds in data: status = ds.get('status') if status == 0: try: Assets.objects.filter(id=server_assets.assets_id).update(sn=ds.get('serial'), model=ds.get('model'), manufacturer=ds.get( 'manufacturer')) except Exception as e: return JsonResponse({'msg': "数据更新失败-查询不到该主机的资产信息", "code": 403}) try: Server_Assets.objects.filter(id=server_id).update(cpu_number=ds.get('cpu_number'), kernel=ds.get('kernel'), selinux=ds.get('selinux'), hostname=ds.get('hostname'), system=ds.get('system'), cpu=ds.get('cpu'), disk_total=ds.get('disk_total'), cpu_core=ds.get('cpu_core'), swap=ds.get('swap'), ram_total=ds.get('ram_total'), vcpu_number=ds.get('vcpu_number') ) recordAssets.delay(user=str(request.user), content="修改服务器资产:{ip}".format(ip=server_assets.ip), type="server", id=server_assets.id) except Exception as e: print(e) return JsonResponse({'msg': "数据更新失败-写入数据失败", "code": 400}) else: return JsonResponse({'msg': "数据更新失败-无法链接主机~", "code": 502}) return JsonResponse({'msg': "数据更新成功", "code": 200}) else: return JsonResponse({'msg': "数据更新失败-请检查Ansible配置", "code": 400}) elif genre == 'crawHw': try: server_assets = Server_Assets.objects.get(id=server_id) assets = Assets.objects.get(id=server_assets.assets_id) if server_assets.keyfile == 1: resource = [{"hostname": server_assets.ip, "port": int(server_assets.port)}] else: resource = [ {"hostname": server_assets.ip, "port": server_assets.port, "username": server_assets.username, "password": server_assets.passwd}] except Exception as e: return JsonResponse({'msg': "数据更新失败-查询不到该主机资料~", "code": 502}) ANS = ANSRunner(resource) ANS.run_model(host_list=[server_assets.ip], module_name='crawHw', module_args="") data = ANS.handle_cmdb_crawHw_data(ANS.get_model_result()) if data: for ds in data: if ds.get('mem_info'): for mem in ds.get('mem_info'): if Ram_Assets.objects.filter(assets=assets, device_slot=mem.get('slot')).count() > 0: try: Ram_Assets.objects.filter(assets=assets, device_slot=mem.get('slot')).update( device_slot=mem.get('slot'), device_model=mem.get('serial'), device_brand=mem.get('manufacturer'), device_volume=mem.get('size'), device_status="Online" ) except Exception as e: return JsonResponse({'msg': "数据更新失败-写入数据失败", "code": 400}) else: try: Ram_Assets.objects.create(device_slot=mem.get('slot'), device_model=mem.get('serial'), device_brand=mem.get('manufacturer'), device_volume=mem.get('size'), device_status="Online", assets=assets ) recordAssets.delay(user=str(request.user), content="修改服务器资产:{ip}".format(ip=server_assets.ip), type="server", id=server_assets.id) except Exception as e: return JsonResponse({'msg': "数据更新失败-写入数据失败", "code": 400}) if ds.get('disk_info'): for disk in ds.get('disk_info'): if Disk_Assets.objects.filter(assets=assets, device_slot=disk.get('slot')).count() > 0: try: Disk_Assets.objects.filter(assets=assets, device_slot=disk.get('slot')).update( device_serial=disk.get('serial'), device_model=disk.get('model'), device_brand=disk.get('manufacturer'), device_volume=disk.get('size'), device_status="Online" ) except Exception as e: return JsonResponse({'msg': "数据更新失败-写入数据失败", "code": 400}) else: try: Disk_Assets.objects.create(device_serial=disk.get('serial'), device_model=disk.get('model'), device_brand=disk.get('manufacturer'), device_volume=disk.get('size'), device_status="Online", assets=assets, device_slot=disk.get('slot') ) recordAssets.delay(user=str(request.user), content="修改服务器资产:{ip}".format(ip=server_assets.ip), type="server", id=server_assets.id) except Exception as e: return JsonResponse({'msg': "数据更新失败-写入数据失败", "code": 400}) return JsonResponse({'msg': "数据更新成功", "code": 200}) else: return JsonResponse({'msg': "数据更新失败,系统可能不支持,未能获取数据", "code": 400}) else: return JsonResponse({'msg': "您没有该项操作的权限~", "code": 400})