def get_inventory_argument(inventory):
    """Get Ansible inventory arguments"""
    inv = ''

    if type(inventory) is list:
        for inv_item in inventory:
            inv += ' -i {}'.format(inv_item)
    else:
        for inv_item in inventory.split(','):
            inv += ' -i {}'.format(inv_item)

    return inv