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
2 changes: 2 additions & 0 deletions .github/workflows/on_merged_pr.yml
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,8 @@ on:
- completed

permissions:
actions: read
contents: read
issues: write

jobs:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,20 +2,18 @@
# SPDX-License-Identifier: MIT-0

import os
from datetime import datetime

import boto3
from aws_lambda_powertools.logging import Logger
from aws_lambda_powertools.metrics import Metrics
from aws_lambda_powertools.tracing import Tracer
from aws_lambda_powertools.event_handler.exceptions import InternalServerError
from schema.unicorn_contracts.contractstatuschanged import AWSEvent, ContractStatusChanged, Marshaller

# Initialise Environment variables
if (SERVICE_NAMESPACE := os.environ.get("SERVICE_NAMESPACE")) is None:
raise InternalServerError("SERVICE_NAMESPACE environment variable is undefined")
raise EnvironmentError("SERVICE_NAMESPACE environment variable is undefined")
if (CONTRACT_STATUS_TABLE := os.environ.get("CONTRACT_STATUS_TABLE")) is None:
raise InternalServerError("CONTRACT_STATUS_TABLE environment variable is undefined")
raise EnvironmentError("CONTRACT_STATUS_TABLE environment variable is undefined")

# Initialise PowerTools
logger: Logger = Logger()
Expand All @@ -26,10 +24,6 @@
dynamodb = boto3.resource("dynamodb")
table = dynamodb.Table(CONTRACT_STATUS_TABLE) # type: ignore

# Get current date
now = datetime.now()
current_date = now.strftime("%d/%m/%Y %H:%M:%S")


@logger.inject_lambda_context(log_event=True) # type: ignore
@metrics.log_metrics(capture_cold_start_metric=True) # type: ignore
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,14 +3,14 @@

import os
import uuid
from datetime import datetime
from datetime import datetime, timezone

import boto3
from boto3.dynamodb.conditions import Attr
from botocore.exceptions import ClientError

from aws_lambda_powertools.logging import Logger
from aws_lambda_powertools.metrics import Metrics
from aws_lambda_powertools.metrics import Metrics, MetricUnit
from aws_lambda_powertools.tracing import Tracer
from aws_lambda_powertools.utilities.data_classes import event_source, SQSEvent
from aws_lambda_powertools.utilities.typing import LambdaContext
Expand Down Expand Up @@ -76,7 +76,7 @@ def create_contract(event: dict) -> None:
DynamoDB put Item response
"""

current_date = datetime.now().strftime("%d/%m/%Y %H:%M:%S")
current_date = datetime.now(timezone.utc).isoformat()
contract = {
"property_id": event["property_id"], # PK
"address": event["address"],
Expand Down Expand Up @@ -105,6 +105,7 @@ def create_contract(event: dict) -> None:

# Annotate trace with contract status
tracer.put_annotation(key="ContractStatus", value=contract["contract_status"])
metrics.add_metric(name="ContractCreated", unit=MetricUnit.Count, value=1)

except ClientError as e:
code = e.response["Error"]["Code"]
Expand Down Expand Up @@ -147,13 +148,13 @@ def update_contract(contract: dict) -> None:

try:
contract["contract_status"] = ContractStatus.APPROVED.name
current_date = datetime.now().strftime("%d/%m/%Y %H:%M:%S")
current_date = datetime.now(timezone.utc).isoformat()

response = table.update_item(
Key={
"property_id": contract["property_id"],
},
UpdateExpression="set contract_status=:t, modified_date=:m",
UpdateExpression="set contract_status=:t, contract_last_modified_on=:m",
ConditionExpression=Attr("property_id").exists()
& Attr("contract_status").is_in([ContractStatus.DRAFT.name]),
ExpressionAttributeValues={
Expand Down
10 changes: 5 additions & 5 deletions unicorn_contracts/uv.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Original file line number Diff line number Diff line change
Expand Up @@ -74,14 +74,18 @@ def publication_approved(event_detail, errors):

property_id = event_detail.property_id
evaluation_result = event_detail.evaluation_result

valid_results = {"APPROVED", "DECLINED"}
if evaluation_result.upper() not in valid_results:
logger.warning(f"Unknown evaluation_result '{evaluation_result}'; skipping DynamoDB update")
return {"result": "Skipped — unknown evaluation result"}

country, city, street, number = property_id.split("/")

pk_details = f"{country}#{city}".replace(" ", "-").lower()
pk = f"PROPERTY#{pk_details}"
sk = f"{street}#{str(number)}".replace(" ", "-").lower()

metrics.add_metric(name="PropertiesAdded", unit=MetricUnit.Count, value=1)

logger.info(f"Storing new property in DynamoDB with PK {pk} and SK {sk}")
dynamodb_response = table.update_item(
Key={
Expand Down
Loading