File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed
Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change 77
88# Try to find a way to get S3 credentials.
99session = None
10+ s3_endpoint = None
1011
1112try :
1213 if session is None :
1516 session = boto3 .Session (
1617 aws_access_key_id = creds ['Credentials' ]['AccessKeyId' ],
1718 aws_secret_access_key = creds ['Credentials' ]['SecretAccessKey' ],
18- aws_session_token = creds ['Credentials' ][ 'SessionToken' ]
19+ aws_session_token = creds ['Credentials' ]. get ( 'SessionToken' ),
1920 )
21+ s3_endpoint = creds .get ('S3Endpoint' )
2022except :
2123 pass
2224
@@ -27,7 +29,7 @@ def upload_pbw(release, file):
2729 filename = f"{ config ['S3_PATH' ]} { release .id } .pbw"
2830 print (f"uploading file { file } to { config ['S3_BUCKET' ]} :{ filename } " )
2931
30- s3 = session .client ('s3' )
32+ s3 = session .client ('s3' , endpoint_url = s3_endpoint )
3133 s3 .upload_file (file , config ['S3_BUCKET' ], filename )
3234
3335def upload_asset (file , mime_type = None ):
@@ -43,7 +45,7 @@ def upload_asset(file, mime_type = None):
4345 else :
4446 mime_type = "image/png"
4547
46- s3 = session .client ('s3' )
48+ s3 = session .client ('s3' , endpoint_url = s3_endpoint )
4749 s3 .upload_file (file , config ['S3_ASSET_BUCKET' ], filename , ExtraArgs = {'ContentType' : mime_type })
4850
4951 return id
You can’t perform that action at this time.
0 commit comments