def edit(request, plugin_id): (mylar_key, mylar_secret) = utils.get_mylar_oauth_creds() url = utils.get_rpc_url(request) trans = OAuthTransport(url, key=mylar_key, secret=mylar_secret) """ Get the Mylar object If it does not exist create a new entry """ try: mylar = models.Mylar.objects.order_by('-id')[0] except IndexError: mylar = models.Mylar.objects.create() try: server = jsonrpclib.Server(url, transport=trans) jail_path = server.plugins.jail.path(plugin_id) jail = json.loads(server.plugins.jail.info(plugin_id))[0]['fields'] jail_ipv4 = jail['jail_ipv4'].split('/')[0] auth = server.plugins.is_authenticated( request.COOKIES.get("sessionid", "")) assert auth except Exception as e: raise if request.method == "GET": form = forms.MylarForm(instance=mylar, jail_path=jail_path) return render(request, "edit.html", {'form': form, 'ipv4': jail_ipv4}) if not request.POST: return JsonResponse(request, error=True, message="A problem occurred.") form = forms.MylarForm(request.POST, instance=mylar, jail_path=jail_path) if form.is_valid(): form.save() cmd = "%s restart" % utils.mylar_control pipe = Popen(cmd, stdin=PIPE, stdout=PIPE, stderr=PIPE, shell=True, close_fds=True) return JsonResponse(request, error=True, message="Mylar settings successfully saved.") return JsonResponse(request, form=form)
def start(request, plugin_id): (mylar_key, mylar_secret) = utils.get_mylar_oauth_creds() url = utils.get_rpc_url(request) trans = OAuthTransport(url, key=mylar_key, secret=mylar_secret) server = jsonrpclib.Server(url, transport=trans) auth = server.plugins.is_authenticated(request.COOKIES.get( "sessionid", "")) jail_path = server.plugins.jail.path(plugin_id) assert auth try: mylar = models.Mylar.objects.order_by('-id')[0] mylar.enable = True mylar.save() except IndexError: mylar = models.Mylar.objects.create(enable=True) try: form = forms.MylarForm(mylar.__dict__, instance=mylar, jail_path=jail_path) form.is_valid() form.save() except ValueError: return HttpResponse(simplejson.dumps({ 'error': True, 'message': ('Mylar data did not validate, configure ' 'it first.'), }), content_type='application/json') cmd = "%s onestart" % utils.mylar_control pipe = Popen(cmd, stdin=PIPE, stdout=PIPE, stderr=PIPE, shell=True, close_fds=True) out = pipe.communicate()[0] return HttpResponse(simplejson.dumps({ 'error': False, 'message': out, }), content_type='application/json')