ben.carrasco/prowler
1
0
Fork 0
mirror of https://github.com/prowler-cloud/prowler.git synced 2026-07-24 13:01:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(Azure): Optimize Entra service to use async funcs (#3706)

Browse Source
This commit is contained in:
Rubén De la Torre Vico
2024-04-09 09:20:06 +02:00
committed by GitHub
parent 06843cd41a
commit 331ae92843
1 changed files with 23 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
prowler/providers/azure/services/entra/entra_service.py
+23 -19
View File
@@ -1,4 +1,4 @@
import asyncio
from asyncio import gather, get_event_loop
from dataclasses import dataclass
from typing import Any, List, Optional
from uuid import UUID
@@ -19,26 +19,30 @@ from prowler.providers.azure.lib.service.service import AzureService
class Entra(AzureService):
def __init__(self, provider: AzureProvider):
super().__init__(GraphServiceClient, provider)
self.users = asyncio.get_event_loop().run_until_complete(self.__get_users__())
self.authorization_policy = asyncio.get_event_loop().run_until_complete(
self.__get_authorization_policy__()
)
self.group_settings = asyncio.get_event_loop().run_until_complete(
self.__get_group_settings__()
)
self.security_default = asyncio.get_event_loop().run_until_complete(
self.__get_security_default__()
)
self.named_locations = asyncio.get_event_loop().run_until_complete(
self.__get_named_locations__()
)
self.directory_roles = asyncio.get_event_loop().run_until_complete(
self.__get_directory_roles__()
)
self.conditional_access_policy = asyncio.get_event_loop().run_until_complete(
self.__get_conditional_access_policy__()
loop = get_event_loop()
# Get users first alone because it is a dependency for other attributes
self.users = loop.run_until_complete(self.__get_users__())
attributes = loop.run_until_complete(
gather(
self.__get_authorization_policy__(),
self.__get_group_settings__(),
self.__get_security_default__(),
self.__get_named_locations__(),
self.__get_directory_roles__(),
self.__get_conditional_access_policy__(),
)
)
self.authorization_policy = attributes[0]
self.group_settings = attributes[1]
self.security_default = attributes[2]
self.named_locations = attributes[3]
self.directory_roles = attributes[4]
self.conditional_access_policy = attributes[5]
async def __get_users__(self):
logger.info("Entra - Getting users...")
users = {}