CE-574 Upgrade Logic

PR URL: https://www.github.com/delphix/couchbase-plugin/pull/61

Author: ayesharustgi-delphix

src/plugin_runner.py

@@ -4,6 +4,7 @@
 #
 
 import logging
+import os
 
 from controller.couchbase_operation import CouchbaseOperation
 from controller.helper_lib import check_server_is_used
@@ -29,6 +30,34 @@
 logger = logging.getLogger(__name__)
 
 
+@plugin.upgrade.linked_source("2023.10.27.01")
+def update_bucket_size(old_linked_source):
+    old_linked_source = dict(old_linked_source)
+    if isinstance(old_linked_source["bucket_size"], int):
+        if old_linked_source["bucket_size"] == 0:
+            old_linked_source["bucket_size"] = []
+        else:
+            d = [{"bname": "*", "bsize": old_linked_source["bucket_size"]}]
+            old_linked_source["bucket_size"] = d
+    return old_linked_source
+
+
+@plugin.upgrade.linked_source("2023.10.27.02")
+def update_archive_name(old_linked_source):
+    old_linked_source = dict(old_linked_source)
+    if "archive_name" not in old_linked_source.keys():
+        if old_linked_source["couchbase_bak_loc"] == "":
+            old_linked_source["archive_name"] = ""
+        else:
+            old_linked_source["archive_name"] = os.path.basename(
+                old_linked_source["couchbase_bak_loc"]
+            )
+            old_linked_source["couchbase_bak_loc"] = os.path.dirname(
+                old_linked_source["couchbase_bak_loc"]
+            )
+    return old_linked_source
+
+
 #
 # Below is an example of the repository discovery operation.
 #