Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -158,8 +158,8 @@ def _put_policy_on_role(
resource: IAMServiceResource = target_account_session.resource("iam")
role = resource.Role(role_name)
role.attach_policy(PolicyArn=policy_arn)
timeout = datetime.utcnow() + timedelta(minutes=timeout_in_mins)
while datetime.utcnow() < timeout:
timeout = datetime.now(datetime.UTC) + timedelta(minutes=timeout_in_mins)
while datetime.now(datetime.UTC) < timeout:
time.sleep(delay)
if self.role_policy_is_attached(
role_name=role_name,
Expand Down Expand Up @@ -187,8 +187,8 @@ def role_policy_is_attached(
def _ensure_role_can_be_assumed(
self, role_name: str, timeout_in_mins: int = 1, delay: int = 5
) -> None:
timeout = datetime.utcnow() + timedelta(minutes=timeout_in_mins)
while datetime.utcnow() < timeout:
timeout = datetime.now(datetime.UTC) + timedelta(minutes=timeout_in_mins)
while datetime.now(datetime.UTC) < timeout:
if self._can_assume_role(role_name=role_name):
return None
time.sleep(delay)
Expand Down
2 changes: 1 addition & 1 deletion sources/aft-lambda-layer/aft_common/metrics.py
Original file line number Diff line number Diff line change
Expand Up @@ -82,7 +82,7 @@ def wrap_event_for_api(
) -> MetricsPayloadType:
payload: MetricsPayloadType = {
"Solution": self.solution_id,
"TimeStamp": datetime.utcnow().isoformat(timespec="seconds"),
"TimeStamp": datetime.now(datetime.UTC).isoformat(timespec="seconds"),
"Version": None,
"UUID": None,
"Data": {},
Expand Down
4 changes: 2 additions & 2 deletions src/aft_lambda/aft_builder/codebuild_trigger.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,8 +33,8 @@ def lambda_handler(event: Dict[str, Any], context: Dict[str, Any]) -> LayerBuild
time.sleep(30)

# 15min Lambda hard-timeout, soft-timeout at 14min
end_time = datetime.datetime.utcnow() + datetime.timedelta(minutes=14)
while datetime.datetime.utcnow() <= end_time:
end_time = datetime.datetime.now(datetime.UTC) + datetime.timedelta(minutes=14)
while datetime.datetime.now(datetime.UTC) <= end_time:
# We pass exactly 1 job ID, so the build list should contain exactly 1 object
job_status = client.batch_get_builds(ids=[job_id])["builds"][0][
"buildStatus"
Expand Down