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

chore(service): service class type hints (#2695)

Browse Source
This commit is contained in:
Pepe Fagoaga
2023-08-10 10:01:54 +02:00
committed by GitHub
parent ad3cd66e08
commit 76db92ea14
3 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
prowler/providers/aws/lib/service/service.py
View File

@@ -4,6 +4,7 @@ from prowler.providers.aws.aws_provider import (
generate_regional_clients, generate_regional_clients,
get_default_region, get_default_region,
) )
from prowler.providers.aws.lib.audit_info.models import AWS_Audit_Info
class AWSService: class AWSService:
@@ -14,7 +15,7 @@ class AWSService:
- Also handles if the AWS Service is Global - Also handles if the AWS Service is Global
""" """
def __init__(self, service, audit_info, global_service=False): def __init__(self, service: str, audit_info: AWS_Audit_Info, global_service=False):
# Audit Information # Audit Information
self.audit_info = audit_info self.audit_info = audit_info
self.audited_account = audit_info.audited_account self.audited_account = audit_info.audited_account

5
prowler/providers/azure/lib/service/service.py
View File

@@ -1,11 +1,12 @@
from prowler.lib.logger import logger from prowler.lib.logger import logger
from prowler.providers.azure.lib.audit_info.models import Azure_Audit_Info
class AzureService: class AzureService:
def __init__( def __init__(
self, self,
service, service: str,
audit_info, audit_info: Azure_Audit_Info,
): ):
self.clients = self.__set_clients__( self.clients = self.__set_clients__(
audit_info.identity.subscriptions, audit_info.credentials, service audit_info.identity.subscriptions, audit_info.credentials, service

9
prowler/providers/gcp/lib/service/service.py
View File

@@ -6,10 +6,17 @@ from googleapiclient import discovery
from prowler.lib.logger import logger from prowler.lib.logger import logger
from prowler.providers.gcp.gcp_provider import generate_client from prowler.providers.gcp.gcp_provider import generate_client
from prowler.providers.gcp.lib.audit_info.models import GCP_Audit_Info
class GCPService: class GCPService:
def __init__(self, service, audit_info, region="global", api_version="v1"): def __init__(
self,
service: str,
audit_info: GCP_Audit_Info,
region="global",
api_version="v1",
):
# We receive the service using __class__.__name__ or the service name in lowercase # We receive the service using __class__.__name__ or the service name in lowercase
# e.g.: APIKeys --> we need a lowercase string, so service.lower() # e.g.: APIKeys --> we need a lowercase string, so service.lower()
self.service = service.lower() if not service.islower() else service self.service = service.lower() if not service.islower() else service