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)