File tree Expand file tree Collapse file tree 2 files changed +10
-4
lines changed
Expand file tree Collapse file tree 2 files changed +10
-4
lines changed Original file line number Diff line number Diff 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 ]:
Original file line number Diff line number Diff line change 99
1010class 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 ,
You can’t perform that action at this time.
0 commit comments