def reoutbound(request, queryset): action = sys._getframe().f_code.co_name action_name = "取消出库" queryset = queryset.filter(actived=False) if not queryset.exists(): return "查无结果" if request.POST.get('post'): for obj in queryset: o = copy.deepcopy(obj) obj.actived = True obj.save() diffs = diff_dict(model_to_dict(o), model_to_dict(obj)) log_action( user_id=request.user.pk, content_type_id=get_content_type_for_model(obj, True).pk, object_id=obj.pk, action_flag=action_name, message=json.dumps(list(diffs.keys())), content=json.dumps(diffs) ) return None context = construct_context(request, queryset, action, action_name) return TemplateResponse(request, 'base/base_confirmation.html', context)
def stopfuzzing(request, queryset): action = sys._getframe().f_code.co_name action_name = "停止" if request.POST.get('post'): for obj in queryset: o = copy.deepcopy(obj) try: #os.kill(int(obj.runningpid), signal.SIGKILL) obj.actived = False kill_all_process(int(obj.runningpid)) obj.runningpid = "" except Exception as e: print(e) obj.save() diffs = diff_dict(model_to_dict(o), model_to_dict(obj)) log_action( user_id=request.user.pk, content_type_id=get_content_type_for_model(obj, True).pk, object_id=obj.pk, action_flag="停止", message=json.dumps(list(diffs.keys())), content=json.dumps(diffs) ) return None context = construct_context(request, queryset, action, action_name) return TemplateResponse(request, 'base/base_confirmation.html', context)
def form_valid(self, form): form.instance.operator = self.request.user if 'onidc' not in form.cleaned_data: form.instance.onidc = self.request.user.onidc d1 = form.initial message = json.dumps(form.changed_data) response = super(EditModelView, self).form_valid(form) d2 = model_to_dict(construct_instance(form, self.object)) diffs = diff_dict(make_dict(d1), make_dict(d2)) content = json.dumps(diffs) log_action(user_id=self.request.user.pk, content_type_id=get_content_type_for_model( self.object, True).pk, object_id=self.object.pk, action_flag="修改", message=message, content=content) # if self.model_name == 'online': # verify = Thread(target=device_post_save, args=(self.object.pk,)) # verify.start() if self.request.is_ajax(): data = { 'message': "Successfully submitted form data.", 'data': form.cleaned_data } return JsonResponse(data) else: return response
def delete(request, queryset): model = queryset.model opts = model._meta action = sys._getframe().f_code.co_name action_name = "删除" modeladmin = admin.site._registry.get(model) # queryset = queryset.filter(actived=False) if not modeladmin.has_delete_permission(request): raise PermissionDenied using = router.db_for_write(modeladmin.model) deletable_objects, model_count, perms_needed, protected = get_deleted_objects( queryset, request, modeladmin.admin_site) if request.POST.get('post') and not protected: if perms_needed: raise PermissionDenied if queryset.count(): for obj in queryset: log_action( user_id=request.user.pk, content_type_id=get_content_type_for_model(obj, True).pk, object_id=obj.pk, action_flag="删除" ) if not SOFT_DELELE: queryset.delete() else: queryset.update(deleted=True, actived=False) return None if len(queryset) == 1: objects_name = force_text(opts.verbose_name) else: objects_name = force_text(opts.verbose_name_plural) meta, menus = construct_model_meta(request, model, action_name) context = dict( objects_name=objects_name, deletable_objects=[deletable_objects], model_count=dict(model_count).items(), queryset=queryset, perms_lacking=perms_needed, protected=protected, opts=opts, meta=meta, action=action, action_name=action_name, menus=menus, ) request.current_app = modeladmin.admin_site.name return TemplateResponse(request, 'base/base_confirmation.html', context)
def get_user_config(user, mark, model): content_type = get_content_type_for_model(model) configs = Configure.objects.filter( creator=user, mark=mark, content_type=content_type).order_by('-pk') if configs.exists(): config = configs.first().content try: return json.loads(config) except BaseException: return None else: return None
def user_config(self, data): newd = dict(list_display=data.getlist('list_display'), list_only_date=data.get('list_only_date', 1)) content = json.dumps(newd) config = Configure.objects.filter( content_type=get_content_type_for_model(self.model), onidc_id=self.onidc_id, creator=self.request.user, mark='list').order_by('-pk') if config.exists(): config = config.update(content=content) else: config = Configure.objects.create( content_type=get_content_type_for_model(self.model), onidc_id=self.onidc_id, creator=self.request.user, mark='list', content=content) key = utils.make_template_fragment_key("{}.{}.{}".format( self.request.user.id, self.model_name, 'list')) cache.delete(key) return config
def startfuzzing(request, queryset): action = sys._getframe().f_code.co_name action_name = "开始" if not os.path.exists(settings.WORKING_ROOT): try: os.mkdir(settings.WORKING_ROOT) os.mkdir(os.path.join(settings.WORKING_ROOT, "fuzzer_in")) os.mkdir(os.path.join(settings.WORKING_ROOT, "fuzzer_out")) except: pass if request.POST.get('post'): for obj in queryset: o = copy.deepcopy(obj) obj.actived = True b_path = os.path.join(settings.BASE_DIR, obj.binary[1:]) os.chmod(b_path,stat.S_IXUSR |stat.S_IRUSR |stat.S_IWUSR | stat.S_IWGRP | stat.S_IRGRP | stat.S_IXOTH | stat.S_IROTH | stat.S_IWOTH) cmd = [sys.executable, os.path.join(settings.BASE_DIR, "workingFuzzer.py"), os.path.join(settings.WORKING_ROOT, "fuzzer_in"), os.path.join(settings.WORKING_ROOT, "fuzzer_out", obj.name)] cmd += obj.cmdargs.split(" ") cmd += [b_path, "--", obj.targetargs] if not os.path.exists(os.path.join(settings.WORKING_ROOT, "fuzzer_out", obj.name)): try: os.mkdir(os.path.join(settings.WORKING_ROOT, "fuzzer_out", obj.name)) except: pass #file_output = open(os.path.join(settings.WORKING_ROOT, "fuzzer_out", obj.name, "ui.log"), "w") file_output = SpooledFile(os.path.join(settings.WORKING_ROOT, "fuzzer_out", obj.name, "ui.log"), mode="w", max_size=100*1024) p = run_sh(cmd,stdin=subprocess.PIPE, stdout=file_output, stderr=subprocess.PIPE, cwd=settings.BASE_DIR) obj.runningpid = p.pid obj.save() diffs = diff_dict(model_to_dict(o), model_to_dict(obj)) log_action( user_id=request.user.pk, content_type_id=get_content_type_for_model(obj, True).pk, object_id=obj.pk, action_flag="开始", message=json.dumps(list(diffs.keys())), content=json.dumps(diffs) ) return None context = construct_context(request, queryset, action, action_name) return TemplateResponse(request, 'base/base_confirmation.html', context)
def form_valid(self, form): form.instance.creator = self.request.user if 'onidc' not in form.cleaned_data: form.instance.onidc = self.request.user.onidc response = super(NewModelView, self).form_valid(form) log_action(user_id=self.request.user.pk, content_type_id=get_content_type_for_model( self.object, True).pk, object_id=self.object.pk, action_flag="新增") # if self.model_name == 'online': # verify = Thread(target=device_post_save, args=(self.object.pk,)) # verify.start() if self.request.is_ajax(): data = { 'message': "Successfully submitted form data.", 'data': form.cleaned_data } return JsonResponse(data) else: return response
def outbound(request, queryset): action = sys._getframe().f_code.co_name action_name = "出库" queryset = queryset.filter(actived=True) if not queryset.exists(): return "选择无结果" total = queryset.aggregate(Sum('amount')) if request.POST.get('post') and request.POST.getlist('items'): def construct_item(index): obj = queryset.get(pk=int(index)) out_amount = int(request.POST.get('count-' + str(index))) out_serials = request.POST.getlist('sn-' + str(index)) copy_needed = True if int(out_amount) == obj.amount: copy_needed = False comment = request.POST.get(('comment-' + index), None) return obj, copy_needed, out_serials, out_amount, comment for item in request.POST.getlist('items'): obj, _copy, out_serials, out_amount, comment = construct_item(item) o = copy.deepcopy(obj) if _copy: hold = [s for s in obj.serials.split( ',') if s not in out_serials] obj.amount -= out_amount obj.serials = ','.join(hold) new_obj = copy.deepcopy(obj) new_obj.pk = None new_obj.amount = out_amount new_obj.serials = ','.join(out_serials) new_obj.actived = False new_obj.creator = request.user new_obj.created = timezone.datetime.now() new_obj.operator = None new_obj.parent = obj new_obj.save() comment_obj = new_obj else: obj.actived = False obj.operator = request.user comment_obj = obj obj.save() if comment: Comment.objects.create( object_repr=comment_obj, content=comment, creator=request.user, onidc=obj.onidc) diffs = diff_dict(model_to_dict(o), model_to_dict(obj)) log_action( user_id=request.user.pk, content_type_id=get_content_type_for_model(obj, True).pk, object_id=comment_obj.pk, action_flag=action_name, message=json.dumps(list(diffs.keys())), content=json.dumps(diffs) ) return None context = construct_context(request, queryset, action, action_name) _extra = dict(total=total) context.update(_extra) return TemplateResponse(request, 'base/items_out.html', context)