Нахождение кратчайшего пути в трехмерной системе координат

def fun(a1, a2, b1, b2):
    # уравнение радиатора в ХОУ
    ax1 = a2[1] - a1[1]
    by1 = a1[0] - a2[0]
    c1 = a1[1]*a2[0] - a1[0]*a2[1]
    norm = (ax1**2 + by1**2)**(-.5)
    
    # уравнение стояка в ХОУ
    ax2 = b2[1] - b1[1]
    by2 = b1[0] - b2[0]
    c2 = b1[1]*b2[0] - b1[0]*b2[1]
    
    if b1[2] == b2[2]:
        # обе трубы горизонтальны
        if ax2:
            tmp = abs((c2*ax1/ax2-c1) * norm)
        elif:
            tmp = abs((c2*bx1/bx2-c1) * norm)
        else:
            print('стояка не существует )))')
            return
            
        return round(abs(b1[2] - a1[2]), 3), round(tmp, 3)
    
    else:
        # вертикальный стояк В1-В2
        
        # v1 - расстояние от стояка до прямой а1-а2
        v1 = abs(norm * (ax1*b1[0] + by1*b1[1] + c1))
        
        # коэф "c" уравнения нормали к а1-а2 из точки М
        cm = b1[1]*ax1 - b1[0]*by1
        
        # координаты точки М
        ym = (cm*ax1 - by1*c1) * norm**2
        xm = -(c1 + by1*ym) / ax1
        
        s1 = (xm - a1[0])**2 + (ym - a1[1])**2
        s2 = (xm - a2[0])**2 + (ym - a2[1])**2
        
        # v2 - минимум от точки М до точек отрезка a1 и а2
        v2 = min(s1, s2) **.5
        
        return round(v1, 3), round(v2, 3)
 
 
 
a1 = (143.604, -106.206, 1.958)
a2 = (145.010, -106.530, 1.958)
 
b1 = (140.200, -108.156, -0.0656)
b2 = (149.279, -110.252, -0.0656)
print(fun(a1, a2, b1, b2))
print()
 
a1 = (143.6040 -120.821, 16.5740)
a2 = (145.010, -121.1463, 16.5740)
 
b1 = (146.274, -120.6302, 13.385)
b2 = (146.274, -120.6302, 18.307)
 
print(fun(a1, a2, b1, b2))