cloud-governance 1.1.354__py3-none-any.whl → 1.1.356__py3-none-any.whl

cloud_governance/policy/aws/ec2_stop.py

@@ -3,7 +3,8 @@ import operator
 
 from cloud_governance.common.clouds.aws.cloudtrail.cloudtrail_operations import CloudTrailOperations
 from cloud_governance.common.logger.init_logger import logger
-from cloud_governance.policy.policy_operations.aws.zombie_non_cluster.run_zombie_non_cluster_policies import NonClusterZombiePolicy
+from cloud_governance.policy.policy_operations.aws.zombie_non_cluster.run_zombie_non_cluster_policies import \
+    NonClusterZombiePolicy
 from operator import ge
 
 
@@ -28,14 +29,17 @@ class EC2Stop(NonClusterZombiePolicy):
         This method list all stopped instances for more than 30 days and terminate if dry_run no
         @return:
         """
-        return self.__fetch_stop_instance(sign=ge, instance_days=self.FIRST_MAIL_NOTIFICATION_INSTANCE_DAYS, delete_instance_days=self.DELETE_INSTANCE_DAYS)
+        return self.__fetch_stop_instance(sign=ge, instance_days=self.FIRST_MAIL_NOTIFICATION_INSTANCE_DAYS,
+                                          delete_instance_days=self.DELETE_INSTANCE_DAYS)
 
     def __fetch_stop_instance(self, instance_days: int, delete_instance_days: int, sign: operator = ge):
         """
         This method list all stopped instances for more than 30 days and terminate if dry_run no
         @return:
         """
-        instances = self._ec2_client.describe_instances(Filters=[{'Name': 'instance-state-name', 'Values': ['stopped']}])['Reservations']
+        instances = \
+            self._ec2_client.describe_instances(Filters=[{'Name': 'instance-state-name', 'Values': ['stopped']}])[
+                'Reservations']
         stopped_instances = []
         stopped_instance_tags = {}
         ec2_types = {}
@@ -51,20 +55,30 @@ class EC2Stop(NonClusterZombiePolicy):
                         stopped_time = datetime.datetime.now()
                     days = self._calculate_days(create_date=stopped_time)
                     user = self._get_tag_name_from_tags(tags=resource.get('Tags'), tag_name='User')
-                    stop_cost = self.get_ebs_cost(resource=resource.get('BlockDeviceMappings'), resource_type='ec2', resource_hours=(self.DAILY_HOURS * days))
+                    stop_cost = self.get_ebs_cost(resource=resource.get('BlockDeviceMappings'), resource_type='ec2',
+                                                  resource_hours=(self.DAILY_HOURS * days))
                     if days in (instance_days, self.SECOND_MAIL_NOTIFICATION_INSTANCE_DAYS):
                         if days == self.SECOND_MAIL_NOTIFICATION_INSTANCE_DAYS:
-                            delta_cost = self.get_ebs_cost(resource=resource.get('BlockDeviceMappings'), resource_type='ec2', resource_hours=(self.DAILY_HOURS * (days-self.FIRST_MAIL_NOTIFICATION_INSTANCE_DAYS)))
+                            delta_cost = self.get_ebs_cost(resource=resource.get('BlockDeviceMappings'),
+                                                           resource_type='ec2', resource_hours=(
+                                        self.DAILY_HOURS * (days - self.FIRST_MAIL_NOTIFICATION_INSTANCE_DAYS)))
                         else:
                             delta_cost = stop_cost
                         if user:
-                            self.__trigger_mail(tags=resource.get('Tags'), stopped_time=stopped_time, resource_id=instance_id, days=days, ec2_type=resource.get("InstanceType"), instance_id=instance_id, message_type='notification', stop_cost=stop_cost, delta_cost=delta_cost)
+                            self.__trigger_mail(tags=resource.get('Tags'), stopped_time=stopped_time,
+                                                resource_id=instance_id, days=days,
+                                                ec2_type=resource.get("InstanceType"), instance_id=instance_id,
+                                                message_type='notification', stop_cost=stop_cost, delta_cost=delta_cost)
                         else:
                             logger.info('User is missing')
                     if days == self.DAYS_TO_NOTIFY_ADMINS:
-                        delta_charge = self.get_ebs_cost(resource=resource.get('BlockDeviceMappings'), resource_type='ec2', resource_hours=(self.DAILY_HOURS * (self.DAYS_TO_NOTIFY_ADMINS-self.SECOND_MAIL_NOTIFICATION_INSTANCE_DAYS)))
-                        self.__trigger_mail(tags=resource.get('Tags'), stopped_time=stopped_time, resource_id=instance_id,
-                                            days=days, ec2_type=resource.get("InstanceType"), instance_id=instance_id, admins=self._admins,
+                        delta_charge = self.get_ebs_cost(resource=resource.get('BlockDeviceMappings'),
+                                                         resource_type='ec2', resource_hours=(self.DAILY_HOURS * (
+                                    self.DAYS_TO_NOTIFY_ADMINS - self.SECOND_MAIL_NOTIFICATION_INSTANCE_DAYS)))
+                        self.__trigger_mail(tags=resource.get('Tags'), stopped_time=stopped_time,
+                                            resource_id=instance_id,
+                                            days=days, ec2_type=resource.get("InstanceType"), instance_id=instance_id,
+                                            admins=self._admins,
                                             message_type='notify-admin', stop_cost=stop_cost, delta_charge=delta_charge)
                     if sign(days, instance_days):
                         if days >= delete_instance_days:
@@ -78,7 +92,7 @@ class EC2Stop(NonClusterZombiePolicy):
                             'StoppedDate': str(resource.get('UsageOperationUpdateTime')),
                             'Name': self._get_tag_name_from_tags(tags=resource.get('Tags'), tag_name='Name'),
                             'User': self._get_tag_name_from_tags(tags=resource.get('Tags'), tag_name='User'),
-                            'LaunchTime': str(resource.get('LaunchTime')),
+                            'ResourceLaunchTime': str(resource.get('LaunchTime')),
                             'Policy': self._get_tag_name_from_tags(tags=resource.get('Tags'), tag_name='Policy')})
         if self._dry_run == "no":
             for instance_id, tags in stopped_instance_tags.items():
@@ -99,7 +113,8 @@ class EC2Stop(NonClusterZombiePolicy):
                     #     logger.info(err)
         return stopped_instances
 
-    def __trigger_mail(self, tags: list, stopped_time: str, resource_id: str, days: int, image_id: str = '', ec2_type: str = '', instance_id: str = '', **kwargs):
+    def __trigger_mail(self, tags: list, stopped_time: str, resource_id: str, days: int, image_id: str = '',
+                       ec2_type: str = '', instance_id: str = '', **kwargs):
         """
         This method send triggering mail
         @param tags:
@@ -110,16 +125,25 @@ class EC2Stop(NonClusterZombiePolicy):
         """
         try:
             special_user_mails = self._literal_eval(self._special_user_mails)
-            user, instance_name = self._get_tag_name_from_tags(tags=tags, tag_name='User'), self._get_tag_name_from_tags(tags=tags, tag_name='Name')
+            user, instance_name = self._get_tag_name_from_tags(tags=tags,
+                                                               tag_name='User'), self._get_tag_name_from_tags(tags=tags,
+                                                                                                              tag_name='Name')
             to = user if user not in special_user_mails else special_user_mails[user]
             ldap_data = self._ldap.get_user_details(user_name=to)
             cc = []
-            subject, body = self._mail_description.ec2_stop(name=ldap_data.get('displayName'), days=days, image_id=image_id, delete_instance_days=self.DELETE_INSTANCE_DAYS, instance_name=instance_name, resource_id=resource_id, stopped_time=stopped_time, ec2_type=ec2_type, extra_purse=kwargs.get('stop_cost'))
+            subject, body = self._mail_description.ec2_stop(name=ldap_data.get('displayName'), days=days,
+                                                            image_id=image_id,
+                                                            delete_instance_days=self.DELETE_INSTANCE_DAYS,
+                                                            instance_name=instance_name, resource_id=resource_id,
+                                                            stopped_time=stopped_time, ec2_type=ec2_type,
+                                                            extra_purse=kwargs.get('stop_cost'))
             if not kwargs.get('admins'):
                 kwargs['admins'] = to
                 cc = [self._account_admin, f'{ldap_data.get("managerId")}@redhat.com']
             else:
                 kwargs['admins'].append(f'{ldap_data.get("managerId")}@redhat.com')
-            self._mail.send_email_postfix(to=kwargs.get('admins'), content=body, subject=subject, cc=cc, resource_id=instance_id, message_type=kwargs.get('message_type'), extra_purse=kwargs.get('delta_cost', 0))
+            self._mail.send_email_postfix(to=kwargs.get('admins'), content=body, subject=subject, cc=cc,
+                                          resource_id=instance_id, message_type=kwargs.get('message_type'),
+                                          extra_purse=kwargs.get('delta_cost', 0))
         except Exception as err:
             logger.info(err)

cloud_governance/policy/ibm/ibm_cost_report.py

@@ -51,6 +51,8 @@ class IBMCostReport:
             hardware_tags = self.collect_tags_from_machines(
                 tags=self.classic_operations.get_hardware_tags(hardware_id=str(hardware.get('id'))))
             hardware_tags['fqdn'] = hardware.get('fullyQualifiedDomainName').lower()
+            if 'budget' not in hardware_tags:
+                hardware_tags['budget'] = self.__environment_variables_dict.get('account')
             collect_machines_data[hardware_tags['fqdn']] = hardware_tags
         return collect_machines_data
 
@@ -86,7 +88,8 @@ class IBMCostReport:
             if fqdn not in collect_machines_data:
                 user = [user for user in users if user in fqdn]
                 if user:
-                    user_tags = self.ibm_account.get_user_tags_from_gsheet(username=f'{user[0]}@redhat.com', user_email='yes')
+                    user_tags = self.ibm_account.get_user_tags_from_gsheet(username=f'{user[0]}@redhat.com',
+                                                                           user_email='yes')
                     invoice_data[fqdn].update(self.collect_tags_from_machines(user_tags))
         return invoice_data
 
@@ -112,15 +115,17 @@ class IBMCostReport:
         collect_machines_data = self.get_hardware_data()
         collect_machines_data.update(self.get_virtual_machine_data())
         invoice_data = self.get_invoice_data(collect_machines_data=collect_machines_data)
-        cost_invoice_resource_data = self.concatenate_dictionaries(resource_data=collect_machines_data, invoice_data=invoice_data)
+        cost_invoice_resource_data = self.concatenate_dictionaries(resource_data=collect_machines_data,
+                                                                   invoice_data=invoice_data)
         for tag_name in self.owned_tags:
             cost_list_items = []
             for _, data in cost_invoice_resource_data.items():
-                if data[tag_name] != 0 or tag_name == 'budget':
+                if data.get(tag_name) != 0 or tag_name == 'budget':
                     cost_list_items.append({
                         tag_name.capitalize(): data[tag_name],
                         'Cost': data['cost'],
                         'Budget': self._elastic_upload.account
                     })
-            self._elastic_upload.es_upload_data(items=cost_list_items, es_index=f'{self._elastic_upload.es_index}-{tag_name.lower()}')
+            self._elastic_upload.es_upload_data(items=cost_list_items,
+                                                es_index=f'{self._elastic_upload.es_index}-{tag_name.lower()}')
         return True

{cloud_governance-1.1.354.dist-info → cloud_governance-1.1.356.dist-info}/METADATA

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

{cloud_governance-1.1.354.dist-info → cloud_governance-1.1.356.dist-info}/RECORD

@@ -158,7 +158,7 @@ cloud_governance/policy/aws/cost_explorer.py,sha256=45GpfojLXXwwYSXREhPOA1ATgy1E
 cloud_governance/policy/aws/cost_explorer_payer_billings.py,sha256=EuTry807RRw0SXNkyCmbK9CGuO1zT7fvv6odIy9338k,18731
 cloud_governance/policy/aws/cost_over_usage.py,sha256=Jhv9VNpgIK1cEGJS6CFitkxwM3o8AUf98N1sRTUL09w,5351
 cloud_governance/policy/aws/ebs_in_use.py,sha256=7vGV2qobV14rzC7tJnJiafC3oAhnY_TTvd53oEtaN4Q,616
-cloud_governance/policy/aws/ec2_stop.py,sha256=21Ju0XZt67A-u_nTUXQjgy_8wUIvZ6N5OUkKS8oP810,8494
+cloud_governance/policy/aws/ec2_stop.py,sha256=FXQNkHmiMCEw6Pz6CalIIVYFRsV24RT5TewqFazyP8M,9641
 cloud_governance/policy/aws/empty_roles.py,sha256=EhsepFRbjU82PjJQ1hADqBM_RO1Ukr-Q4DZAz3bM4ho,3155
 cloud_governance/policy/aws/ip_unattached.py,sha256=OnHlTIKSgyxwdAgU9jBApdvDQpzHgECjSodN6KoXWqU,3176
 cloud_governance/policy/aws/monthly_report.py,sha256=jxdObuFEWPobRFtVKgo-sMT4gmMC_KS9gkPVF8ZqjpY,6722
@@ -200,7 +200,7 @@ cloud_governance/policy/ibm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMp
 cloud_governance/policy/ibm/cost_billing_reports.py,sha256=3qQlgugtBQGw_W3PhrsQCehi4r4o1QPr6G50ZWGATnE,4118
 cloud_governance/policy/ibm/cost_usage_reports.py,sha256=UoiGugo9wCtoEKBWd-a4hbkbFhfWOf5BgM8Z6VkWnNU,2298
 cloud_governance/policy/ibm/ibm_cost_over_usage.py,sha256=C9ZkpMjRao9FQ2iZSXFdXqGxctBZVoLe-JXEupqIJ10,3595
-cloud_governance/policy/ibm/ibm_cost_report.py,sha256=3DbC_Os_vy-V4dD7i63JvWvd25dXgZa7oJHQIHETrec,5385
+cloud_governance/policy/ibm/ibm_cost_report.py,sha256=xv-TX4eDgnTt5EM_iCLrUwj5hXRmHrsCfE6bs-bSlIo,5716
 cloud_governance/policy/ibm/tag_baremetal.py,sha256=7vVh3t4mqs_RDEzSwUS1E2xQzHnwJYPs3lGn5V7d3RU,5329
 cloud_governance/policy/ibm/tag_resources.py,sha256=g0ypZuKsQrJxscWDmcCdXHm9AS2fpMWsqpMCorF_Tdc,6324
 cloud_governance/policy/ibm/tag_vm.py,sha256=G3QLYwuB2cWSHAsoR40BHVBV5nmGVcVzLtJLmM3MenA,4836
@@ -263,8 +263,8 @@ cloud_governance/policy/policy_runners/elasticsearch/__init__.py,sha256=47DEQpj8
 cloud_governance/policy/policy_runners/elasticsearch/upload_elastic_search.py,sha256=pOwUJWXjJbyTy8iv3Ap8xJGnqQe-5lZgoR8-vGfAVos,1881
 cloud_governance/policy/policy_runners/ibm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cloud_governance/policy/policy_runners/ibm/policy_runner.py,sha256=V0E_f7F3hXit0aSq4BlfX1Jd4vjR2NEvOWsJ5upvZ4o,1302
-cloud_governance-1.1.354.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-cloud_governance-1.1.354.dist-info/METADATA,sha256=nG7sgx5aOcgWvLHt4NVvtPRZyARdCfXX97SJ81k4KfU,11364
-cloud_governance-1.1.354.dist-info/WHEEL,sha256=tTnHoFhvKQHCh4jz3yCn0WPTYIy7wXx3CJtJ7SJGV7c,91
-cloud_governance-1.1.354.dist-info/top_level.txt,sha256=jfB1fgj7jvx3YZkZA4G6hFeS1RHO7J7XtnbjuMNMRww,17
-cloud_governance-1.1.354.dist-info/RECORD,,
+cloud_governance-1.1.356.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+cloud_governance-1.1.356.dist-info/METADATA,sha256=mNvoFNdQ_M3Q5wLHAUhE4wmpfqZ_voKMcz_0AeMkiJw,11364
+cloud_governance-1.1.356.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+cloud_governance-1.1.356.dist-info/top_level.txt,sha256=jfB1fgj7jvx3YZkZA4G6hFeS1RHO7J7XtnbjuMNMRww,17
+cloud_governance-1.1.356.dist-info/RECORD,,

{cloud_governance-1.1.354.dist-info → cloud_governance-1.1.356.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (77.0.1)
+Generator: setuptools (78.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any