| 
 | 1 | +#!/usr/bin/env python3  | 
 | 2 | + | 
 | 3 | + | 
 | 4 | +import argparse  | 
 | 5 | +import os  | 
 | 6 | +import os.path  | 
 | 7 | +import sys  | 
 | 8 | +import urllib.request  | 
 | 9 | + | 
 | 10 | +import tinys3  | 
 | 11 | + | 
 | 12 | + | 
 | 13 | +def main():  | 
 | 14 | +    parser = argparse.ArgumentParser(description='S3 File Uploader')  | 
 | 15 | +    parser.add_argument(  | 
 | 16 | +        '--s3-bucket',  | 
 | 17 | +        help=('S3 bucket name (defaults to $S3_UPLOAD_BUCKET)'),  | 
 | 18 | +        default=os.environ.get('S3_UPLOAD_BUCKET'))  | 
 | 19 | +    parser.add_argument(  | 
 | 20 | +        '--s3-region',  | 
 | 21 | +        help=('S3 region (defaults to $S3_UPLOAD_REGION)'),  | 
 | 22 | +        default=os.environ.get('S3_UPLOAD_REGION'))  | 
 | 23 | +    parser.add_argument(  | 
 | 24 | +        '--s3-username',  | 
 | 25 | +        help=('S3 username (defaults to $S3_UPLOAD_USERNAME)'),  | 
 | 26 | +        default=os.environ.get('S3_UPLOAD_USERNAME'))  | 
 | 27 | +    parser.add_argument(  | 
 | 28 | +        '--s3-key',  | 
 | 29 | +        help=('S3 access key (defaults to $S3_UPLOAD_ACCESSKEY)'),  | 
 | 30 | +        default=os.environ.get('S3_UPLOAD_ACCESSKEY'))  | 
 | 31 | +    parser.add_argument(  | 
 | 32 | +        '--s3-secret',  | 
 | 33 | +        help=('S3 secret (defaults to $S3_UPLOAD_SECRET)'),  | 
 | 34 | +        default=os.environ.get('S3_UPLOAD_SECRET'))  | 
 | 35 | +    parser.add_argument(  | 
 | 36 | +        '--destdir',  | 
 | 37 | +        help='Destination directory.')  | 
 | 38 | +    parser.add_argument(  | 
 | 39 | +        'package', metavar='PACKAGE',  | 
 | 40 | +        help='Package name and version to download.')  | 
 | 41 | + | 
 | 42 | +    args = parser.parse_args()  | 
 | 43 | + | 
 | 44 | +    if args.s3_region:  | 
 | 45 | +        endpoint = 's3-{}.amazonaws.com'.format(args.s3_region.lower())  | 
 | 46 | +    else:  | 
 | 47 | +        endpoint = 's3.amazonaws.com'  | 
 | 48 | + | 
 | 49 | +    conn = tinys3.Connection(  | 
 | 50 | +        access_key=args.s3_key,  | 
 | 51 | +        secret_key=args.s3_secret,  | 
 | 52 | +        default_bucket=args.s3_bucket,  | 
 | 53 | +        tls=True,  | 
 | 54 | +        endpoint=endpoint,  | 
 | 55 | +    )  | 
 | 56 | + | 
 | 57 | +    files = []  | 
 | 58 | + | 
 | 59 | +    for entry in conn.list(args.package):  | 
 | 60 | +        files.append(entry['key'])  | 
 | 61 | + | 
 | 62 | +    destdir = args.destdir or os.getpwd()  | 
 | 63 | + | 
 | 64 | +    for file in files:  | 
 | 65 | +        print('Downloading {}...'.format(file))  | 
 | 66 | +        url = 'https://{}/{}/{}'.format(endpoint, args.s3_bucket, file)  | 
 | 67 | +        target = os.path.join(destdir, file)  | 
 | 68 | +        urllib.request.urlretrieve(url, target)  | 
 | 69 | + | 
 | 70 | +    return 0  | 
 | 71 | + | 
 | 72 | + | 
 | 73 | +if __name__ == '__main__':  | 
 | 74 | +    sys.exit(main())  | 
0 commit comments