minor backups changes

Modified pluggable module exports to allow for the overwrite of an existing file. Backups of pluggables using the backups module now overwrite an already exported pluggable module

Author: mbomb67

onefuse/__init__.py

@@ -4,5 +4,5 @@
 Enables the execution of OneFuse policies via Python
 """
 
-__version__ = "9.4.7.2"
+__version__ = "9.4.7.3"
 __credits__ = 'Cloudbolt Software, Inc.'

onefuse/admin.py

@@ -437,7 +437,8 @@ def deprovision_ipam(self, ipam_id: int):
         return path
 
     # Pluggable Modules
-    def export_pluggable_module(self, module_name: str, save_path: str):
+    def export_pluggable_module(self, module_name: str, save_path: str,
+                                overwrite: bool = False):
         """
         Export a Pluggable Module to a file path. The exported module will be
         saved to a file
@@ -450,6 +451,9 @@ def export_pluggable_module(self, module_name: str, save_path: str):
             File path to save the exported module zip file to
             Windows: 'C:\\temp\\onefuse_backups\\'
             Linux: '/tmp/onefuse_backups/'
+        overwrite : bool
+            Boolean value whether to overwrite an existing file in save_path
+            Defaults to False
         """
         path = 'modules'
         module = self.get_policy_by_name(path, module_name)
@@ -465,7 +469,10 @@ def export_pluggable_module(self, module_name: str, save_path: str):
                     raise
         file_path = f'{save_path}{module_name}.zip'
         if os.path.isfile(file_path):
-            raise OneFuseError(f'Zip file already exists for module_name: '
+            if overwrite:
+                os.remove(file_path)
+            else:
+                raise OneFuseError(f'Zip file already exists for module_name: '
                                f'{module_name} in save_path: {save_path}')
         with open(file_path, 'wb') as fd:
             for chunk in response.iter_content(chunk_size=128):

onefuse/backups.py

@@ -137,7 +137,10 @@ def create_json_files(self, response, policy_type: str, backups_path: str):
                                   f'{policy["name"]}')
             file_path = f'{backups_path}{policy_type}{path_char}'
             if policy_type == 'modules':
-                self.ofm.export_pluggable_module(policy["name"], file_path)
+                # Modules will overwrite existing modules in the file path if
+                # found
+                self.ofm.export_pluggable_module(policy["name"], file_path,
+                                                 True)
                 continue
             if policy_type == "endpoints":
                 if "credential" in policy["_links"]:

setup.py

@@ -5,7 +5,7 @@
 
 setup(
     name='onefuse',
-    version='9.4.7.2',
+    version='9.4.7.3',
     author='Cloudbolt Software, Inc.',
     author_email='support@cloudbolt.io',
     description='OneFuse upstream provider package for Python',