mirror of
https://github.com/retspen/webvirtcloud
synced 2024-11-01 20:14:15 +00:00
85 lines
2.9 KiB
Python
85 lines
2.9 KiB
Python
from secrets.forms import AddSecret
|
|
|
|
from admin.decorators import superuser_only
|
|
from computes.models import Compute
|
|
from django.contrib import messages
|
|
from django.http import HttpResponseRedirect
|
|
from django.shortcuts import get_object_or_404, render
|
|
from libvirt import (
|
|
VIR_SECRET_USAGE_TYPE_CEPH,
|
|
VIR_SECRET_USAGE_TYPE_ISCSI,
|
|
VIR_SECRET_USAGE_TYPE_NONE,
|
|
VIR_SECRET_USAGE_TYPE_TLS,
|
|
VIR_SECRET_USAGE_TYPE_VOLUME,
|
|
libvirtError,
|
|
)
|
|
from vrtManager.secrets import wvmSecrets
|
|
|
|
|
|
@superuser_only
|
|
def secrets(request, compute_id):
|
|
"""
|
|
:param request:
|
|
:param compute_id:
|
|
:return:
|
|
"""
|
|
|
|
secrets_all = []
|
|
compute = get_object_or_404(Compute, pk=compute_id)
|
|
secret_usage_types = {
|
|
VIR_SECRET_USAGE_TYPE_NONE: "none",
|
|
VIR_SECRET_USAGE_TYPE_VOLUME: "volume",
|
|
VIR_SECRET_USAGE_TYPE_CEPH: "ceph",
|
|
VIR_SECRET_USAGE_TYPE_ISCSI: "iscsi",
|
|
VIR_SECRET_USAGE_TYPE_TLS: "tls",
|
|
}
|
|
|
|
try:
|
|
conn = wvmSecrets(compute.hostname, compute.login, compute.password, compute.type)
|
|
secrets = conn.get_secrets()
|
|
|
|
for uuid in secrets:
|
|
secrt = conn.get_secret(uuid)
|
|
try:
|
|
secrt_value = conn.get_secret_value(uuid)
|
|
except libvirtError:
|
|
secrt_value = None
|
|
secrets_all.append(
|
|
{
|
|
"usage": secrt.usageID(),
|
|
"uuid": secrt.UUIDString(),
|
|
"usageType": secret_usage_types[secrt.usageType()],
|
|
"value": secrt_value,
|
|
}
|
|
)
|
|
if request.method == "POST":
|
|
if "create" in request.POST:
|
|
form = AddSecret(request.POST)
|
|
if form.is_valid():
|
|
data = form.cleaned_data
|
|
conn.create_secret(
|
|
data["ephemeral"],
|
|
data["private"],
|
|
data["usage_type"],
|
|
data["data"],
|
|
)
|
|
return HttpResponseRedirect(request.get_full_path())
|
|
else:
|
|
for msg_err in form.errors.values():
|
|
messages.error(request, msg_err.as_text())
|
|
if "delete" in request.POST:
|
|
uuid = request.POST.get("uuid", "")
|
|
conn.delete_secret(uuid)
|
|
return HttpResponseRedirect(request.get_full_path())
|
|
if "set_value" in request.POST:
|
|
uuid = request.POST.get("uuid", "")
|
|
value = request.POST.get("value", "")
|
|
try:
|
|
conn.set_secret_value(uuid, value)
|
|
except Exception as err:
|
|
messages.error(request, err)
|
|
return HttpResponseRedirect(request.get_full_path())
|
|
except libvirtError as err:
|
|
messages.error(request, err)
|
|
|
|
return render(request, "secrets.html", locals())
|