Skip to content

Commit 4022a32

Browse files
committed
add GCS support to s3.py
1 parent 898a13f commit 4022a32

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

appstore/s3.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77

88
# Try to find a way to get S3 credentials.
99
session = None
10+
s3_endpoint = None
1011

1112
try:
1213
if session is None:
@@ -15,8 +16,9 @@
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')
2022
except:
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

3335
def 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

0 commit comments

Comments
 (0)