mathix420/pylone

import os
import boto3
import shutil

from ...utils.checksum import md5_file
from botocore.client import ClientError


DEFAULT_BUCKET_NAME = 'pylone-bucket'


def init_s3(self):
    self.s3 = boto3.resource(
        's3',
        region_name=self.gb['region'],
        aws_access_key_id=self.creds['access_id'],
        aws_secret_access_key=self.creds['secret_key'],
    )


def _bucket_exist(self, bucket):
    try:
        self.s3.meta.client.head_bucket(Bucket=bucket)
    except ClientError:
        return False
    return True


def _send_to_s3(self, path, config):
    if config.get('as-module') and os.path.isdir(os.path.join('..', path)):
        shutil.make_archive('/tmp/update', 'zip', '..', path)
    elif config.get('as-module'):
        raise Exception('Push as module only support directories')
    elif '.zip' in path:
        shutil.move(path, '/tmp/update.zip')
    elif os.path.isdir(path):
        shutil.make_archive('/tmp/update', 'zip', path)
    elif os.path.isfile(path):
        root_dir = os.path.dirname(path)
        filename = os.path.basename(path)
        shutil.make_archive('/tmp/update', 'zip', root_dir, filename)
    else:
        raise Exception('Not a file or directory')
    key = md5_file('/tmp/update.zip')
    bucket_name = self.gb.get('bucket-name', DEFAULT_BUCKET_NAME)
    if not self._bucket_exist(bucket_name):
        self.s3.create_bucket(
            Bucket=bucket_name,
            # CreateBucketConfiguration={'LocationConstraint': self.gb['region']}
        )
    self.s3.meta.client.upload_file(
        '/tmp/update.zip',
        bucket_name,
        key,
    )
    return {
        'S3Bucket': bucket_name,
        'S3Key': key
    }