def test_correlate_management_grp_data(self):
parquet_file = "TestingCorrelation.parquet"
xlsx_file = "TestingCorrelation.xlsx"
with change_dir(OPERATIONSPATH):
if os.path.isfile(parquet_file):
os.remove(parquet_file)
subscription = os.environ["SUB"]
management_grp = os.environ["MANGRP"]
credentials = AzureConnections().get_authenticated_client()
df = EvaluatePolicies(credentials).correlate_management_grp_data(
management_grp=management_grp, subscription=subscription)
df = df.drop(columns=['serialize', 'timestamp'])
df.to_excel(xlsx_file)
pm = pluggy.PluginManager("Analysis")
pm.add_hookspecs(DatbasePlugins)
pm.register(SQLServerDatabase())
engine_list = pm.hook.get_db_engine(
driver=self.driver,
server=self.server,
database=self.database,
user=self.user,
password=self.password,
)
engine = engine_list[0]
df.to_sql('test_noncompliant_002',
con=engine,
if_exists='append',
chunksize=1000)
def __init__(self, credentials=None):
load_dotenv()
with change_dir(OPERATIONSPATH):
self.config = ConfigParser()
self.config.read(CONFVARIABLES)
self.credentials = AzureConnections().get_authenticated_client()
def test_aggregation(self):
subscription = os.environ["SUB"]
management_grp = os.environ["MANGRP"]
credentials = AzureConnections().get_authenticated_client()
eval_policies = EvaluatePolicies(credentials)
df = eval_policies.process_management_grp_subscriptions(management_grp)
def __init__(self):
load_dotenv()
self.credentials = AzureConnections().get_authenticated_client()
with change_dir(OPERATIONSPATH):
self.config = ConfigParser()
self.config.read(CONFVARIABLES)
self.subscriptions = Subscriptions().list_displayname_and_id()
self.resource_cls = Resource()
def __init__(self):
load_dotenv()
self.credentials = AzureConnections().get_authenticated_client()
self.subscription_client = SubscriptionClient(self.credentials, base_url=None)
with change_dir(OPERATIONSPATH):
self.config = ConfigParser()
self.config.read(CONFVARIABLES)
logging_level = int(self.config['LOGGING']['level'])
logging.basicConfig(level=logging_level)
def test_subscription_list_displayname_id(self):
"""
:return:
"""
credentials = AzureConnections().get_authenticated_client()
results = Subscriptions().list_displayname_and_id()
self.assertGreater(len(results), 0)
for i in results:
logging.debug(i)
def __init__(self, client_id, key, tenant_id, url, cosmos_key):
load_dotenv()
self.url = url
self.__comos_key = cosmos_key
self.cosmos_client = cosmos_client.CosmosClient(url, cosmos_key)
self.credentials = AzureConnections().authenticate_device_code(
CLIENT=client_id, KEY=key, TENANT_ID=tenant_id)
with change_dir(OPERATIONSPATH):
self.config = ConfigParser()
self.config.read(CONFVARIABLES)
def __init__(self):
load_dotenv()
self.credentials = AzureConnections().get_authenticated_client()
# Not the default path is debug. This exists in order to fail safe. Production must be explicit.
if 'mop_debug' is os.environ and os.environ['mop_debug'].lower() in 'false':
config_file = CONFVARIABLES
else:
config_file = TESTVARIABLES
with change_dir(OPERATIONSPATH):
self.config = ConfigParser()
self.config.read(config_file)
logging_level = int(self.config['LOGGING']['level'])
logging.basicConfig(level=logging_level)