Merge pull request #6 from bmrodrigues/main

Latest MDEcli version

MDEcli/app.py

@@ -24,7 +24,6 @@ try:
     tenantid = input('Please enter your Tenant ID: ')
     appid = input('Please enter your Application ID: ')
     secret = getpass.getpass('Please enter your Application Secret:')
-    webhook = getpass.getpass('Please enter your webhook from Teams: ')
     new_token = gettoken.GetToken(tenantid, appid, secret)
     new_token.gettoken()
     token = new_token.aadToken
@@ -33,14 +32,12 @@ try:
     if args.Commands == 'actions':
         if args.offboard:
             filename = input('Please enter the CSV filename (export from MDE/Devices): ')
-            comment = input('COmment (mandatory): ')
+            comment = input('Comment (mandatory): ')
             new_csv = read.Csv(filename)
             new_csv.open()
             for ids in new_csv.list_ids:
-                new_offboard = offboard.Offboard(token, ids, comment)
-                new_offboard.offboard()
-                print(new_offboard.results)
-
+                new_offboard_action = offboard.Offboard(token, ids, comment)
+                new_offboard_action.offboard()
         elif args.quick:
             print("Let's quick scan")
         else:

MDEcli/mde_api_obj/offboard.py

@@ -9,6 +9,7 @@ class Offboard:
         self.comment = comment
 
     def offboard(self):
+        try:
             url = f"https://api.securitycenter.microsoft.com/api/machines/{self.id}/offboard"
             headers = { 
                 'Content-Type' : 'application/json',
@@ -21,5 +22,6 @@ class Offboard:
             req = urllib.request.Request(url, data, headers)
             response = urllib.request.urlopen(req)
             jsonResponse = json.loads(response.read())
-        self.results = jsonResponse["Results"]
-        return self.results
+            print(jsonResponse)
+        except Exception as e:
+            print(e)