hera_librarian.py

# Python bindings of Librarian RPCs

from hera_rpc import *

def get_site(site_name):
    config = get_client_config()
    sites = config['sites']
    try:
        return sites[site_name]
    except IndexError:
        return None

def create_observation(site_name, obs_id, julian_date, polarization, length):
    site = get_site(site_name)
    if site == None:
        return error_struct("no such site")
    req = {'operation': 'create_observation',
        'authenticator': site['authenticator'],
        'id': obs_id,
        'julian_date': julian_date,
        'polarization': polarization,
        'length': length}
    return do_http_post(req, site)

def create_file(site_name, store_name, file_name, type, obs_id, size, md5):
    site = get_site(site_name)
    if site == None:
        return error_struct("no such site")
    req = {'operation': 'create_file',
        'authenticator': site['authenticator'],
        'store_name': store_name,
        'file_name': file_name,
        'type': type,
        'obs_id': obs_id,
        'size': size,
        'md5': md5
    }
    return do_http_post(req, site)

def delete_file(site_name, file_name, store_name):
    site = get_site(site_name)
    if site == None:
        return error_struct("no such site")
    req = {'operation': 'delete_file',
        'authenticator': site['authenticator'],
        'name': file_name,
        'store_name': store_name
    }
    return do_http_post(req, site)

def get_store_list(site_name):
    site = get_site(site_name)
    if site == None:
        return error_struct("no such site")
    req = {'operation': 'get_store_list',
        'authenticator': site['authenticator']
    }
    return do_http_post(req, site)

def recommended_store(site_name, file_size):
    site = get_site(site_name)
    if site == None:
        return error_struct("no such site")
    req = {'operation': 'recommended_store',
        'authenticator': site['authenticator'],
        'file_size': file_size
    }
    return do_http_post(req, site)

def create_copy_task(task_type, local_site_name, local_store_name, file_name,
    remote_site_name, remote_store_name, delete_when_done):
    site = get_site(local_site_name)
    if site == None:
        return error_struct("no such site")
    req = {'operation': 'create_copy_task',
        'authenticator': site['authenticator'],
        'task_type': task_type,
        'local_store_name': local_store_name,
        'file_name': file_name,
        'remote_site_name': remote_site_name,
        'remote_store_name': remote_store_name,
        'delete_when_done': delete_when_done
    }
    return do_http_post(req, site)