cloud-governance 1.1.385__tar.gz → 1.1.386__tar.gz

{cloud_governance-1.1.385/cloud_governance.egg-info → cloud_governance-1.1.386}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cloud-governance
-Version: 1.1.385
+Version: 1.1.386
 Summary: Cloud Governance Tool
 Home-page: https://github.com/redhat-performance/cloud-governance
 Author: Red Hat

{cloud_governance-1.1.385 → cloud_governance-1.1.386}/cloud_governance/common/clouds/azure/subscriptions/azure_operations.py

@@ -24,6 +24,7 @@ class AzureOperations:
         self.billing_client = BillingManagementClient(credential=self.__default_creds,
                                                       subscription_id=self.subscription_id)
         self.__account_id = self.__environment_variables_dict.get('AZURE_ACCOUNT_ID')
+        self.global_tags = self.__environment_variables_dict.get('GLOBAL_TAGS')
         self.cloud_name = 'AZURE'
         self.scope = f'subscriptions/{self.subscription_id}'
 

{cloud_governance-1.1.385 → cloud_governance-1.1.386}/cloud_governance/main/environment_variables.py

@@ -70,7 +70,7 @@ class EnvironmentVariables:
         self._environment_variables_dict['account'] = EnvironmentVariables.get_env('account', '').upper().strip()
         self._environment_variables_dict['AWS_DEFAULT_REGION'] = EnvironmentVariables.get_env('AWS_DEFAULT_REGION', '')
         self._environment_variables_dict['log_level'] = EnvironmentVariables.get_env('log_level', 'INFO')
-
+        self._environment_variables_dict['GLOBAL_TAGS'] = literal_eval(EnvironmentVariables.get_env('GLOBAL_TAGS', "{}"))
         self._environment_variables_dict['DAYS_TO_TAKE_ACTION'] = int(
             EnvironmentVariables.get_env('DAYS_TO_TAKE_ACTION', "7"))
         if not hasattr(self, 'POLICIES_LIST'):
@@ -108,6 +108,7 @@ class EnvironmentVariables:
                                                              'cost_explorer_payer_billings', 'spot_savings_analysis']
         self._environment_variables_dict['ibm_policies'] = ['tag_baremetal', 'tag_vm', 'ibm_cost_report',
                                                             'ibm_cost_over_usage']
+        self._environment_variables_dict['azure_policies'] = ['tag_azure_resource_group']
         if self._environment_variables_dict['policy'] in self._environment_variables_dict['cost_policies']:
             es_index = 'cloud-governance-global-cost-billing-index'
 
@@ -150,10 +151,9 @@ class EnvironmentVariables:
         self._environment_variables_dict['AZURE_ACCOUNT_ID'] = EnvironmentVariables.get_env('AZURE_ACCOUNT_ID', '')
         self._environment_variables_dict['AZURE_CLIENT_ID'] = EnvironmentVariables.get_env('AZURE_CLIENT_ID', '')
         self._environment_variables_dict['AZURE_TENANT_ID'] = EnvironmentVariables.get_env('AZURE_TENANT_ID', '')
-        self._environment_variables_dict['AZURE_CLIENT_SECRET'] = EnvironmentVariables.get_env('AZURE_CLIENT_SECRET',
-                                                                                               '')
-        self._environment_variables_dict['AZURE_SUBSCRIPTION_ID'] = EnvironmentVariables.get_env(
-            'AZURE_SUBSCRIPTION_ID', '')
+        self._environment_variables_dict['AZURE_CLIENT_SECRET'] = EnvironmentVariables.get_env('AZURE_CLIENT_SECRET', '')
+        self._environment_variables_dict['AZURE_SUBSCRIPTION_ID'] = EnvironmentVariables.get_env('AZURE_SUBSCRIPTION_ID', '')
+        self._environment_variables_dict['AZURE_SUBSCRIPTION_ID'] = EnvironmentVariables.get_env('AZURE_SUBSCRIPTION_ID', '')
         if self._environment_variables_dict['AZURE_CLIENT_ID'] and self._environment_variables_dict['AZURE_TENANT_ID'] \
                 and self._environment_variables_dict['AZURE_CLIENT_SECRET']:
             self._environment_variables_dict['PUBLIC_CLOUD_NAME'] = 'AZURE'
@@ -411,7 +411,6 @@ class EnvironmentVariables:
 
 
 environment_variables = EnvironmentVariables()
-
 # env vars examples
 # os.environ['AWS_DEFAULT_REGION'] = 'us-east-2'
 # os.environ['AWS_DEFAULT_REGION'] = 'all'

{cloud_governance-1.1.385 → cloud_governance-1.1.386}/cloud_governance/main/main.py

@@ -254,7 +254,7 @@ def main():
                 if environment_variables_dict.get('PUBLIC_CLOUD_NAME') and environment_variables_dict.get(
                         'PUBLIC_CLOUD_NAME').upper() == 'AZURE':
                     azure_cost_policy_runner = None
-                    is_azure_policy_runner = policy in environment_variables_dict.get('cost_policies')
+                    is_azure_policy_runner = policy in environment_variables_dict.get('cost_policies') or policy in environment_variables_dict.get('azure_policies')
                     if is_azure_policy_runner:
                         azure_cost_policy_runner = AzurePolicyRunner()
 

cloud_governance-1.1.386/cloud_governance/policy/azure/tag_azure_resource_group.py

@@ -0,0 +1,84 @@
+from azure.mgmt.resource import ResourceManagementClient
+from cloud_governance.common.logger.init_logger import logger
+from cloud_governance.common.clouds.azure.subscriptions.azure_operations import AzureOperations
+
+
+class TagAzureResourceGroup:
+    def __init__(self):
+        """
+        Initialize Azure clients and set the tags to apply.
+        """
+        self.__azure_operations = AzureOperations()
+        self.__subscription_id = self.__azure_operations.subscription_id
+        self.__credential = self.__azure_operations._AzureOperations__default_creds  # reuse credentials
+        self.resource_client = ResourceManagementClient(self.__credential, self.__subscription_id)
+        self.__tags_to_add = self.__azure_operations.global_tags
+        if not self.__tags_to_add:
+            raise ValueError("No tags provided to add. Please set GLOBAL_TAGS environment variable.")
+
+    def tag_all(self):
+        """
+        Tag all resource groups and their contained resources with the specified tags.
+        """
+        resource_groups = self.resource_client.resource_groups.list()
+
+        for rg in resource_groups:
+            logger.info(f"Processing resource group: {rg.name}")
+            # Tag the resource group
+            try:
+                self.resource_client.resource_groups.update(
+                    resource_group_name=rg.name,
+                    parameters={"tags": self.__tags_to_add}
+                )
+                logger.info(f"Tagged resource group: {rg.name}")
+            except Exception as e:
+                logger.error(f"Failed to tag resource group {rg.name}: {e}")
+                continue
+
+            # Tag all resources in the resource group
+            for resource in self.resource_client.resources.list_by_resource_group(rg.name):
+                logger.info(f"Tagging resource: {resource.name} ({resource.type})")
+
+                existing_tags = resource.tags or {}
+                updated_tags = {**existing_tags, **self.__tags_to_add}
+
+                try:
+                    parts = resource.type.split('/')
+                    provider_ns = parts[0]
+                    resource_type_str = '/'.join(parts[1:])
+
+                    provider = self.resource_client.providers.get(provider_ns)
+
+                    # Some resource_type may not match exactly, use a relaxed matching
+                    resource_type_info = next(
+                        (rt for rt in provider.resource_types if rt.resource_type.lower() == resource_type_str.lower()),
+                        None
+                    )
+
+                    if not resource_type_info:
+                        logger.warning(f"Could not find exact match for resource type: {resource_type_str} in provider {provider_ns}")
+                        continue
+
+                    api_version = next(
+                        (v for v in resource_type_info.api_versions if 'preview' not in v.lower()),
+                        resource_type_info.api_versions[0]
+                    )
+
+                    poller = self.resource_client.resources.begin_update_by_id(
+                        resource_id=resource.id,
+                        api_version=api_version,
+                        parameters={
+                            'location': resource.location,
+                            'tags': updated_tags
+                        }
+                    )
+                    poller.result()
+                    logger.info(f"Tagged resource: {resource.name}")
+                except Exception as e:
+                    logger.error(f"Failed to tag resource {resource.name}: {e}")
+
+    def run(self):
+        """
+        This method tags all Azure resources.
+        """
+        self.tag_all()

{cloud_governance-1.1.385 → cloud_governance-1.1.386/cloud_governance.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cloud-governance
-Version: 1.1.385
+Version: 1.1.386
 Summary: Cloud Governance Tool
 Home-page: https://github.com/redhat-performance/cloud-governance
 Author: Red Hat

{cloud_governance-1.1.385 → cloud_governance-1.1.386}/cloud_governance.egg-info/SOURCES.txt

@@ -190,6 +190,7 @@ cloud_governance/policy/aws/monitor/__init__.py
 cloud_governance/policy/aws/monitor/cluster_run.py
 cloud_governance/policy/azure/__init__.py
 cloud_governance/policy/azure/cost_billing_reports.py
+cloud_governance/policy/azure/tag_azure_resource_group.py
 cloud_governance/policy/azure/cleanup/__init__.py
 cloud_governance/policy/azure/cleanup/instance_idle.py
 cloud_governance/policy/azure/cleanup/instance_run.py

{cloud_governance-1.1.385 → cloud_governance-1.1.386}/setup.py

@@ -2,7 +2,7 @@ from codecs import open
 from os import path
 from setuptools import setup, find_packages
 
-__version__ = '1.1.385'
+__version__ = '1.1.386'
 
 here = path.abspath(path.dirname(__file__))