python - 「戻る」機能

Question

私はpythonで測定機を作ろうとしていますが、測定値を入力するとコードが続きます。結果が得られたら処理を停止したいのですが、これまでに行ったコードは次のとおりです。

print math.sqrt(int(h1)**2 + int(h2)**2)" "の後に停止する必要があります

need = raw_input("What do you need to Use?")

if need == "pythagoras" or "Pythagoras":
    pythagoras = raw_input("What side do you Need?")

if pythagoras == "hypotenuse" or "Hypotenuse":
    h1 = int(raw_input("Known Side 1"))
    h2 = int(raw_input("Known Side 2"))

import math
print math.sqrt(int(h1)**2 + int(h2)**2)

Answer 1

を呼び出すと、関数の場合sys.exitと同様に、プログラムの実行がすぐに終了します。return

import sys

...
print math.sqrt(int(h1)**2 + int(h2)**2)
sys.exit()