CLI for delete all objects in MediaStore container

delete-all-objects.py

#!/usr/bin/python

import sys
import subprocess
import json

args = sys.argv

def deleteObject(endpoint, path, name):
  print ('deleteObject("{0}", "{1}", "{2}"))'.format(endpoint, path, name))
  res = subprocess.check_output(['aws', 'mediastore-data', 'delete-object', '--endpoint', endpoint, '--path', path + '/' + name])

def deleteFolder(endpoint, path):
  print ('deleteFolder("{0}", "{1}"))'.format(endpoint, path))
  res = subprocess.check_output(['aws', 'mediastore-data', 'list-items', '--endpoint', endpoint, '--path', path])
  obj = json.loads(res)
  for item in obj['Items']:
    if item['Type'] == 'OBJECT':
      deleteObject(endpoint, path, item['Name']) 
    elif item['Type'] == 'FOLDER':
      deleteFolder(endpoint, path + '/' + item['Name']) 

def deleteContainer(name):
  print ("Delete all objects in '" + name + "'")
  res = subprocess.check_output(['aws', 'mediastore', 'list-containers'])
  obj = json.loads(res)
  for container in obj['Containers']:
    if container['Name'] != name:
      continue
    deleteFolder(container['Endpoint'], '')

for arg in args[1:]:
  deleteContainer(arg)