def get(self, request, *args, **kwargs):
self.preference_file_form = PreferenceFileForm(
prefix='pff',
initial=PreferenceFileForm.get_initial(self.preference_file))
self.key_form_set = KeyFormSet(prefix='kfs',
initial=KeyFormSet.get_initial(
self.preference_file))
return super().get(request, *args, **kwargs)
def post(self, request, *args, **kwargs):
self.preference_file_form = PreferenceFileForm(request.POST,
prefix='pff')
self.key_form_set = KeyFormSet(request.POST, prefix='kfs')
if self.preference_file_form.is_valid() and self.key_form_set.is_valid(
):
return self.forms_valid()
else:
return self.render_to_response(self.get_context_data())
def post(self, request, *args, **kwargs):
self.preference_file_form = PreferenceFileForm(request.POST, prefix='pff')
self.key_form_set = KeyFormSet(request.POST, prefix='kfs')
if self.preference_file_form.is_valid() and self.key_form_set.is_valid():
return self.forms_valid()
else:
return self.render_to_response(self.get_context_data())
class UpdateComplianceProbePreferenceFileView(LoginRequiredMixin,
TemplateView):
template_name = "osquery/preference_file_form.html"
def dispatch(self, request, *args, **kwargs):
self.probe_source = get_object_or_404(ProbeSource,
pk=kwargs["probe_id"])
self.probe = self.probe_source.load()
self.preference_file_id = int(kwargs["pf_id"])
try:
self.preference_file = self.probe.preference_files[
self.preference_file_id]
except IndexError:
raise Http404
return super().dispatch(request, *args, **kwargs)
def get_context_data(self, **kwargs):
ctx = super().get_context_data(**kwargs)
ctx['title'] = 'Update compliance probe preference file'
ctx["probes"] = True
ctx['probe_source'] = self.probe_source
ctx['probe'] = self.probe
ctx['cancel_url'] = self.probe_source.get_absolute_url(
"osquery_compliance")
ctx['preference_file_form'] = self.preference_file_form
ctx['key_form_set'] = self.key_form_set
return ctx
def forms_valid(self):
preference_file = self.preference_file_form.get_item_d()
preference_file['keys'] = self.key_form_set.get_keys()
def func(probe_d):
probe_d["preference_files"][
self.preference_file_id] = preference_file
self.probe_source.update_body(func)
return HttpResponseRedirect(
self.probe_source.get_absolute_url("preference_files"))
def get(self, request, *args, **kwargs):
self.preference_file_form = PreferenceFileForm(
prefix='pff',
initial=PreferenceFileForm.get_initial(self.preference_file))
self.key_form_set = KeyFormSet(prefix='kfs',
initial=KeyFormSet.get_initial(
self.preference_file))
return super().get(request, *args, **kwargs)
def post(self, request, *args, **kwargs):
self.preference_file_form = PreferenceFileForm(request.POST,
prefix='pff')
self.key_form_set = KeyFormSet(request.POST, prefix='kfs')
if self.preference_file_form.is_valid() and self.key_form_set.is_valid(
):
return self.forms_valid()
else:
return self.render_to_response(self.get_context_data())
class UpdateComplianceProbePreferenceFileView(TemplateView):
template_name = "osquery/preference_file_form.html"
def dispatch(self, request, *args, **kwargs):
self.probe_source = get_object_or_404(ProbeSource, pk=kwargs["probe_id"])
self.probe = self.probe_source.load()
self.preference_file_id = int(kwargs["pf_id"])
try:
self.preference_file = self.probe.preference_files[self.preference_file_id]
except IndexError:
raise Http404
return super().dispatch(request, *args, **kwargs)
def get_context_data(self, **kwargs):
ctx = super().get_context_data(**kwargs)
ctx['title'] = 'Update compliance probe preference file'
ctx["probes"] = True
ctx['probe_source'] = self.probe_source
ctx['probe'] = self.probe
ctx['cancel_url'] = self.probe_source.get_absolute_url("osquery_compliance")
ctx['preference_file_form'] = self.preference_file_form
ctx['key_form_set'] = self.key_form_set
return ctx
def forms_valid(self):
preference_file = self.preference_file_form.get_item_d()
preference_file['keys'] = self.key_form_set.get_keys()
def func(probe_d):
probe_d["preference_files"][self.preference_file_id] = preference_file
self.probe_source.update_body(func)
return HttpResponseRedirect(self.probe_source.get_absolute_url("preference_files"))
def get(self, request, *args, **kwargs):
self.preference_file_form = PreferenceFileForm(prefix='pff',
initial=PreferenceFileForm.get_initial(self.preference_file))
self.key_form_set = KeyFormSet(prefix='kfs',
initial=KeyFormSet.get_initial(self.preference_file))
return super().get(request, *args, **kwargs)
def post(self, request, *args, **kwargs):
self.preference_file_form = PreferenceFileForm(request.POST, prefix='pff')
self.key_form_set = KeyFormSet(request.POST, prefix='kfs')
if self.preference_file_form.is_valid() and self.key_form_set.is_valid():
return self.forms_valid()
else:
return self.render_to_response(self.get_context_data())
class AddComplianceProbePreferenceFileView(LoginRequiredMixin, TemplateView):
template_name = "osquery/preference_file_form.html"
def dispatch(self, request, *args, **kwargs):
self.probe_source = get_object_or_404(ProbeSource, pk=kwargs["probe_id"])
self.probe = self.probe_source.load()
return super().dispatch(request, *args, **kwargs)
def get_context_data(self, **kwargs):
ctx = super().get_context_data(**kwargs)
ctx['title'] = 'Add compliance probe preference file'
ctx["probes"] = True
ctx['probe_source'] = self.probe_source
ctx['probe'] = self.probe
ctx['cancel_url'] = self.probe_source.get_absolute_url("osquery_compliance")
ctx['preference_file_form'] = self.preference_file_form
ctx['key_form_set'] = self.key_form_set
return ctx
def forms_valid(self):
preference_file = self.preference_file_form.get_item_d()
preference_file['keys'] = self.key_form_set.get_keys()
def func(probe_d):
preference_files = probe_d.setdefault("preference_files", [])
preference_files.append(preference_file)
self.probe_source.update_body(func)
return HttpResponseRedirect(self.probe_source.get_absolute_url("preference_files"))
def get(self, request, *args, **kwargs):
self.preference_file_form = PreferenceFileForm(prefix='pff')
self.key_form_set = KeyFormSet(prefix='kfs')
return super().get(request, *args, **kwargs)
def post(self, request, *args, **kwargs):
self.preference_file_form = PreferenceFileForm(request.POST, prefix='pff')
self.key_form_set = KeyFormSet(request.POST, prefix='kfs')
if self.preference_file_form.is_valid() and self.key_form_set.is_valid():
return self.forms_valid()
else:
return self.render_to_response(self.get_context_data())
def get(self, request, *args, **kwargs):
self.preference_file_form = PreferenceFileForm(prefix='pff')
self.key_form_set = KeyFormSet(prefix='kfs')
return super().get(request, *args, **kwargs)
