doomcc/doomcc/put.py

22 lines
655 B
Python

import boto3
import doomcc.config
import doomcc.doom_base
class Put(doomcc.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"},
)