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})
