infra-utils/s3utils/move_builds.py

28 lines
866 B
Python
Executable File

#!/usr/bin/env python3
import os
import boto3
import botocore
SRC_BUCKET = 'status-im'
DST_BUCKET = 'status-im-prs'
session = boto3.session.Session()
s3 = session.client('s3',
region_name='ams3',
endpoint_url='https://ams3.digitaloceanspaces.com',
aws_access_key_id=os.environ['DO_ID'],
aws_secret_access_key=os.environ['DO_SECRET'])
for f in s3.list_objects_v2(Bucket=SRC_BUCKET)['Contents']:
name = f['Key']
# ignore is build is not a pr
if 'pr' != name.split('-')[-1].split('.')[0]:
continue
print('{:<25} {} -> {}'.format(name, SRC_BUCKET, DST_BUCKET))
path = '/tmp/{}'.format(name)
s3.download_file(SRC_BUCKET, name, path)
s3.upload_file(path, DST_BUCKET, name)
s3.delete_object(Bucket=SRC_BUCKET, Key=name)
os.remove(path)
break