jumpserver/jasset/views.py

331 lines
10 KiB
Python
Raw Normal View History

# coding:utf-8
2015-02-09 11:02:25 +00:00
2015-03-19 10:32:10 +00:00
import ast
2015-02-09 11:02:25 +00:00
from django.db.models import Q
from django.template import RequestContext
from django.shortcuts import get_object_or_404
2015-09-08 16:19:17 +00:00
from jasset.asset_api import *
2015-03-26 10:42:52 +00:00
from jumpserver.api import *
2015-10-31 01:23:10 +00:00
from jasset.forms import AssetForm
2015-11-03 15:53:12 +00:00
from jasset.models import Asset, IDC, AssetGroup, ASSET_TYPE, ASSET_STATUS
2014-12-22 09:18:51 +00:00
2015-09-08 16:19:17 +00:00
@require_role('admin')
def group_add(request):
"""
Add asset group
添加资产组
"""
header_title, path1, path2 = u'添加资产组', u'资产管理', u'添加资产组'
asset_all = Asset.objects.all()
2015-03-24 10:34:00 +00:00
if request.method == 'POST':
2015-09-08 16:19:17 +00:00
name = request.POST.get('name', '')
asset_select = request.POST.getlist('asset_select', [])
comment = request.POST.get('comment', '')
2015-09-08 16:19:17 +00:00
try:
if not name:
error = u'组名不能为空'
raise ServerError(error)
2015-09-08 16:19:17 +00:00
asset_group_test = get_object(AssetGroup, name=name)
if asset_group_test:
error = u"该组名 %s 已存在" % name
raise ServerError(error)
2015-04-14 09:25:56 +00:00
2015-09-08 16:19:17 +00:00
except ServerError:
pass
2015-11-03 09:07:46 +00:00
2015-09-08 16:19:17 +00:00
else:
db_add_group(name=name, comment=comment, asset_select=asset_select)
2015-11-03 09:07:46 +00:00
smg = u"主机组 %s 添加成功" % name
2015-04-14 09:25:56 +00:00
2015-09-08 16:19:17 +00:00
return my_render('jasset/group_add.html', locals(), request)
2015-04-14 09:25:56 +00:00
2015-11-03 09:07:46 +00:00
@require_role('admin')
def group_edit(request):
"""
Edit asset group
编辑资产组
"""
header_title, path1, path2 = u'编辑主机组', u'资产管理', u'编辑主机组'
group_id = request.GET.get('id', '')
group = get_object(AssetGroup, id=group_id)
asset_all = Asset.objects.all()
asset_select = Asset.objects.filter(group=group)
asset_no_select = [a for a in asset_all if a not in asset_select]
if request.method == 'POST':
name = request.POST.get('name', '')
asset_select = request.POST.getlist('asset_select', [])
comment = request.POST.get('comment', '')
try:
if not name:
emg = u'组名不能为空'
raise ServerError(emg)
if group.name != name:
asset_group_test = get_object(AssetGroup, name=name)
if asset_group_test:
emg = u"该组名 %s 已存在" % name
raise ServerError(emg)
except ServerError:
pass
else:
group.asset_set.clear()
db_update_group(id=group_id, name=name, comment=comment, asset_select=asset_select)
smg = u"主机组 %s 添加成功" % name
return HttpResponseRedirect('/jasset/group_list')
return my_render('jasset/group_edit.html', locals(), request)
@require_role('admin')
def group_detail(request):
""" 主机组详情 """
header_title, path1, path2 = u'主机组详情', u'资产管理', u'主机组详情'
group_id = request.GET.get('id', '')
group = get_object(AssetGroup, id=group_id)
asset_all = Asset.objects.filter(group=group).order_by('ip')
contact_list, p, contacts, page_range, current_page, show_first, show_end = pages(asset_all, request)
return my_render('jasset/group_detail.html', locals(), request)
2015-09-08 16:19:17 +00:00
@require_role('admin')
def group_list(request):
"""
list asset group
列出资产组
"""
header_title, path1, path2 = u'查看资产组', u'资产管理', u'查看资产组'
keyword = request.GET.get('keyword', '')
gid = request.GET.get('gid')
sid = request.GET.get('sid')
asset_group_list = AssetGroup.objects.all()
2015-02-05 10:53:57 +00:00
2015-09-08 16:19:17 +00:00
if keyword:
2015-09-10 14:51:42 +00:00
asset_group_list = asset_group_list.filter(Q(name__contains=keyword) | Q(comment__contains=keyword))
2015-02-05 10:53:57 +00:00
2015-09-08 16:19:17 +00:00
asset_group_list, p, asset_groups, page_range, current_page, show_first, show_end = pages(asset_group_list, request)
return my_render('jasset/group_list.html', locals(), request)
2015-09-10 14:51:42 +00:00
@require_role('admin')
def group_del(request):
"""
del asset group
删除主机组
"""
group_ids = request.GET.get('id', '')
group_id_list = group_ids.split(',')
for group_id in group_id_list:
AssetGroup.objects.filter(id=group_id).delete()
return HttpResponse(u'删除成功')
2015-09-08 16:19:17 +00:00
@require_role('admin')
def asset_add(request):
"""
Asset add view
添加资产
"""
header_title, path1, path2 = u'添加资产', u'资产管理', u'添加资产'
2015-09-10 14:51:42 +00:00
asset_group_all = AssetGroup.objects.all()
2015-10-31 01:23:10 +00:00
af = AssetForm()
if request.method == 'POST':
2015-11-03 09:07:46 +00:00
af_post = AssetForm(request.POST)
2015-11-08 14:39:30 +00:00
print af_post
ip = request.POST.get('ip', '')
is_active = True if request.POST.get('is_active') == '1' else False
use_default_auth = request.POST.get('use_default_auth', '')
2015-09-08 16:19:17 +00:00
try:
if Asset.objects.filter(ip=str(ip)):
error = u'该IP %s 已存在!' % ip
raise ServerError(error)
2015-04-07 11:15:45 +00:00
2015-09-08 16:19:17 +00:00
except ServerError:
pass
2015-11-03 09:07:46 +00:00
else:
if af_post.is_valid():
asset_save = af_post.save(commit=False)
2015-11-08 14:39:30 +00:00
if not use_default_auth:
password = request.POST.get('password', '')
password_encode = CRYPTOR.encrypt(password)
asset_save.password = password_encode
asset_save.is_active = True if is_active else False
2015-11-03 09:07:46 +00:00
asset_save.save()
af_post.save_m2m()
2015-11-08 14:39:30 +00:00
2015-11-03 09:07:46 +00:00
msg = u'主机 %s 添加成功' % ip
else:
esg = u'主机 %s 添加失败' % ip
2015-09-08 16:19:17 +00:00
return my_render('jasset/asset_add.html', locals(), request)
@require_role('admin')
def asset_del(request):
"""
del a asset
删除主机
"""
asset_id = request.GET.get('id', '')
if asset_id:
Asset.objects.filter(id=asset_id).delete()
2015-11-03 15:53:12 +00:00
if request.method == 'POST':
asset_batch = request.GET.get('arg', '')
asset_id_all = str(request.POST.get('asset_id_all', ''))
if asset_batch:
for asset_id in asset_id_all.split(','):
asset = get_object(Asset, id=asset_id)
asset.delete()
return HttpResponse(u'删除成功')
2015-09-08 16:19:17 +00:00
@require_role(role='super')
def asset_edit(request):
2015-11-03 09:07:46 +00:00
"""
edit a asset
修改主机
"""
2015-09-08 16:19:17 +00:00
header_title, path1, path2 = u'修改资产', u'资产管理', u'修改资产'
2015-04-13 13:25:43 +00:00
2015-09-08 16:19:17 +00:00
asset_id = request.GET.get('id', '')
2015-11-08 14:39:30 +00:00
username = request.session.get('username', 'admin')
# if not asset_id:
# return HttpResponse('没有该主机')
2015-09-08 16:19:17 +00:00
asset = get_object(Asset, id=asset_id)
2015-11-03 09:07:46 +00:00
af = AssetForm(instance=asset)
2015-01-11 11:27:21 +00:00
if request.method == 'POST':
2015-11-03 09:07:46 +00:00
af_post = AssetForm(request.POST, instance=asset)
ip = request.POST.get('ip', '')
# ip = request.POST.get('ip')
# port = request.POST.get('port')
# groups = request.POST.getlist('groups')
# use_default_auth = True if request.POST.getlist('use_default_auth', []) else False
# is_active = True if request.POST.get('is_active') else False
# comment = request.POST.get('comment')
# if not use_default_auth:
2015-11-08 14:39:30 +00:00
# username = request.POST.get('username')
# password = request.POST.get('password')
2015-11-03 09:07:46 +00:00
# if password == asset.password:
# password_encode = password
# else:
# password_encode = CRYPTOR.encrypt(password)
# else:
# username = None
# password_encode = None
2015-01-11 11:27:21 +00:00
2015-04-13 13:25:43 +00:00
try:
2015-09-08 16:19:17 +00:00
asset_test = get_object(Asset, ip=ip)
2015-11-08 14:39:30 +00:00
if asset_test and asset_id != unicode(asset_test.id):
2015-09-08 16:19:17 +00:00
error = u'该IP %s 已存在!' % ip
raise ServerError(error)
except ServerError:
2015-04-13 13:25:43 +00:00
pass
2015-01-11 11:27:21 +00:00
else:
2015-11-03 09:07:46 +00:00
if af_post.is_valid():
af_save = af_post.save(commit=False)
af_save.save()
af_post.save_m2m()
2015-11-08 14:39:30 +00:00
info = asset_diff(af_post.__dict__.get('initial'), request.POST)
print info
db_asset_alert(asset, username, info)
2015-11-03 09:07:46 +00:00
msg = u'主机 %s 修改成功' % ip
else:
emg = u'主机 %s 修改失败' % ip
2015-09-08 16:19:17 +00:00
return HttpResponseRedirect('/jasset/asset_detail/?id=%s' % asset_id)
return my_render('jasset/asset_edit.html', locals(), request)
2015-11-03 15:53:12 +00:00
@require_role('user')
2015-11-08 14:39:30 +00:00
def asset_list(request):
2015-11-03 15:53:12 +00:00
"""
2015-11-08 14:39:30 +00:00
asset list view
2015-11-03 15:53:12 +00:00
"""
idc_all = IDC.objects.filter()
asset_group_all = AssetGroup.objects.all()
2015-11-08 14:39:30 +00:00
asset_types = ASSET_TYPE
2015-11-03 15:53:12 +00:00
asset_status = ASSET_STATUS
2015-09-08 16:19:17 +00:00
2015-11-03 15:53:12 +00:00
idc_name = request.GET.get('idc', '')
group_name = request.GET.get('group', '')
asset_type = request.GET.get('asset_type', '')
status = request.GET.get('status', '')
keyword = request.GET.get('keyword', '')
2015-11-08 14:39:30 +00:00
export = request.GET.get("export", False)
2015-09-08 16:19:17 +00:00
2015-11-08 14:39:30 +00:00
asset_find = Asset.objects.all()
if idc_name:
asset_find = asset_find.filter(idc__name__contains=idc_name)
if group_name:
asset_find = asset_find.filter(group__name__contains=group_name)
if asset_type:
asset_find = asset_find.filter(asset_type__contains=asset_type)
if status:
asset_find = asset_find.filter(status__contains=status)
if keyword:
2015-11-03 15:53:12 +00:00
asset_find = asset_find.filter(
Q(hostname__contains=keyword) |
2015-11-08 14:39:30 +00:00
Q(other_ip__contains=keyword) |
2015-11-03 15:53:12 +00:00
Q(ip__contains=keyword) |
Q(remote_ip__contains=keyword) |
Q(comment__contains=keyword) |
Q(group__name__contains=keyword) |
Q(cpu__contains=keyword) |
Q(memory__contains=keyword) |
Q(disk__contains=keyword))
2015-11-08 14:39:30 +00:00
if export:
s = write_excel(asset_find)
if s[0]:
file_name = s[1]
smg = 'excel文件已生成请点击下载!'
return my_render('jasset/asset_excel_download.html', locals(), request)
assets_list, p, assets, page_range, current_page, show_first, show_end = pages(asset_find, request)
return my_render('jasset/asset_list.html', locals(), request)
2015-11-03 15:53:12 +00:00
2015-11-08 14:39:30 +00:00
@require_role('admin')
def asset_edit_batch(request):
af = AssetForm()
asset_group_all = AssetGroup.objects.all()
return my_render('jasset/asset_edit_batch.html', locals(), request)
@require_role('admin')
def asset_detail(request):
"""
主机详情
"""
header_title, path1, path2 = u'主机详细信息', u'资产管理', u'主机详情'
asset_id = request.GET.get('id', '')
asset = get_object(Asset, id=asset_id)
asset_record = AssetRecord.objects.filter(asset=asset).order_by('-alert_time')
return my_render('jasset/asset_detail.html', locals(), request)