make_lots_of_expired_files.py

from argparse import ArgumentParser
import os
import sys
from swiftclient.client import Connection
import time

parser = ArgumentParser('make a lot of files')
parser.add_argument('--container', default='lots-of-files',
                    help='the container name')
parser.add_argument('--prefix', default='obj',
                    help='the object name <PREFIX><N>')
parser.add_argument('--number', default=100, type=int,
                    help='number <N> of objects to make')
parser.add_argument('--expire-in', default=30, type=int,
                    help='number <N> seconds before expire')
parser.add_argument('--sleep-for', default=0.1, type=float,
                    help='number <N> seconds between each')

auth_url = os.environ['ST_AUTH']
user = os.environ['ST_USER']
key = os.environ['ST_KEY']


def main():
    args = parser.parse_args()

    conn = Connection(auth_url, user, key, insecure=True)
    container = args.container
    conn.put_container(container)
    fill_size = len(str(args.number))
    for i in range(args.number):
        obj_name = '%s%s' % (args.prefix, str(i).zfill(fill_size))
        conn.put_object(container, obj_name, None, headers={
            'x-delete-after': args.expire_in,
        })
        print('uploaded %s/%s' % (container, obj_name))
        time.sleep(args.sleep_for)


if __name__ == "__main__":
    sys.exit(main())