upd: fixing issues within the search mssp and export feature

certeu · Sep 23, 2024 · b96a50c · b96a50c
1 parent 2e99256
commit b96a50c
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/droid/platforms/sentinel.py b/src/droid/platforms/sentinel.py
@@ -192,7 +192,7 @@ def get_workspaces(self, credential, export_mode=False):
             else:
                 entry_dict = {
                     "customer": workspace_name,
-                    graph_key: graph_value
+                    "workspace_id": graph_value
                 }
 
             workspace_list.append(entry_dict)
@@ -435,18 +435,20 @@ def create_rule(self, rule_content, rule_converted, rule_file):
             techniques=self.mitre_techniques(rule_content)
         )
 
-        credential = self.get_credentials()
-
         if self._export_mssp:
             if self._export_list_mssp:
                 self.logger.info("Exporting to restricted customers")
                 for group, info in self._export_list_mssp.items():
+
                     workspace_name = info['workspace_name']
+                    self._tenant_id = info['tenant_id']
                     resource_group_name = info['resource_group_name']
                     subscription_id = info['subscription_id']
 
                     self.logger.debug(f"Exporting to {workspace_name}")
 
+                    credential = self.get_credentials()
+
                     # Create a new SecurityInsights client for the target subscription
                     client = SecurityInsights(credential, subscription_id)