DEVOPS-271 azure policy exemption automation

devwithkrishna · Aug 5, 2024 · f469472 · f469472
1 parent 0fd80c0
commit f469472
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/common_utils.py b/common_utils.py
@@ -8,9 +8,9 @@ def calculate_expiry(expires_after: int, unit: str) -> str:
     now = datetime.now(pytz.utc)  # Get the current time in UTC
 
     if unit == 'day':
-        expiry_date = now + timedelta(days=expires_after)
+        expiry_date = now + timedelta(days=float(expires_after))
     elif unit == 'month':
-        expiry_date = now + relativedelta(months=expires_after)
+        expiry_date = now + relativedelta(months=int(expires_after))
     elif unit == 'hour':
         expiry_date = now + timedelta(hours=float(expires_after))
     else:

diff --git a/policy_exception.py b/policy_exception.py
@@ -95,11 +95,11 @@ def create_exemption_for_policy(subscription_id: str, policy_name:str, expires_a
 				expires_on = expiry_date,
 				display_name = policy_exemption_name,
 				description=policy_exemption_description
-
 			)
 
 		exemption = client.policy_exemptions.create_or_update(scope=scope,policy_exemption_name=policy_exemption_name, parameters=parameters)
 		print("Policy exemption created or updated successfully.")
+		print(f'Policy Exemption will expire at {expiry_date}')
 
 	except HttpResponseError as err:
 		# Handle HTTP response errors