class AssetAdd(LoginRequiredMixin, CreateView):
"""
资产增加
"""
model = AssetInfo
form_class = AssetForm
template_name = 'asset/asset-add-update.html'
success_url = reverse_lazy('asset:asset_list')
@method_decorator(permission_required_or_404('asset.add_assetinfo'))
def dispatch(self, *args, **kwargs):
return super().dispatch(*args, **kwargs)
def get_context_data(self, **kwargs):
context = {
"asset_active": "active",
"asset_list_active": "active",
}
if '__next__' in self.request.POST:
context['i__next__'] = self.request.POST['__next__']
else:
try:
context['i__next__'] = self.request.META['HTTP_REFERER']
except Exception as e:
logger.error(e)
kwargs.update(context)
return super().get_context_data(**kwargs)
class AssetBusinessAllDel(LoginRequiredMixin, View):
"""
资产业务 删除
"""
model = AssetBusiness
@method_decorator(permission_required_or_404('asset.delete_assetbusiness'))
def dispatch(self, *args, **kwargs):
return super().dispatch(*args, **kwargs)
@staticmethod
def post(request):
ret = {'status': True, 'error': None, }
try:
if request.POST.get('nid'):
ids = request.POST.get('nid', None)
AssetBusiness.objects.get(id=ids).delete()
else:
ids = request.POST.getlist('id', None)
idstring = ','.join(ids)
AssetBusiness.objects.extra(where=['id IN (' + idstring + ')']).delete()
except Exception as e:
ret['status'] = False
ret['error'] = '删除请求错误,没有权限{}'.format(e)
finally:
return HttpResponse(json.dumps(ret))
class NameUpdate(LoginRequiredMixin, UpdateView):
"""
系统用户更新"""
model = Names
form_class = NameForm
template_name = 'name/name-add-update.html'
success_url = reverse_lazy('name:name_list')
@method_decorator(permission_required_or_404('name.change_names'))
def dispatch(self, *args, **kwargs):
return super().dispatch(*args, **kwargs)
def get_context_data(self, **kwargs):
context = {
"name_active": "active",
"name_list_active": "active",
}
kwargs.update(context)
return super().get_context_data(**kwargs)
def form_valid(self, form):
pk = self.kwargs.get(self.pk_url_kwarg, None)
obj = Names.objects.get(id=pk)
old_password = obj.password
new_password = form.cleaned_data['password']
forms = form.save()
if new_password == "1":
forms.password = old_password
else:
forms.set_password(new_password)
forms.save()
return super().form_valid(form)
class NameAdd(LoginRequiredMixin, CreateView):
"""
系统用户 增加
"""
model = Names
form_class = NameForm
template_name = 'name/name-add-update.html'
success_url = reverse_lazy('name:name_list')
@method_decorator(permission_required_or_404('name.add_names'))
def dispatch(self, *args, **kwargs):
return super().dispatch(*args, **kwargs)
def get_context_data(self, **kwargs):
context = {
"name_active": "active",
"name_list_active": "active",
}
kwargs.update(context)
return super().get_context_data(**kwargs)
def form_valid(self, form):
forms = form.save()
password_form = form.cleaned_data['password']
if password_form is not None:
forms.set_password(password_form)
forms.save()
return super().form_valid(form)
class GroupsObjectAdd(LoginRequiredMixin, CreateView):
"""
系统组对象 增加
"""
model = GroupObjectPermission
form_class = GroupsObjectForm
template_name = 'name/groups-object-add-update.html'
success_url = reverse_lazy('name:groups_object_list')
@method_decorator(
permission_required_or_404('guardian.add_groupobjectpermission'))
def dispatch(self, *args, **kwargs):
return super().dispatch(*args, **kwargs)
def get_context_data(self, **kwargs):
context = {
"name_active": "active",
"groups_object_list_active": "active",
}
kwargs.update(context)
return super().get_context_data(**kwargs)
def form_invalid(self, form):
print(form.errors)
return super().form_invalid(form)
class AssetProjectAdd(LoginRequiredMixin, CreateView):
"""
资产项目增加
"""
model = AssetProject
form_class = AssetProjectForm
template_name = 'asset/asset-project-add-update.html'
success_url = reverse_lazy('asset:asset_project_list')
@method_decorator(permission_required_or_404('asset.add_assetproject'))
def dispatch(self, *args, **kwargs):
return super().dispatch(*args, **kwargs)
def get_context_data(self, **kwargs):
context = {
"asset_active": "active",
"asset_project_list_active": "active",
}
kwargs.update(context)
return super().get_context_data(**kwargs)
class GroupsAdd(LoginRequiredMixin, CreateView):
"""
系统组 增加
"""
model = Groups
form_class = GroupsForm
template_name = 'name/groups-add-update.html'
success_url = reverse_lazy('name:groups_list')
@method_decorator(permission_required_or_404('name.add_groups'))
def dispatch(self, *args, **kwargs):
return super().dispatch(*args, **kwargs)
def get_context_data(self, **kwargs):
context = {
"name_active": "active",
"groups_list_active": "active",
}
kwargs.update(context)
return super().get_context_data(**kwargs)
class AssetBusinessUpdate(LoginRequiredMixin, UpdateView):
"""
资产业务 更新
"""
model = AssetBusiness
form_class = AssetBusinessForm
template_name = 'asset/asset-business-add-update.html'
success_url = reverse_lazy('asset:asset_business_list')
@method_decorator(permission_required_or_404('asset.change_assetbusiness'))
def dispatch(self, *args, **kwargs):
return super().dispatch(*args, **kwargs)
def get_context_data(self, **kwargs):
context = {
"asset_active": "active",
"asset_business_list_active": "active",
}
kwargs.update(context)
return super().get_context_data(**kwargs)
class AssetUserAdd(LoginRequiredMixin, CreateView):
"""
资产用户 增加
"""
model = AssetLoginUser
form_class = AssetUserForm
template_name = 'asset/asset-user-add-update.html'
success_url = reverse_lazy('asset:asset_user_list')
@method_decorator(permission_required_or_404('asset.add_assetloginuser'))
def dispatch(self, *args, **kwargs):
return super().dispatch(*args, **kwargs)
def get_context_data(self, **kwargs):
context = {
"asset_active": "active",
"asset_user_list_active": "active",
}
kwargs.update(context)
return super().get_context_data(**kwargs)
def form_valid(self, form):
forms = form.save()
password_form = form.cleaned_data['password']
private_key = form.cleaned_data['private_key']
if password_form is not None:
password = encrypt_p(password_form)
forms.password = password
forms.save()
if private_key is not None:
try:
name = form.cleaned_data['hostname']
private_key_path = AssetLoginUser.objects.get(
hostname=name).private_key.name
system("chmod 600 {0}".format(private_key_path))
except Exception as e:
logger.error(e)
return super().form_valid(form)
class GroupListAll(LoginRequiredMixin, ListView):
"""
系统组 列表
"""
template_name = 'name/groups.html'
model = Groups
context_object_name = "groups_list"
queryset = Groups.objects.all()
ordering = ('-id', )
@method_decorator(permission_required_or_404('name.add_groups'))
def dispatch(self, *args, **kwargs):
return super().dispatch(*args, **kwargs)
def get_context_data(self, **kwargs):
context = {
"name_active": "active",
"groups_list_active": "active",
}
kwargs.update(context)
return super().get_context_data(**kwargs)
