Coverage for polar/file/s3.py: 100%

1from polar.config import settings 1 ctx1a

2from polar.integrations.aws.s3 import S3Service 1 ctx1a

3from polar.models.file import FileServiceTypes 1 ctx1a

4 

5 

6def _get_s3_service(bucket: str) -> S3Service: 1 ctx1a

7 return S3Service( 1 ctx1a

8 bucket=bucket, 

9 presign_ttl=settings.S3_FILES_PRESIGN_TTL, 

10 ) 

11 

12 

13S3_SERVICES: dict[FileServiceTypes, S3Service] = { 1 ctx1a

14 FileServiceTypes.downloadable: _get_s3_service(settings.S3_FILES_BUCKET_NAME), 

15 FileServiceTypes.product_media: _get_s3_service( 

16 settings.S3_FILES_PUBLIC_BUCKET_NAME 

17 ), 

18 FileServiceTypes.organization_avatar: _get_s3_service( 

19 settings.S3_FILES_PUBLIC_BUCKET_NAME 

20 ), 

21}