Populate Shares

Author: lakepry25

ctools.py

@@ -19,6 +19,7 @@
 from windows.ImportSharesWindow import importSharesWindow
 from windows.AddMembersWindow import addMembersWindow
 from windows.ReportZonesWindow import reportZonesWindow
+from windows.PopulateCloudFoldersWindow import populateCloudFoldersWindow
 #from windows.SMBAuditWindow import smbAuditWindow
 
 from PySide6 import QtCore
@@ -39,7 +40,7 @@ def main():
 
     widget = QStackedWidget()
 
-    widget.setWindowTitle("CTools v3.1a")
+    widget.setWindowTitle("CTools v3.1b")
 
     widget.setWindowIcon(QIcon('icon.jpeg'))
 
@@ -83,6 +84,9 @@ def main():
     report_zones = reportZonesWindow(widget)
     widget.addWidget(report_zones)
 
+    populate_shares = populateCloudFoldersWindow(widget)
+    widget.addWidget(populate_shares)
+
     ## STEP7- Add new windows above this line ##
 
     widget.setCurrentWidget(run_cmd)

populate.py

@@ -0,0 +1,65 @@
+from cterasdk import *
+from cterasdk import settings
+from filer import get_filer, get_filers
+
+import urllib3
+urllib3.disable_warnings()
+
+import logging
+
+from login import global_admin_login
+
+def get_cloud_folders(self):
+  """Get all cloud folders on a tenant."""
+  try:
+    result = self.cloudfs.drives.all()
+    logging.info("Finished get_cloud_folders task.")
+
+    return result
+  except CTERAException as ce:
+    logging.error(ce)
+    logging.error("Failed get_cloud_folders task.")
+      
+  return None
+
+def create_shares_from_folders_list(self, folders, edge_filer):
+  """Create shares from a list of cloud folders."""
+  try:
+    for folder in folders:
+      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)
+
+      
+  except CTERAException as ce:
+    logging.error(ce)
+    logging.error("Failed create_shares_from_folders_list task.")
+
+def run_populate(address, username, password, device_name):
+  try:
+    settings.sessions.management.ssl = False
+    global_admin = global_admin_login(address, username, password, True)
+    folders = get_cloud_folders(global_admin)
+
+    filer = get_filer(global_admin, device_name)
+
+    create_shares_from_folders_list(global_admin, folders, filer)
+  except CTERAException as ce:
+    logging.error(ce)
+    logging.error("Failed creating share task.")
+  finally:
+    global_admin.logout()
+
+
+"""if __name__ == "__main__":
+  run_populate("192.168.22.197", "admin", "lap6056*", "labgw")"""

share_add.py

@@ -0,0 +1,4 @@
+from populate import run_populate
+
+if __name__ == "__main__":
+  run_populate("192.168.22.197", "admin", "lap6056*", "labgw")

ui_help.py

@@ -29,15 +29,16 @@ def create_tool_bar(widget, currentWindow):
     delete_shares = QPushButton("Delete Shares")
     import_shares = QPushButton("Copy Shares")
     add_members = QPushButton("Add/Remove Members")
-    zones_report = QPushButton("Zones Report")
+    report_zones = QPushButton("Report Zones")
+    populate_shares = QPushButton("Populate Shares")
 
     #STEP8 - Create the push button above so you can navigate to the tool
 
 
     # STEP9 - Add the button you just created to the list below. MAKE SURE YOU PUT IT BEFORE EXIT
-    tool_list = [run_cmd, show_status, suspend_sync, unsuspend_sync, enable_ssh, disable_ssh, enable_telnet, reset_password, cloud_folders, delete_shares, import_shares, add_members, zones_report]
+    tool_list = [run_cmd, show_status, suspend_sync, unsuspend_sync, enable_ssh, disable_ssh, enable_telnet, reset_password, cloud_folders, delete_shares, import_shares, add_members, report_zones, populate_shares]
 
-    tool_list[currentWindow].setStyleSheet("color: darkblue; background-color: lightblue; ")   
+    tool_list[currentWindow].setStyleSheet("color: darkblue; background-color: lightblue;")   
 
     tools.addWidget(label, alignment=Qt.AlignTop)
 

windows/PopulateCloudFoldersWindow.py

@@ -0,0 +1,98 @@
+import logging
+from log_setter import set_logging
+## STEP6a - import the tool function from the file you imported into the CTOOLS3 project folder
+from populate import run_populate
+from ui_help import gen_tool_layout, gen_custom_tool_layout, create_tool_bar
+from login import global_admin_login
+from PySide6.QtCore import Qt
+from PySide6.QtWidgets import (
+    QMainWindow,
+    QWidget,
+    QPushButton,
+    QVBoxLayout,
+    QLabel,
+    QHBoxLayout,
+    QTextEdit,
+    QFrame,
+)
+from PySide6.QtGui import (
+    QPixmap
+)
+WINDOW_WIDTH = 600
+WINDOW_HEIGHT = 500
+OUTPUT_HEIGHT = 250
+class populateCloudFoldersWindow(QMainWindow):
+    """PyCalc's main window (GUI or view)."""
+    def __init__(self, widget):
+        super().__init__()
+        self.widget = widget
+        self.setWindowTitle("CTools 3.0")
+        self.setFixedSize(WINDOW_WIDTH, WINDOW_HEIGHT)
+        self.generalLayout = QVBoxLayout()
+        self.top = QHBoxLayout()
+        welcome = QLabel("<h2>Welcome to CTools!</h2><h5>One tool for all</h5>")
+        pic_label = QLabel(self)
+        pixmap = QPixmap("logo.png")
+        pic_label.setPixmap(pixmap)
+        #pic_label.setScaledContents(True)
+        self.top.addWidget(welcome)
+        self.top.addStretch()
+        self.top.addWidget(pic_label)
+        self.mainContent = QHBoxLayout()
+        centralWidget = QWidget(self)
+        centralWidget.setLayout(self.generalLayout)
+        self.setCentralWidget(centralWidget)
+        self.generalLayout.addLayout(self.top)
+        self.generalLayout.addLayout(self.mainContent)
+        self._createToolBar()
+        self._createToolViewLayout()
+    def _createToolBar(self):
+        tools = create_tool_bar(self.widget, 13)
+        # Add line separator between Tool List and Tool View
+        line = QFrame()
+        line.setFrameShape(QFrame.VLine)
+        line.setFrameShadow(QFrame.Sunken)
+        line.setLineWidth(1)
+        self.mainContent.addLayout(tools)
+        self.mainContent.addWidget(line)
+    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"])
+        toolView.addLayout(BoilerLayout)
+        # Create action buttons
+        actionButtonLayout = QHBoxLayout()
+        self.cancel = QPushButton("Cancel")
+        self.start = QPushButton("Start")
+        actionButtonLayout.addWidget(self.cancel)
+        actionButtonLayout.addWidget(self.start)
+        toolView.addLayout(actionButtonLayout)
+        # STEP5 - Add button listeners
+        self.start.clicked.connect(self.tool)
+        # Create Output box
+        self.output = QTextEdit()
+        self.output.setReadOnly(True)
+        toolView.addWidget(self.output)
+        self.mainContent.addLayout(toolView)
+    # STEP4 - Grab the arguments for you tool
+    def tool(self):
+        portal_address = self.input_widgets[0].text()
+        portal_username = self.input_widgets[1].text()
+        portal_password = self.input_widgets[2].text()
+        device_name = self.input_widgets[3].text()
+        ignore_cert = self.input_widgets[4].isChecked()
+        verbose = self.input_widgets[5].isChecked()
+        if verbose:
+            set_logging(logging.DEBUG, 'debug-log.txt')
+        else:
+            set_logging()
+        ## Step6b - Run the tool here
+        # Ex: run_status(global_admin, filename, all_tenants_flag)
+        run_populate(portal_address, portal_username, portal_password, device_name)
+        self._updateOutput()
+    def _updateOutput(self):
+        file = open("output.tmp", 'r')
+        with file:
+            text = file.read()
+            self.output.setText(text)
+        self.output.verticalScrollBar().setValue(self.output.verticalScrollBar().maximum())

windows/ReportZonesWindow.py

@@ -79,6 +79,7 @@ def _createToolViewLayout(self):
 
         password = QLabel("Portal Admin Password:")
         self.password_field = QLineEdit()
+        self.password_field.setEchoMode(QLineEdit.Password)
 
         output_dir = QLabel("Output Location:")
         self.output_field = QHBoxLayout()