Example #1
0
File: views.py Project: Arcylus/PBI
def stop(request, plugin_id):
    (couchpotato_key, couchpotato_secret) = utils.get_couchpotato_oauth_creds()

    url = utils.get_rpc_url(request)
    trans = OAuthTransport(url, key=couchpotato_key,
        secret=couchpotato_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:
        couchpotato = models.Couchpotato.objects.order_by('-id')[0]
        couchpotato.enable = False
        couchpotato.save()
    except IndexError:
        couchpotato = models.Couchpotato.objects.create(enable=False)

    try:
        form = forms.CouchpotatoForm(couchpotato.__dict__, instance=couchpotato, jail_path=jail_path)
        form.is_valid()
        form.save()
    except ValueError:
        pass

    cmd = "%s onestop" % utils.couchpotato_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')
Example #2
0
def treemenu(request, plugin_id):
    """
    This is how we inject nodes to the Tree Menu

    The FreeNAS GUI will access this view, expecting for a JSON
    that describes a node and possible some children.
    """

    (couchpotato_key, couchpotato_secret) = utils.get_couchpotato_oauth_creds()
    url = utils.get_rpc_url(request)
    trans = OAuthTransport(url, key=couchpotato_key, secret=couchpotato_secret)
    server = jsonrpclib.Server(url, transport=trans)
    jail = json.loads(server.plugins.jail.info(plugin_id))[0]
    jail_name = jail['fields']['jail_host']
    number = jail_name.rsplit('_', 1)
    name = "CouchPotato"
    if len(number) == 2:
        try:
            number = int(number)
            if number > 1:
                name = "CouchPotato (%d)" % number
        except:
            pass

    plugin = {
        'name': name,
        'append_to': 'plugins',
        'icon': reverse('treemenu_icon', kwargs={'plugin_id': plugin_id}),
        'type': 'pluginsfcgi',
        'url': reverse('couchpotato_edit', kwargs={'plugin_id': plugin_id}),
        'kwargs': {'plugin_name': 'couchpotato', 'plugin_id': plugin_id },
    }

    return HttpResponse(json.dumps([plugin]), content_type='application/json')
Example #3
0
def edit(request, plugin_id):
    from syslog import *
    (couchpotato_key, couchpotato_secret) = utils.get_couchpotato_oauth_creds()

    url = utils.get_rpc_url(request)
    trans = OAuthTransport(url, key=couchpotato_key, secret=couchpotato_secret)
    """
    Get the Couchpotato object
    If it does not exist create a new entry
    """
    try:
        couchpotato = models.Couchpotato.objects.order_by('-id')[0]
    except IndexError:
        couchpotato = models.Couchpotato.objects.create()

    try:
        server = jsonrpclib.Server(url, transport=trans)
        jail_path = server.plugins.jail.path(plugin_id)
        auth = server.plugins.is_authenticated(
            request.COOKIES.get("sessionid", ""))
        assert auth
    except Exception:
        raise

    if request.method == "GET":
        form = forms.CouchpotatoForm(instance=couchpotato, jail_path=jail_path)
        return render(request, "edit.html", {
            'form': form,
        })

    if not request.POST:
        return JsonResponse(request, error=True, message="A problem occurred.")

    form = forms.CouchpotatoForm(request.POST,
                                 instance=couchpotato,
                                 jail_path=jail_path)
    if form.is_valid():
        form.save()

        cmd = "%s restart" % utils.couchpotato_control
        pipe = Popen(cmd,
                     stdin=PIPE,
                     stdout=PIPE,
                     stderr=PIPE,
                     shell=True,
                     close_fds=True)

        return JsonResponse(request,
                            error=True,
                            message="Couchpotato settings successfully saved.")

    return JsonResponse(request, form=form)
Example #4
0
def open_view(request, plugin_id):
    (couchpotato_key,
    couchpotato_secret) = utils.get_couchpotato_oauth_creds()
    url = utils.get_rpc_url(request)
    trans = OAuthTransport(url, key=couchpotato_key,
        secret=couchpotato_secret)
    server = jsonrpclib.Server(url, transport=trans)
    jail = json.loads(server.plugins.jail.info(plugin_id))[0]
    jail_ipv4 = jail['fields']['jail_ipv4'].split('/', 1)[0]

    return render(request, "open.html", {
        'ipv4': jail_ipv4,
    })
Example #5
0
def start(request, plugin_id):
    (couchpotato_key, couchpotato_secret) = utils.get_couchpotato_oauth_creds()

    url = utils.get_rpc_url(request)

    trans = OAuthTransport(url, key=couchpotato_key, secret=couchpotato_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:
        couchpotato = models.Couchpotato.objects.order_by('-id')[0]
        couchpotato.enable = True
        couchpotato.save()
    except IndexError:
        couchpotato = models.Couchpotato.objects.create(enable=True)

    try:
        form = forms.CouchpotatoForm(couchpotato.__dict__,
                                     instance=couchpotato,
                                     jail_path=jail_path)
        form.is_valid()
        form.save()
    except ValueError:
        return HttpResponse(simplejson.dumps({
            'error':
            True,
            'message':
            'Couchpotato data did not validate, please configure it first.',
        }),
                            content_type='application/json')

    cmd = "%s onestart" % utils.couchpotato_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')
Example #6
0
File: views.py Project: Arcylus/PBI
def edit(request, plugin_id):
    from syslog import *
    (couchpotato_key, couchpotato_secret) = utils.get_couchpotato_oauth_creds()

    url = utils.get_rpc_url(request)
    trans = OAuthTransport(url, key=couchpotato_key,
        secret=couchpotato_secret)

    """
    Get the Couchpotato object
    If it does not exist create a new entry
    """
    try:
        couchpotato = models.Couchpotato.objects.order_by('-id')[0]
    except IndexError:
        couchpotato = models.Couchpotato.objects.create()

    try:
        server = jsonrpclib.Server(url, transport=trans)
        jail_path = server.plugins.jail.path(plugin_id)
        auth = server.plugins.is_authenticated(request.COOKIES.get("sessionid", ""))
        assert auth
    except Exception:
        raise

    if request.method == "GET":
        form = forms.CouchpotatoForm(instance=couchpotato,
            jail_path=jail_path)
        return render(request, "edit.html", {
            'form': form,
        })

    if not request.POST:
        return JsonResponse(request, error=True, message="A problem occurred.")

    form = forms.CouchpotatoForm(request.POST,
        instance=couchpotato,
        jail_path=jail_path)
    if form.is_valid():
        form.save()

        cmd = "%s restart" % utils.couchpotato_control
        pipe = Popen(cmd, stdin=PIPE, stdout=PIPE, stderr=PIPE, shell=True, close_fds=True)

        return JsonResponse(request, error=True, message="Couchpotato settings successfully saved.")

    return JsonResponse(request, form=form)
Example #7
0
def treemenu(request, plugin_id):
    """
    This is how we inject nodes to the Tree Menu

    The FreeNAS GUI will access this view, expecting for a JSON
    that describes a node and possible some children.
    """

    (couchpotato_key, couchpotato_secret) = utils.get_couchpotato_oauth_creds()

    url = utils.get_rpc_url(request)
    trans = OAuthTransport(url, key=couchpotato_key, secret=couchpotato_secret)
    server = jsonrpclib.Server(url, transport=trans)
    jail = json.loads(server.plugins.jail.info(plugin_id))[0]
    jail_name = jail['fields']['jail_host']
    number = jail_name.rsplit('_', 1)
    name = "Couchpotato"

    if len(number) == 2:
        try:
            number = int(number)
            if number > 1:
                name = "Couchpotato (%d)" % number
        except:
            pass

    plugin = {
        'name': name,
        'append_to': 'plugins',
        'icon': reverse('treemenu_icon', kwargs={'plugin_id': plugin_id}),
        'type': 'pluginsfcgi',
        'url': reverse('couchpotato_edit', kwargs={'plugin_id': plugin_id}),
        'kwargs': {
            'plugin_name': 'couchpotato',
            'plugin_id': plugin_id
        },
    }

    return HttpResponse(json.dumps([plugin]), content_type='application/json')