def arangodb(delete, db_name):
    """Setup ArangoDB database

    db_name: Either 'belns' or 'edgestore' - must be one or the other

    This will create the database, collections and indexes on the collection if it doesn't exist.

    The --delete option will force removal of the database if it exists."""

    if delete:
        client = bel.db.arangodb.get_client()
        bel.db.arangodb.delete_database(client, db_name)

    if db_name == "belns":
        bel.db.arangodb.get_belns_handle(client)
    elif db_name == "edgestore":
        bel.db.arangodb.get_edgestore_handle(client)