Project Euler #28

euler28.py

def spiral(n):
    max = n * n
    s = 1
    n = 1
    i = 2
    while n <= max:
        for j in range(4):
            if n >= max:
                return s
            else:
                n += i
                s += n
        i = i + 2
    return s

print spiral(1001)