Merge pull request #44 from ctera/copy-shares-nfs

Added a couple of tools as well as support for copying nfs shares in …

Author: lakepry25

cli-testing.py

@@ -0,0 +1,85 @@
+import cterasdk
+from filer import get_filer, get_filers
+from cterasdk import Edge, edge_types, edge_enum
+from login import global_admin_login
+
+def copy_nfs_share(ga, share, dest):
+  try:
+    tenant = ga.users.session().user.tenant
+    filer_dest = get_filer(ga, dest, tenant)
+
+    client_list = []
+    for client in share.trustedNFSClients:
+      if client.accessLevel == "ReadWrite":
+        client_list.append(edge_types.NFSv3AccessControlEntry(client.address, client.netmask, edge_enum.FileAccessMode.RW))
+      elif client.accessLevel == "ReadOnly":
+        client_list.append(edge_types.NFSv3AccessControlEntry(client.address, client.netmask, edge_enum.FileAccessMode.RO))
+    
+    filer_dest.shares.add(share.name, share.directory[1:], export_to_nfs=True, trusted_nfs_clients=client_list)
+  except Exception as e:
+    print("Failed to copy share: %s" % share.name)
+    print(e)
+
+def get_nfs_shares(ga, source):
+  tenant = ga.users.session().user.tenant
+  filer = get_filer(ga, source, tenant)
+
+  try:
+    shares = filer.shares.get()
+
+    nfs_shares = []
+
+    for share in shares:
+      if share.exportToNFS:
+        nfs_shares.append(share)
+    
+    return nfs_shares
+  except Exception as e:
+    print("Failed to get shares from device: %s" % source)
+    print(e)
+    return None
+
+def copy_nfs_shares(ga, source, dest):
+  try:
+    tenant = ga.users.session().user.tenant
+    filer_source = get_filer(ga, source, tenant)
+    filer_dest = get_filer(ga, dest, tenant)
+
+    nfs_shares_source = get_nfs_shares(ga, source)
+
+    for share in nfs_shares_source:
+      try:
+        #get the nfs clients from the source share
+        source_nfs_clients = filer_source.shares.get(share.name).trustedNFSClients
+
+        print(dir(source_nfs_clients[0]))
+
+        print("nfs client accessLevel: %s" % source_nfs_clients[0].accessLevel)
+        print("nfs client address: %s" % source_nfs_clients[0].address)
+        print("nfs client netmask: %s" % source_nfs_clients[0].netmask)
+
+        client_list = []
+        for client in source_nfs_clients:
+          if client.accessLevel == "ReadWrite":
+            client_list.append(edge_types.NFSv3AccessControlEntry(client.address, client.netmask, edge_enum.FileAccessMode.RW))
+          elif client.accessLevel == "ReadOnly":
+            client_list.append(edge_types.NFSv3AccessControlEntry(client.address, client.netmask, edge_enum.FileAccessMode.RO))
+        
+        filer_dest.shares.add(share.name, share.directory[1:], export_to_nfs=True, trusted_nfs_clients=client_list)
+      except Exception as e:
+        print("Failed to copy share: %s" % share.name)
+        print(e)
+
+  except Exception as e:
+    print("Failed to copy shares from device: %s to device: %s" % (source, dest))
+    print(e)
+
+
+"""if __name__ == "__main__":
+  ga = global_admin_login("192.168.22.197", "admin", "lap6056*", True)
+  try:
+    copy_nfs_shares(ga, "lakegw2", "labgw")
+  except Exception as e:
+    print(e)
+  finally:
+    ga.logout()"""

copynfs.py

@@ -1,5 +1,6 @@
 import cterasdk
 from filer import get_filer, get_filers
+from copynfs import copy_nfs_share
 from cterasdk import Edge, edge_types, edge_enum
 import logging
 
@@ -42,6 +43,10 @@ def copyshares(self, source, dest):
       if share.name == 'public' or share.name == 'cloud' or share.name == 'backups':
         logging.info("%s skipped" % str(share.name))
         continue
+      elif share.exportToNFS:
+        logging.info("Copying NFS share: %s" % str(share.name))
+        copy_nfs_share(self, share, dest)
+        continue
       try:
         acl_entries = []
         for acl in share.acl:

copyshares.py

@@ -40,7 +40,7 @@ def main():
 
     widget = QStackedWidget()
 
-    widget.setWindowTitle("CTools v3.1b")
+    widget.setWindowTitle("CTools v3.1c")
 
     widget.setWindowIcon(QIcon('icon.jpeg'))
 

ctools.py

@@ -29,16 +29,20 @@ def create_shares_from_folders_list(self, folders, edge_filer):
       if folder.name != 'My Files':
         logging.info("Creating share for folder: " + folder.name)
 
-        users = self.users.list_local_users(include = ['name', 'email', 'firstName', 'lastName'])
-        match = None
-        for user in users:
-          if user.name == folder.owner.split('/')[-1]:
-            match = user
-            break
-        
-        if match is not None:
-          path = "cloud/users/"+ match.firstName + " " + match.lastName + "/" + folder.name
-          edge_filer.shares.add(folder.name, path)
+        try:
+          users = self.users.list_local_users(include = ['name', 'email', 'firstName', 'lastName'])
+          match = None
+          for user in users:
+            if user.name == folder.owner.split('/')[-1]:
+              match = user
+              break
+          
+          if match is not None:
+            path = "cloud/users/"+ match.firstName + " " + match.lastName + "/" + folder.name
+            edge_filer.shares.add(folder.name, path)
+        except CTERAException as ce:
+          logging.error(ce)
+          logging.error("Failed creating share for folder: " + folder.name)
 
 
   except CTERAException as ce:

populate.py

@@ -18,8 +18,8 @@
 from PySide6.QtGui import (
     QPixmap
 )
-WINDOW_WIDTH = 600
-WINDOW_HEIGHT = 500
+WINDOW_WIDTH = 700
+WINDOW_HEIGHT = 600
 OUTPUT_HEIGHT = 250
 class populateCloudFoldersWindow(QMainWindow):
     """PyCalc's main window (GUI or view)."""
@@ -32,7 +32,7 @@ def __init__(self, widget):
         self.top = QHBoxLayout()
         welcome = QLabel("<h2>Welcome to CTools!</h2><h5>One tool for all</h5>")
         pic_label = QLabel(self)
-        pixmap = QPixmap("logo.png")
+        pixmap = QPixmap("C:\\Users\\lakea\\Desktop\\CTERA\\ctools\\logo.png")#replace with image location
         pic_label.setPixmap(pixmap)
         #pic_label.setScaledContents(True)
         self.top.addWidget(welcome)
@@ -58,7 +58,7 @@ def _createToolBar(self):
     def _createToolViewLayout(self):
         toolView = QVBoxLayout()
         # Step3 - You will change the next two lines according to the KB
-        BoilerLayout, self.input_widgets = gen_custom_tool_layout("Populate Cloud Folders", ["Device Name"], ["Ignore cert warnings for login", "Verbose Logging"])
+        BoilerLayout, self.input_widgets = gen_custom_tool_layout("Populate Cloud Folders as Shares (Except My Files)", ["Device Name"], ["Ignore cert warnings for login", "Verbose Logging"])
         toolView.addLayout(BoilerLayout)
         # Create action buttons
         actionButtonLayout = QHBoxLayout()