doomcc/dcc/put.py

import boto3
import dcc.config
import dcc.doom_base


class Put(dcc.doom_base.WadMap):
    def get_parser(self, prog_name):
        parser = super().get_parser(prog_name)
        return parser

    # TODO: accept configuration for bucket name
    def take_action(self, parsed_args):
        s3_client = boto3.client("s3")
        demo = self.demo_in_path()
        bucket = self.target_bucket()
        print("Uploading {} to bucket {}.".format(demo, bucket))
        s3_client.upload_file(
            demo,
            "yrriban",
            bucket,
            ExtraArgs={"ContentType": "binary/octet-stream", "ACL": "public-read"},
        )