mirror of https://github.com/jumpserver/jumpserver
551 lines
19 KiB
Python
551 lines
19 KiB
Python
# coding:utf-8
|
|
from __future__ import absolute_import, unicode_literals
|
|
|
|
from django.utils.translation import ugettext as _
|
|
from django.conf import settings
|
|
from django.db.models import Q
|
|
from django.views.generic import TemplateView, ListView
|
|
from django.views.generic.edit import CreateView, DeleteView, FormView, UpdateView
|
|
from django.urls import reverse_lazy
|
|
from django.contrib.messages.views import SuccessMessageMixin
|
|
from django.views.generic.detail import DetailView, SingleObjectMixin
|
|
from django.shortcuts import get_object_or_404, reverse, redirect
|
|
|
|
from common.utils import int_seq
|
|
from .models import Asset, AssetGroup, IDC, AssetExtend, AdminUser, SystemUser, Tag
|
|
from .forms import AssetCreateForm, AssetGroupForm, IDCForm, AdminUserForm, SystemUserForm
|
|
from .hands import AdminUserRequiredMixin
|
|
|
|
|
|
class AssetListView(AdminUserRequiredMixin, ListView):
|
|
paginate_by = settings.CONFIG.DISPLAY_PER_PAGE
|
|
model = Asset
|
|
context_object_name = 'asset_list'
|
|
template_name = 'assets/asset_list.html'
|
|
|
|
def get_queryset(self):
|
|
queryset = super(AssetListView, self).get_queryset()
|
|
queryset = sorted(queryset, key=self.sorted_by_valid_and_ip)
|
|
return queryset
|
|
|
|
@staticmethod
|
|
def sorted_by_valid_and_ip(asset):
|
|
ip_list = int_seq(asset.ip.split('.'))
|
|
ip_list.insert(0, asset.is_valid()[0])
|
|
return ip_list
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = {
|
|
'app': 'Assets',
|
|
'action': 'Asset list',
|
|
}
|
|
kwargs.update(context)
|
|
return super(AssetListView, self).get_context_data(**kwargs)
|
|
|
|
|
|
class AssetCreateView(AdminUserRequiredMixin, CreateView):
|
|
model = Asset
|
|
form_class = AssetCreateForm
|
|
template_name = 'assets/asset_create.html'
|
|
success_url = reverse_lazy('assets:asset-list')
|
|
|
|
def form_invalid(self, form):
|
|
print(form.errors)
|
|
return super(AssetCreateView, self).form_invalid(form)
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = {
|
|
'app': 'Assets',
|
|
'action': 'Create asset',
|
|
}
|
|
kwargs.update(context)
|
|
return super(AssetCreateView, self).get_context_data(**kwargs)
|
|
|
|
|
|
class AssetUpdateView(AdminUserRequiredMixin, UpdateView):
|
|
model = Asset
|
|
form_class = AssetCreateForm
|
|
template_name = 'assets/asset_update.html'
|
|
success_url = reverse_lazy('assets:asset-list')
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = {
|
|
'app': 'Assets',
|
|
'action': 'Update asset',
|
|
}
|
|
kwargs.update(context)
|
|
return super(AssetUpdateView, self).get_context_data(**kwargs)
|
|
|
|
|
|
class AssetDeleteView(DeleteView):
|
|
model = Asset
|
|
template_name = 'assets/delete_confirm.html'
|
|
success_url = reverse_lazy('assets:asset-list')
|
|
|
|
|
|
class AssetDetailView(DetailView):
|
|
model = Asset
|
|
context_object_name = 'asset'
|
|
template_name = 'assets/asset_detail.html'
|
|
|
|
def get_context_data(self, **kwargs):
|
|
asset_groups = self.object.groups.all()
|
|
context = {
|
|
'app': 'Assets',
|
|
'action': 'Asset detail',
|
|
'asset_groups_remain': [asset_group for asset_group in AssetGroup.objects.all()
|
|
if asset_group not in asset_groups],
|
|
'asset_groups': asset_groups,
|
|
}
|
|
kwargs.update(context)
|
|
return super(AssetDetailView, self).get_context_data(**kwargs)
|
|
|
|
class AssetModalListView(AdminUserRequiredMixin, ListView):
|
|
paginate_by = settings.CONFIG.DISPLAY_PER_PAGE
|
|
model = Asset
|
|
context_object_name = 'asset_modal_list'
|
|
template_name = 'assets/asset_modal_list.html'
|
|
|
|
|
|
def get_context_data(self, **kwargs):
|
|
group_id = self.request.GET.get('group_id')
|
|
if group_id:
|
|
group = AssetGroup.objects.get(id=group_id)
|
|
context = {
|
|
'group_assets':[x.id for x in group.assets.all()]
|
|
}
|
|
kwargs.update(context)
|
|
return super(AssetModalListView, self).get_context_data(**kwargs)
|
|
|
|
class AssetGroupCreateView(AdminUserRequiredMixin, CreateView):
|
|
model = AssetGroup
|
|
form_class = AssetGroupForm
|
|
template_name = 'assets/asset_group_create.html'
|
|
success_url = reverse_lazy('assets:asset-group-list')
|
|
|
|
# Todo: Asset group create template select assets so hard, need be resolve next
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = {
|
|
'app': _('Assets'),
|
|
'action': _('Create asset group'),
|
|
'assets_count': 0,
|
|
}
|
|
kwargs.update(context)
|
|
return super(AssetGroupCreateView, self).get_context_data(**kwargs)
|
|
|
|
|
|
def form_valid(self, form):
|
|
asset_group = form.save()
|
|
assets_id_list = self.request.POST.getlist('assets', [])
|
|
assets = [get_object_or_404(Asset, id=int(asset_id)) for asset_id in assets_id_list]
|
|
print assets
|
|
asset_group.created_by = self.request.user.username or 'Admin'
|
|
asset_group.assets.add(*tuple(assets))
|
|
asset_group.save()
|
|
return super(AssetGroupCreateView, self).form_valid(form)
|
|
|
|
class AssetGroupListView(AdminUserRequiredMixin, ListView):
|
|
model = AssetGroup
|
|
paginate_by = settings.CONFIG.DISPLAY_PER_PAGE
|
|
context_object_name = 'asset_group_list'
|
|
template_name = 'assets/asset_group_list.html'
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = {
|
|
'app': _('Assets'),
|
|
'action': _('Asset group list'),
|
|
'keyword': self.request.GET.get('keyword', '')
|
|
}
|
|
kwargs.update(context)
|
|
return super(AssetGroupListView, self).get_context_data(**kwargs)
|
|
|
|
def get_queryset(self):
|
|
self.queryset = super(AssetGroupListView, self).get_queryset()
|
|
self.keyword = keyword = self.request.GET.get('keyword', '')
|
|
self.sort = sort = self.request.GET.get('sort', '-date_created')
|
|
if keyword:
|
|
self.queryset = self.queryset.filter(Q(name__icontains=keyword) |
|
|
Q(comment__icontains=keyword))
|
|
if sort:
|
|
self.queryset = self.queryset.order_by(sort)
|
|
return self.queryset
|
|
|
|
|
|
class AssetGroupDetailView(SingleObjectMixin, AdminUserRequiredMixin, ListView):
|
|
template_name = 'assets/asset_group_detail.html'
|
|
paginate_by = settings.CONFIG.DISPLAY_PER_PAGE
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
self.object = self.get_object(queryset=AssetGroup.objects.all())
|
|
return super(AssetGroupDetailView, self).get(request, *args, **kwargs)
|
|
|
|
def get_queryset(self):
|
|
return self.object.assets.all()
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = {
|
|
'app': _('Assets'),
|
|
'action': _('Asset group detail'),
|
|
'asset_group': self.object,
|
|
}
|
|
kwargs.update(context)
|
|
return super(AssetGroupDetailView, self).get_context_data(**kwargs)
|
|
|
|
class AssetGroupUpdateView(AdminUserRequiredMixin, UpdateView):
|
|
model = AssetGroup
|
|
form_class = AssetGroupForm
|
|
template_name = 'assets/asset_group_create.html'
|
|
success_url = reverse_lazy('assets:asset-group-list')
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
self.object = self.get_object(queryset=AssetGroup.objects.all())
|
|
return super(AssetGroupUpdateView, self).get(request, *args, **kwargs)
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = {
|
|
'app': _('Assets'),
|
|
'action': _('Create asset group'),
|
|
# 'assets': Asset.objects.all(),
|
|
'assets_count': self.object.assets.all().count(),
|
|
'group_id':self.object.id,
|
|
}
|
|
kwargs.update(context)
|
|
return super(AssetGroupUpdateView, self).get_context_data(**kwargs)
|
|
|
|
|
|
class AssetGroupDeleteView(AdminUserRequiredMixin, DeleteView):
|
|
template_name = 'assets/delete_confirm.html'
|
|
model = AssetGroup
|
|
success_url = reverse_lazy('assets:asset-group-list')
|
|
|
|
|
|
class IDCListView(AdminUserRequiredMixin, ListView):
|
|
model = IDC
|
|
paginate_by = settings.CONFIG.DISPLAY_PER_PAGE
|
|
context_object_name = 'idc_list'
|
|
template_name = 'assets/idc_list.html'
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = {
|
|
'app': _('Assets'),
|
|
'action': _('IDC list'),
|
|
'keyword': self.request.GET.get('keyword', '')
|
|
}
|
|
kwargs.update(context)
|
|
return super(IDCListView, self).get_context_data(**kwargs)
|
|
|
|
def get_queryset(self):
|
|
self.queryset = super(IDCListView, self).get_queryset()
|
|
self.keyword = keyword = self.request.GET.get('keyword', '')
|
|
self.sort = sort = self.request.GET.get('sort', '-date_created')
|
|
|
|
if keyword:
|
|
self.queryset = self.queryset.filter(Q(name__icontains=keyword) |
|
|
Q(comment__icontains=keyword))
|
|
|
|
if sort:
|
|
self.queryset = self.queryset.order_by(sort)
|
|
return self.queryset
|
|
|
|
|
|
class IDCCreateView(AdminUserRequiredMixin, CreateView):
|
|
model = IDC
|
|
form_class = IDCForm
|
|
template_name = 'assets/idc_create_update.html'
|
|
success_url = reverse_lazy('assets:idc-list')
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = {
|
|
'app': _('assets'),
|
|
'action': _('Create IDC'),
|
|
}
|
|
kwargs.update(context)
|
|
return super(IDCCreateView, self).get_context_data(**kwargs)
|
|
|
|
def form_valid(self, form):
|
|
IDC = form.save(commit=False)
|
|
IDC.created_by = self.request.user.username or 'System'
|
|
IDC.save()
|
|
# IDC_add_success_next(user)
|
|
return super(IDCCreateView, self).form_valid(form)
|
|
|
|
class IDCUpdateView(AdminUserRequiredMixin, UpdateView):
|
|
model = IDC
|
|
form_class = IDCForm
|
|
template_name = 'assets/idc_create_update.html'
|
|
context_object_name = 'idc'
|
|
success_url = reverse_lazy('assets:idc-list')
|
|
|
|
def form_valid(self, form):
|
|
idc = form.save(commit=False)
|
|
idc.save()
|
|
return super(IDCUpdateView, self).form_valid(form)
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = {
|
|
'app': _('assets'),
|
|
'action': _('Update IDC'),
|
|
}
|
|
kwargs.update(context)
|
|
return super(IDCUpdateView, self).get_context_data(**kwargs)
|
|
|
|
|
|
class IDCDetailView(AdminUserRequiredMixin, DetailView):
|
|
pass
|
|
|
|
|
|
class IDCDeleteView(AdminUserRequiredMixin, DeleteView):
|
|
model = IDC
|
|
template_name = 'assets/delete_confirm.html'
|
|
success_url = reverse_lazy('assets:idc-list')
|
|
|
|
|
|
|
|
class AdminUserListView(AdminUserRequiredMixin, ListView):
|
|
model = AdminUser
|
|
paginate_by = settings.CONFIG.DISPLAY_PER_PAGE
|
|
context_object_name = 'admin_user_list'
|
|
template_name = 'assets/admin_user_list.html'
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = {
|
|
'app': _('Assets'),
|
|
'action': _('Admin user list'),
|
|
'keyword': self.request.GET.get('keyword', '')
|
|
}
|
|
kwargs.update(context)
|
|
return super(AdminUserListView, self).get_context_data(**kwargs)
|
|
|
|
def get_queryset(self):
|
|
# Todo: Default order by lose asset connection num
|
|
self.queryset = super(AdminUserListView, self).get_queryset()
|
|
self.keyword = keyword = self.request.GET.get('keyword', '')
|
|
self.sort = sort = self.request.GET.get('sort', '-date_created')
|
|
|
|
if keyword:
|
|
self.queryset = self.queryset.filter(Q(name__icontains=keyword) |
|
|
Q(comment__icontains=keyword))
|
|
|
|
if sort:
|
|
self.queryset = self.queryset.order_by(sort)
|
|
return self.queryset
|
|
|
|
|
|
class AdminUserCreateView(AdminUserRequiredMixin, SuccessMessageMixin, CreateView):
|
|
model = AdminUser
|
|
form_class = AdminUserForm
|
|
template_name = 'assets/admin_user_create_update.html'
|
|
success_url = reverse_lazy('assets:admin-user-list')
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = {
|
|
'app': 'assets',
|
|
'action': 'Create admin user'
|
|
}
|
|
kwargs.update(context)
|
|
return super(AdminUserCreateView, self).get_context_data(**kwargs)
|
|
|
|
def get_success_message(self, cleaned_data):
|
|
success_message = _('Create admin user <a href="%s">%s</a> successfully.' %
|
|
(
|
|
reverse_lazy('assets:admin-user-detail', kwargs={'pk': self.object.pk}),
|
|
self.object.name,
|
|
))
|
|
return success_message
|
|
|
|
|
|
class AdminUserUpdateView(AdminUserRequiredMixin, UpdateView):
|
|
model = AdminUser
|
|
form_class = AdminUserForm
|
|
template_name = 'assets/admin_user_create_update.html'
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = {
|
|
'app': 'assets',
|
|
'action': 'Update admin user'
|
|
}
|
|
kwargs.update(context)
|
|
return super(AdminUserUpdateView, self).get_context_data(**kwargs)
|
|
|
|
def get_success_url(self):
|
|
success_url = reverse_lazy('assets:admin-user-detail', pk=self.object.pk)
|
|
return success_url
|
|
|
|
|
|
class AdminUserDetailView(AdminUserRequiredMixin, SingleObjectMixin, ListView):
|
|
paginate_by = settings.CONFIG.DISPLAY_PER_PAGE
|
|
template_name = 'assets/admin_user_detail.html'
|
|
context_object_name = 'admin_user'
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
self.object = self.get_object(queryset=AdminUser.objects.all())
|
|
return super(AdminUserDetailView, self).get(request, *args, **kwargs)
|
|
|
|
# Todo: queryset default order by connectivity, need ops support
|
|
def get_queryset(self):
|
|
return self.object.assets.all()
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = {
|
|
'app': 'assets',
|
|
'action': 'Admin user detail'
|
|
}
|
|
kwargs.update(context)
|
|
return super(AdminUserDetailView, self).get_context_data(**kwargs)
|
|
|
|
|
|
class AdminUserDeleteView(AdminUserRequiredMixin, DeleteView):
|
|
model = AdminUser
|
|
template_name = 'assets/delete_confirm.html'
|
|
success_url = reverse_lazy('assets:admin-user-list')
|
|
|
|
|
|
class SystemUserListView(AdminUserRequiredMixin, ListView):
|
|
model = SystemUser
|
|
paginate_by = settings.CONFIG.DISPLAY_PER_PAGE
|
|
context_object_name = 'system_user_list'
|
|
template_name = 'assets/system_user_list.html'
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = {
|
|
'app': _('Assets'),
|
|
'action': _('System user list'),
|
|
'keyword': self.request.GET.get('keyword', '')
|
|
}
|
|
kwargs.update(context)
|
|
return super(SystemUserListView, self).get_context_data(**kwargs)
|
|
|
|
def get_queryset(self):
|
|
# Todo: Default order by lose asset connection num
|
|
self.queryset = super(SystemUserListView, self).get_queryset()
|
|
self.keyword = keyword = self.request.GET.get('keyword', '')
|
|
self.sort = sort = self.request.GET.get('sort', '-date_created')
|
|
|
|
if keyword:
|
|
self.queryset = self.queryset.filter(Q(name__icontains=keyword) |
|
|
Q(comment__icontains=keyword))
|
|
|
|
if sort:
|
|
self.queryset = self.queryset.order_by(sort)
|
|
return self.queryset
|
|
|
|
|
|
class SystemUserCreateView(AdminUserRequiredMixin, SuccessMessageMixin, CreateView):
|
|
model = SystemUser
|
|
form_class = SystemUserForm
|
|
template_name = 'assets/system_user_create_update.html'
|
|
success_url = reverse_lazy('assets:system-user-list')
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = {
|
|
'app': _('Assets'),
|
|
'action': _('Create system user'),
|
|
}
|
|
kwargs.update(context)
|
|
return super(SystemUserCreateView, self).get_context_data(**kwargs)
|
|
|
|
def get_success_message(self, cleaned_data):
|
|
success_message = _('Create system user <a href="%s">%s</a> successfully.' %
|
|
(
|
|
reverse_lazy('assets:system-user-detail', kwargs={'pk': self.object.pk}),
|
|
self.object.name,
|
|
))
|
|
|
|
return success_message
|
|
|
|
|
|
class SystemUserUpdateView(AdminUserRequiredMixin, UpdateView):
|
|
model = SystemUser
|
|
form_class = SystemUserForm
|
|
template_name = 'assets/system_user_create_update.html'
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = {
|
|
'app': _('Assets'),
|
|
'action': _('Update system user')
|
|
}
|
|
kwargs.update(context)
|
|
return super(SystemUserUpdateView, self).get_context_data(**kwargs)
|
|
|
|
def get_success_url(self):
|
|
success_url = reverse_lazy('assets:system-user-detail', kwargs={'pk': self.object.pk})
|
|
return success_url
|
|
|
|
|
|
class SystemUserDetailView(AdminUserRequiredMixin, DetailView):
|
|
template_name = 'assets/system_user_detail.html'
|
|
context_object_name = 'system_user'
|
|
model = SystemUser
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = {
|
|
'app': _('Assets'),
|
|
'action': _('System user detail')
|
|
}
|
|
kwargs.update(context)
|
|
return super(SystemUserDetailView, self).get_context_data(**kwargs)
|
|
|
|
|
|
class SystemUserDeleteView(AdminUserRequiredMixin, DeleteView):
|
|
model = SystemUser
|
|
template_name = 'assets/delete_confirm.html'
|
|
success_url = reverse_lazy('assets:system-user-list')
|
|
|
|
|
|
class SystemUserAssetView(AdminUserRequiredMixin, SingleObjectMixin, ListView):
|
|
paginate_by = settings.CONFIG.DISPLAY_PER_PAGE
|
|
template_name = 'assets/system_user_asset.html'
|
|
context_object_name = 'system_user'
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
self.object = self.get_object(queryset=SystemUser.objects.all())
|
|
return super(SystemUserAssetView, self).get(request, *args, **kwargs)
|
|
|
|
def get_asset_groups(self):
|
|
return self.object.asset_groups.all()
|
|
|
|
# Todo: queryset default order by connectivity, need ops support
|
|
def get_queryset(self):
|
|
return list(self.object.get_assets())
|
|
|
|
def get_context_data(self, **kwargs):
|
|
asset_groups = self.get_asset_groups()
|
|
assets = self.get_queryset()
|
|
context = {
|
|
'app': 'assets',
|
|
'action': 'System user asset',
|
|
'assets_remain': [asset for asset in Asset.objects.all() if asset not in assets],
|
|
'asset_groups': asset_groups,
|
|
'asset_groups_remain': [asset_group for asset_group in AssetGroup.objects.all()
|
|
if asset_group not in asset_groups]
|
|
}
|
|
kwargs.update(context)
|
|
return super(SystemUserAssetView, self).get_context_data(**kwargs)
|
|
|
|
|
|
# class SystemUserAssetGroupView(AdminUserRequiredMixin, SingleObjectMixin, ListView):
|
|
# paginate_by = settings.CONFIG.DISPLAY_PER_PAGE
|
|
# template_name = 'assets/system_user_asset_group.html'
|
|
# context_object_name = 'system_user'
|
|
#
|
|
# def get(self, request, *args, **kwargs):
|
|
# self.object = self.get_object(queryset=SystemUser.objects.all())
|
|
# return super(SystemUserAssetGroupView, self).get(request, *args, **kwargs)
|
|
#
|
|
# Todo: queryset default order by connectivity, need ops support
|
|
# def get_queryset(self):
|
|
# return self.object.asset_groups.all()
|
|
#
|
|
# def get_context_data(self, **kwargs):
|
|
# context = {
|
|
# 'app': 'assets',
|
|
# 'action': 'System user asset group',
|
|
# 'asset_groups': self.get_queryset(),
|
|
# }
|
|
# kwargs.update(context)
|
|
# return super(SystemUserAssetGroupView, self).get_context_data(**kwargs)
|
|
|
|
|
|
|