26 lines
1.2 KiB
Python
26 lines
1.2 KiB
Python
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)
|
|
parser.add_argument("-d", "--dsda", action="store_true")
|
|
parser.add_argument("-s", "--single", action="store_true")
|
|
return parser
|
|
|
|
# TODO: accept configuration for bucket name
|
|
def take_action(self, parsed_args):
|
|
s3_client = boto3.client('s3')
|
|
demo = dcc.config.DemoInPath(parsed_args.wad, parsed_args.map, parsed_args.name)
|
|
bucket = dcc.config.DemoName(parsed_args.wad, parsed_args.map, parsed_args.name)
|
|
if parsed_args.dsda:
|
|
pattern = ("{}{}".format(parsed_args.wad[0:2], parsed_args.map) if not parsed_args.single else parsed_args.wad[0:4]) + "*.zip"
|
|
zips = [x for x in dcc.config.DEMOS.joinpath(parsed_args.wad).glob(pattern)]
|
|
if len(zips) != 1:
|
|
raise Exception("Unable to identify exactly one suitable upload candidate (was {})".format(zips))
|
|
demo = zips[0]
|
|
bucket = "dsda/{}".format(demo.name)
|
|
print("Uploading {} to bucket doom/{}.".format(demo, bucket))
|
|
s3_client.upload_file(demo, 'yrriban', 'doom/' + bucket,
|
|
ExtraArgs={'ContentType': 'binary/octet-stream', 'ACL': 'public-read'})
|