Pagination python

pagination.py

# total - total pages count
# page - current page
# neighbour_count - nighbour pages count, from each side
#
# 'first' - link to first page
# 'last' - link to last page

def pagination(total, page=1, neighbour_count=2):
    result = []
    if page-neighbour_count > 2:
            result.append('first')
    if page-neighbour_count == 2:
            result.append(1)
    for i in range(page-neighbour_count, page+neighbour_count+1):
        if 0 < i <= total:
            result.append(i)
    if page+neighbour_count < total-1:
        result.append('last')
    if page+neighbour_count == total-1:
        result.append(total)
    return result