From: francisco.garcia <francisco.garcia@baculasystems.com>
Date: Fri, 11 Aug 2023 10:25:28 +0000 (+0200)
Subject: Fix #0010329: No module named 'csi_snapshot'
X-Git-Tag: Beta-15.0.0~138
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=43c4e194f852c72119523146e19daa2c17a2dcc1;p=thirdparty%2Fbacula.git

Fix #0010329: No module named 'csi_snapshot'
---

diff --git a/bacula/src/plugins/fd/kubernetes-backend/baculak8s/plugins/k8sbackend/csi_snapshot.py b/bacula/src/plugins/fd/kubernetes-backend/baculak8s/plugins/k8sbackend/csi_snapshot.py
new file mode 100644
index 000000000..c0ae5d74f
--- /dev/null
+++ b/bacula/src/plugins/fd/kubernetes-backend/baculak8s/plugins/k8sbackend/csi_snapshot.py
@@ -0,0 +1,106 @@
+# -*- coding: UTF-8 -*-
+#
+#  Bacula(R) - The Network Backup Solution
+#
+#   Copyright (C) 2000-2023 Kern Sibbald
+#
+#   The original author of Bacula is Kern Sibbald, with contributions
+#   from many others, a complete list can be found in the file AUTHORS.
+#
+#   You may use this file and others of this release according to the
+#   license defined in the LICENSE file, which includes the Affero General
+#   Public License, v3.0 ("AGPLv3") and some additional permissions and
+#   terms pursuant to its AGPLv3 Section 7.
+#
+#   This notice must be preserved when any source code is
+#   conveyed and/or propagated.
+#
+#   Bacula(R) is a registered trademark of Kern Sibbald.
+#
+
+SNAPSHOT_DRIVER_COMPATIBLE='csi'
+
+K8SOBJ_SNAPSHOT_GROUP = 'snapshot.storage.k8s.io'
+K8SOBJ_SNAPSHOT_VERSION = 'v1beta1'
+K8SOBJ_SNAPSHOT_PLURAL = 'volumesnapshots'
+K8SOBJ_SNAPSHOT_KIND = 'VolumeSnapshot'
+K8SOBJ_SNAPSHOT_NAME_TEMPLATE = 'bacula-vsnap-{pvc}-{jobid}'
+BACKUP_PVC_FROM_SNAPSHOT_TEMPLATE = 'bacula-pvcfs-{pvc}-{jobid}'
+K8SOBJ_SNAPSHOT_CLASS = 'csi-hostpath-snapclass'
+
+def csi_snapshots_read_namespaced(crd_api, namespace, name):
+    return crd_api.get_namespaced_custom_object(K8SOBJ_SNAPSHOT_GROUP, K8SOBJ_SNAPSHOT_VERSION, namespace, K8SOBJ_SNAPSHOT_PLURAL, name)
+    
+
+def csi_snapshots_namespaced_names(crd_api, namespace, labels=""):
+    snapdict = {}
+    snaps = crd_api.list_namespaced_custom_object(K8SOBJ_SNAPSHOT_GROUP, K8SOBJ_SNAPSHOT_VERSION, namespace, K8SOBJ_SNAPSHOT_PLURAL, watch=False, label_selector=labels)
+    for snap in snaps.items:
+        snapdict[snap.metadata.name] = {
+            'name': snap.metadata.name,
+            'api_version': snap.apiVersion,
+            'kind': snap.kind,
+            'namespace': snap.metadata.namespace,
+            'resourceVersion': snap.metadata.resourceVersion,
+            'uid': snap.metadata.uid,
+            'pvc_source': snap.spec.source.persistentVolumeClaimName,
+            'class_name': snap.spec.volumeSnapshotClassName,
+            'creation_time': snap.status.creationTime,
+            'ready_to_use': snap.status.readyToUse,
+            'restore_size': snap.status.restoreSize
+        }
+    return snapdict
+
+
+def prepare_create_snapshot_body(namespace, pvc_name, jobid):
+    return {
+        "group": K8SOBJ_SNAPSHOT_GROUP,
+        "version": K8SOBJ_SNAPSHOT_VERSION,
+        "namespace": namespace,
+        "plural": K8SOBJ_SNAPSHOT_PLURAL,
+        "body": {
+            "apiVersion": "{}/{}".format(K8SOBJ_SNAPSHOT_GROUP, K8SOBJ_SNAPSHOT_VERSION),
+            "kind": K8SOBJ_SNAPSHOT_KIND,
+            "metadata": {
+                "name": K8SOBJ_SNAPSHOT_NAME_TEMPLATE.format(pvc=pvc_name, jobid=jobid),
+                "namespace": namespace
+            },
+            "spec": {
+                "volumeSnapshotClassName": K8SOBJ_SNAPSHOT_CLASS,
+                "source": {
+                    "persistentVolumeClaimName": pvc_name
+                }
+            }
+        }
+    }
+
+def prepare_snapshot_action(namespace, vsnapshot_name):
+    return {
+        "group": K8SOBJ_SNAPSHOT_GROUP,
+        "version": K8SOBJ_SNAPSHOT_VERSION,
+        "namespace": namespace,
+        "plural": K8SOBJ_SNAPSHOT_PLURAL,
+        "name": vsnapshot_name 
+    }
+
+def prepare_pvc_from_vsnapshot_body(namespace, pvcdata, jobid):
+    vsnapshot_name = K8SOBJ_SNAPSHOT_NAME_TEMPLATE.format(pvc=pvcdata.get('name'), jobid=jobid)
+    return {
+        'api_version': 'v1',
+        'kind': 'PersistentVolumeClaim',
+        'metadata': {
+            'name': BACKUP_PVC_FROM_SNAPSHOT_TEMPLATE.format(pvc=pvcdata.get('name'), jobid=jobid),
+            'namespace': namespace
+        },
+        'spec': {
+            'storageClassName': pvcdata.get('storage_class_name'),
+            'dataSource': {
+                'name': vsnapshot_name,
+                'kind': K8SOBJ_SNAPSHOT_KIND,
+                'apiGroup': K8SOBJ_SNAPSHOT_GROUP
+            },
+            'accessModes': ['ReadOnlyMany'],
+            'resources': { 'requests': {'storage': pvcdata.get('capacity')}}
+        }
+    }
+