add advanced mappings tool added

Author: lakepry25

add_mapping.py

@@ -0,0 +1,47 @@
+from login import global_admin_login
+from filer import get_filers, get_filer
+from cterasdk import settings
+
+import logging
+
+from cterasdk import common_types
+
+def get_adv_mapping(edge):
+    highest_id = 0
+    mappings = []
+    for domain, mapping in edge.directoryservice.get_advanced_mapping().items():
+      print(f"Domain: {domain}")
+      print(f"Mapping: {mapping}")
+      if mapping.maxID > highest_id:
+          highest_id = mapping.maxID
+      mappings.append(mapping)
+
+    return mappings, highest_id
+  
+def add_mapping(address, username, password, domain, device_name=None, tenant=None):
+  try:
+    settings.sessions.management.ssl = False
+    ga = global_admin_login(address, username, password, True)
+    if device_name:
+      filers=[get_filer(ga, device_name, tenant)]
+    elif tenant:
+      filers = get_filers(ga, tenant=tenant, all_tenants=False)
+    else:
+      filers = get_filers(ga, all_tenants=True)
+
+    for edge in filers:
+      mappings, highest_id = get_adv_mapping(edge)
+
+      mappings.append(common_types.ADDomainIDMapping(domain, highest_id + 1, highest_id + 200000))
+
+      for mapping in mappings:
+          print(f"Mapping: {mapping}")
+
+      logging.info(f"Attempting to add mapping for domain {domain} to {edge.name}")
+      edge.directoryservice.set_advanced_mapping(mappings)
+      logging.info(f"Added mapping for domain {domain} to {edge.name}")
+  except Exception as e:
+    logging.error(f"Error: {e}")
+  finally:
+     ga.logout()
+    

ctools.py

@@ -20,6 +20,7 @@
 from windows.AddMembersWindow import addMembersWindow
 from windows.ReportZonesWindow import reportZonesWindow
 from windows.PopulateCloudFoldersWindow import populateCloudFoldersWindow
+from windows.AddMappingWindow import addMappingWindow
 #from windows.SMBAuditWindow import smbAuditWindow
 
 from PySide6 import QtCore
@@ -87,6 +88,9 @@ def main():
     populate_shares = populateCloudFoldersWindow(widget)
     widget.addWidget(populate_shares)
 
+    add_mapping = addMappingWindow(widget)
+    widget.addWidget(add_mapping)
+
     ## STEP7- Add new windows above this line ##
 
     widget.setCurrentWidget(run_cmd)

ui_help.py

@@ -31,12 +31,13 @@ def create_tool_bar(widget, currentWindow):
     add_members = QPushButton("Add/Remove Members")
     report_zones = QPushButton("Report Zones")
     populate_shares = QPushButton("Populate Shares")
+    add_mapping = QPushButton("Add Domain to Advanced Mapping")
 
     #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, report_zones, populate_shares]
+    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, add_mapping]
 
     tool_list[currentWindow].setStyleSheet("color: darkblue; background-color: lightblue;")   
 

windows/AddMappingWindow.py

@@ -0,0 +1,106 @@
+import logging
+from log_setter import set_logging
+## STEP6a - import the tool function from the file you imported into the CTOOLS3 project folder
+from add_mapping import add_mapping
+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 = 700
+WINDOW_HEIGHT = 600
+OUTPUT_HEIGHT = 250
+class addMappingWindow(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("C:\\Users\\lakea\\Desktop\\CTERA\\ctools\\logo.png")#replace with image location
+        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, 14)
+        # 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("Add Domain to Advanced Mapping", ["Tenant (Empty if all devices on all tenants)", "Device Name (Empty if multiple devices)", "Domain to be Added"], ["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()
+        tenant = self.input_widgets[3].text()
+        device_name = self.input_widgets[4].text()
+        domain_name = self.input_widgets[5].text()
+        ignore_cert = self.input_widgets[6].isChecked()
+        verbose = self.input_widgets[7].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)
+        if device_name:
+            add_mapping(portal_address, portal_username, portal_password, domain_name, device_name=device_name, tenant=tenant)
+        elif tenant:
+            add_mapping(portal_address, portal_username, portal_password, domain_name, tenant=tenant)
+        else:
+            add_mapping(portal_address, portal_username, portal_password, domain_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())