실수 또는 정수만 입력받는 함수 (파이썬)

# 실수 입력 함수 

def readFloat(Msg, errMsg):

    while True:

        val = input(Msg)

        try:

            val = float(val)

            return val

        except:

          print(errMsg)

          

# 정수 입력 함수

def readInt(Msg, errMsg):

    while True:

        val = input(Msg)

        try:

            val = int(val)

            return val

        except:

            print(errMsg)

# 지정형 입력 함수

def readWhat(typename, Msg, errMsg):

    while True:

        val = input(Msg)

        try:

            val = typename(val)

            return val

        except:

            print(errMsg)

# 테스트 예제 

f = readFloat("실수를 입력하세요", "잘못된 값입니다.")

print(f)

i = readInt("정수를 입력하세요", "잘못된 값입니다.")

print(i)

f = readWhat(float, "실수를 입력하세요", "잘못된 값입니다.")

print(f)

i = readWhat(int, "정수를 입력하세요", "잘못된 값입니다.")

print(i)