eviltnan/freeturn

from django.core.exceptions import ImproperlyConfigured
from django.template.loader import render_to_string
from django.conf import settings


def render_policy(bucket, account, user):
    variables = {
        'bucket': bucket,
        'account_id': account,
        'user': user
    }
    return render_to_string('s3_policy.json.template', context=variables)


def install_policy(bucket, account, user):
    import boto3
    policy = render_policy(bucket, account, user)
    s3 = boto3.client('s3')
    return s3.put_bucket_policy(
        Bucket=bucket,
        Policy=policy
    )


def install_storage_policy():
    bucket = settings.AWS_STORAGE_BUCKET_NAME
    account = settings.AWS_STORAGE_ACCOUNT_ID
    user = settings.AWS_STORAGE_USER
    if not (bucket and account and user):
        raise ImproperlyConfigured

    return install_policy(bucket, account, user)