Skip to content

Commit 0db6327

Browse files
committed
PRMP-643 Refactor document services to use cached environment variables for table names and S3 buckets
1 parent db76083 commit 0db6327

File tree

2 files changed

+10
-4
lines changed

2 files changed

+10
-4
lines changed

lambdas/services/document_service.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,16 +21,18 @@ class DocumentService:
2121
def __init__(self):
2222
self.s3_service = S3Service()
2323
self.dynamo_service = DynamoDBService()
24+
self._lg_table_name = os.getenv("LLOYD_GEORGE_DYNAMODB_NAME")
25+
self._lg_s3_bucket = os.getenv("LLOYD_GEORGE_BUCKET_NAME")
2426

2527
@property
2628
def table_name(self) -> str:
2729
"""DynamoDB table name. Can be overridden by child classes."""
28-
return os.getenv("LLOYD_GEORGE_DYNAMODB_NAME")
30+
return self._lg_table_name
2931

3032
@property
3133
def s3_bucket(self) -> str:
3234
"""S3 bucket name. Can be overridden by child classes."""
33-
return os.getenv("LLOYD_GEORGE_BUCKET_NAME")
35+
return self._lg_s3_bucket
3436

3537
@property
3638
def model_class(self) -> type[BaseModel]:

lambdas/services/document_upload_review_service.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,18 +9,22 @@
99

1010
class DocumentUploadReviewService(DocumentService):
1111
"""Service for handling DocumentUploadReviewReference operations."""
12+
def __init__(self):
13+
super().__init__()
14+
self._table_name = os.environ.get("DOCUMENT_REVIEW_DYNAMODB_NAME")
15+
self._s3_bucket = os.environ.get("DOCUMENT_REVIEW_S3_BUCKET_NAME")
1216

1317
@property
1418
def table_name(self) -> str:
15-
return os.environ.get("DOCUMENT_REVIEW_DYNAMODB_NAME")
19+
return self._table_name
1620

1721
@property
1822
def model_class(self) -> type:
1923
return DocumentUploadReviewReference
2024

2125
@property
2226
def s3_bucket(self) -> str:
23-
return os.environ.get("DOCUMENT_REVIEW_S3_BUCKET_NAME")
27+
return self._s3_bucket
2428

2529
def update_document_review_custodian(
2630
self,

0 commit comments

Comments
 (0)