generates a level of the pascal triangle

pascal.py

'''
Pascal's Triangle by Andrew Li
Prints out the level of the pascal's triangle requested by the user
'''


def next_lvl(old_level):
    length = len(old_level)
    new_lvl = [1]

    for i in range(length - 1):
        new_lvl.append(old_level[i] + old_level[i+1])

    return new_lvl + [1]

def get_level(lvl_request):

    # special cases
    # level 1 is just 1
    if lvl_request == 1:
        print([1])
        return
    elif lvl_request == 2:
        print([1, 1])
        return 

    level = [1, 1]

    for _ in range(lvl_request - 2):
        level = next_lvl(level)

    print(level)

def main():
    try:
        lvl_request = int(input('What level in pascal\'s triangle do you want (level 1 is just 1): '))
        assert lvl_request >= 0, 'Level must be over 1'
    except ValueError:
        print('That is not a valid request')
        return
    except AssertionError as error:
        print(error)
        return

    get_level(lvl_request)

if __name__ == "__main__":
    main()