20. None을 반환하기보다는 예외를 발생시켜라¶

In [1]:

def careful_divide(a, b):
    try:
        return a / b
    except ZeroDivisionError:
        return None

In [2]:

x, y = 1, 0

In [3]:

result = careful_divide(x, y)

In [4]:

if result is None:
    print('error')

error

In [5]:

if not result:
    print('error')

error

False와 동등한 반환 값을 잘못 해석하는 경우는 None이 특별한 의미를 가지는 파이썬 코드에서 흔히 저지르는 실수이다.

실수 가능성을 줄이는 방법 두가지

2-튜플로 분리하는 것

첫 번째 부분은 연산이 성공인지 실패인지 표시, 두번째 부분은 성공시 결과 값

In [6]:

def careful_divide(a, b):
    try:
        return True, a / b
    except ZeroDivisionError:
        return False, None

In [7]:

success, result = careful_divide(x, y)

In [8]:

if not success:
    print('error')

error

문제점은 첫번째를 무시할수 있다는 점

In [9]:

_, result = careful_divide(x, y)
if not success:
    print('error')

error

이런 실수를 줄일 수 있는 더 나은 두번째 방법
2. None을 반환하지 않는 것

In [11]:

def careful_divide(a, b):
    try:
        return a / b
    except ZeroDivisionError as e:
        raise ValueError('error')

In [13]:

x, y = 5, 2
try:
    result = careful_divide(x, y)
except ValueError as e:
    print(e)
else:
    print('결과는 %.1f 입니다' % result)

결과는 2.5 입니다

In [14]:

x, y = 5, 0
try:
    result = careful_divide(x, y)
except ValueError as e:
    print(e)
else:
    print('결과는 %.1f 입니다' % result)

error

이 방법을 확장해서 타입 애너테이션을 사용하는 코드에도 적용이 가능하다.

독스트링과 타입 애너테이션까지 포함시키면 다음과 같다

In [15]:

def careful_divide(a: float, b: float) -> float:
    """a를 b로 나눈다.
    
    Raises:
        ValueError: b가 0이어서 나눗셈을 할 수 없을 때
    """
    try:
        return a / b
    except ZeroDivisionError as e:
        raise ValueError('error')

기억해야 할 내용¶

특별한 의미를 표시하는 None을 반환하는 함수를 사용하면 None과 다른 값(0 이나 빈 문자열)이 조건문에서 False로 평가될 수 있기 때문에 실수하기 쉽다.
특별한 상황을 표현하기 위해 None을 반환하는 대신 예외를 발생시켜라. 문서에 예외 정보를 기록해 호출자가 예외를 제대로 처리하도록 하라.
함수가 특별한 경우를 포함하는 그 어떤 경우에도 절대로 None을 반환하지 않는다는 사실을 타입 애너테이션으로 명시할 수 있다.