螺旋队列问题 python

设1点的坐标为（0,0），x方向向右为正，y向下为正，如7的坐标是（-1，-1）。编程实现任意输入一个点的坐标（x,y）。输出所对应的数字。

7 8 9
6 1 2
5 4 3

def foo(x, y):
    t = max(abs(x), abs(y)) #属于第几圈
    u = 2*t + 1  
    w = u*u  #该圈最大的数
    if x == -t:
        v = w - 3*t - y
    elif y == -t:
        v = w - t + x
    elif y == t:
        v = w - 5*t - x
    else:
        v = w - 7*t + y
    return v
print(foo(2,3))