ghndrx/prowler
1
0
Fork 0
mirror of https://github.com/ghndrx/prowler.git synced 2026-02-10 06:45:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(azure_service): New parent class (#2642)

Browse Source
This commit is contained in:
Pepe Fagoaga
2023-07-31 16:03:49 +02:00
committed by GitHub
parent dbbb70027a
commit 3aa1fbced9
5 changed files with 18 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
prowler/providers/azure/lib/service/__init__.py Normal file
View File

8
prowler/providers/azure/lib/service/service.py Normal file
View File

@@ -0,0 +1,8 @@
class AzureService:
def __init__(self, service, audit_info):
# We receive the service using __class__.__name__ or the service name in lowercase
# e.g.: Storage --> we need a lowercase string, so service.lower()
self.service = service.lower() if not service.islower() else service
self.credentials = audit_info.credentials
self.subscriptions = audit_info.identity.subscriptions

8
prowler/providers/azure/services/defender/defender_service.py
View File

@@ -4,14 +4,14 @@ from azure.mgmt.security import SecurityCenter
from pydantic import BaseModel
from prowler.lib.logger import logger
from prowler.providers.azure.lib.service.service import AzureService
########################## Defender
class Defender:
class Defender(AzureService):
def __init__(self, audit_info):
self.service = "defender"
self.credentials = audit_info.credentials
self.subscriptions = audit_info.identity.subscriptions
super().__init__(__class__.__name__, audit_info)
self.clients = self.__set_clients__(
audit_info.identity.subscriptions, audit_info.credentials
)

7
prowler/providers/azure/services/iam/iam_service.py
View File

@@ -4,14 +4,13 @@ from azure.mgmt.authorization import AuthorizationManagementClient
from azure.mgmt.authorization.v2022_04_01.models import Permission
from prowler.lib.logger import logger
from prowler.providers.azure.lib.service.service import AzureService
########################## IAM
class IAM:
class IAM(AzureService):
def __init__(self, audit_info):
self.service = "iam"
self.credentials = audit_info.credentials
self.subscriptions = audit_info.identity.subscriptions
super().__init__(__class__.__name__, audit_info)
self.clients = self.__set_clients__(
audit_info.identity.subscriptions, audit_info.credentials
)

7
prowler/providers/azure/services/storage/storage_service.py
View File

@@ -4,14 +4,13 @@ from azure.mgmt.storage import StorageManagementClient
from azure.mgmt.storage.v2022_09_01.models import NetworkRuleSet
from prowler.lib.logger import logger
from prowler.providers.azure.lib.service.service import AzureService
########################## Storage
class Storage:
class Storage(AzureService):
def __init__(self, audit_info):
self.service = "storage"
self.credentials = audit_info.credentials
self.subscriptions = audit_info.identity.subscriptions
super().__init__(__class__.__name__, audit_info)
self.clients = self.__set_clients__(
audit_info.identity.subscriptions, audit_info.credentials
)