sample_store.py

from war import app
import swiftclient
import os


def _get_swift_client():
    object_store_config = app.config['OBJECT_STORE']

    return swiftclient.Connection(
        authurl=object_store_config['AUTH_URL'],
        user=object_store_config['USERNAME'],
        key=object_store_config['PASSWORD'],
        tenant_name=object_store_config['TENANT_NAME'],
        auth_version='2',
        os_options={'tenant_id': object_store_config['TENANT_ID'], 'region_name': object_store_config['REGION_NAME']}
    )


def save_locally(file, sample_id):
    sample_file_path = get_local_path(sample_id)

    file.save(sample_file_path)


def save_remotely(file, sample_id):
    client = _get_swift_client()

    client.put_object(
        'samples',
        sample_id + '.wav',
        contents=file,
        content_type='audio/wav'
    )


def download_from_remote(sample_id):
    client = _get_swift_client()

    resp_headers, obj_contents =  client.get_object('samples', sample_id + '.wav')

    with open(get_local_path(sample_id), 'wb') as local:
        local.write(obj_contents)


def get_local_path(sample_id, check_if_exists=False):
    sample_file_destination = os.path.abspath(app.config['SAMPLES_PATH'])
    sample_file_name = '{}.wav'.format(sample_id)
    sample_file_path = os.path.join(sample_file_destination, sample_file_name)

    if check_if_exists and not os.path.exists(sample_file_path):
        raise Exception('The sample file "{}" does not exists on the file system'.format(sample_file_path))

    return sample_file_path


def get_remote_path(sample_id):
    return app.config['OBJECT_STORE']['STORAGE_URL'] + app.config['OBJECT_STORE']['TENANT_ID'] + '/samples/' + sample_id + '.wav'


def delete_locally(sample_id):
    sample_file_path = get_local_path(sample_id)

    if os.path.exists(sample_file_path):
        os.remove(sample_file_path)


def delete_remotely(sample_id):
    client = _get_swift_client()

    client.delete_object('samples', sample_id + '.wav')