chore(azure): Manage new errors in the Defender service (prowler-clou…

…d#3534)
garym-krrv · Mar 13, 2024 · ed60958 · ed60958
1 parent 85f4595
commit ed60958
Showing 1 changed file with 21 additions and 3 deletions.
diff --git a/prowler/providers/azure/services/defender/defender_service.py b/prowler/providers/azure/services/defender/defender_service.py
@@ -1,7 +1,11 @@
 from datetime import timedelta
 from typing import Dict
 
-from azure.core.exceptions import HttpResponseError
+from azure.core.exceptions import (
+    ClientAuthenticationError,
+    HttpResponseError,
+    ResourceNotFoundError,
+)
 from azure.mgmt.security import SecurityCenter
 from pydantic import BaseModel
 
@@ -48,6 +52,11 @@ def __get_pricings__(self):
                             )
                         }
                     )
+            except ResourceNotFoundError as error:
+                if "Subscription Not Registered" in error.message:
+                    logger.error(
+                        f"Subscription name: {subscription_name} -- {error.__class__.__name__}[{error.__traceback__.tb_lineno}]: Subscription Not Registered - Please register to Microsoft.Security in order to view your security status"
+                    )
             except Exception as error:
                 logger.error(
                     f"Subscription name: {subscription_name} -- {error.__class__.__name__}[{error.__traceback__.tb_lineno}]: {error}"
@@ -72,10 +81,14 @@ def __get_auto_provisioning_settings__(self):
                             )
                         }
                     )
+            except ClientAuthenticationError as error:
+                if "Subscription Not Registered" in error.message:
+                    logger.error(
+                        f"Subscription name: {subscription_name} -- {error.__class__.__name__}[{error.__traceback__.tb_lineno}]: Subscription Not Registered - Please register to Microsoft.Security in order to view your security status"
+                    )
             except Exception as error:
-                logger.error(f"Subscription name: {subscription_name}")
                 logger.error(
-                    f"{error.__class__.__name__}[{error.__traceback__.tb_lineno}]: {error}"
+                    f"Subscription name: {subscription_name} -- {error.__class__.__name__}[{error.__traceback__.tb_lineno}]: {error}"
                 )
         return auto_provisioning
 
@@ -122,6 +135,11 @@ def __get_settings__(self):
                             )
                         }
                     )
+            except ClientAuthenticationError as error:
+                if "Subscription Not Registered" in error.message:
+                    logger.error(
+                        f"Subscription name: {subscription_name} -- {error.__class__.__name__}[{error.__traceback__.tb_lineno}]: Subscription Not Registered - Please register to Microsoft.Security in order to view your security status"
+                    )
             except Exception as error:
                 logger.error(
                     f"Subscription name: {subscription_name} -- {error.__class__.__name__}[{error.__traceback__.tb_lineno}]: {error}"