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'})