파이썬 입력값이 정수가 아니면 에러 출력하고 다시 입력받기 - paisseon iblyeoggabs-i jeongsuga animyeon eleo chullyeoghago dasi iblyeogbadgi

# 연습문제 14-3

# 입력받은 두 정수의 합을 구하는 프로그램.

# 실수가 입력되면 예외가 발생하고, 이 경우 "정수를 입력하세요" 라는 문자열이 출력된다.

try: # 첫 번째 정수 입력, 정수 대신 실수를 입력할 때 의 오류 메시지는 "ValueError"

val1=int(input("첫 번째 정수를 입력하세요: "))

except ValueError: # 에러가 발생한 경우

print("정수를 입력하세요")

else: # 첫 번째 정수가 잘 입력된 경우 두 번째 정수를 입력한다. 과정은 동일하다.

try:

val2=int(input("두 번째 정수를 입력하세요: "))

except ValueError:

print("정수를 입력하세요")

else:

print(val1+val2)

finally: # 항상 실행되는 안내문

print("프로그램이 실행되었습니다.")

프로그램을 만들다 보면 수없이 많은 오류를 만나게 된다. 물론 오류가 발생하는 이유는 프로그램이 잘못 동작하는 것을 막기 위한 파이썬의 배려이다. 이번 장에서는 파이썬에서 오류를 처리하는 방법에 대해서 알아보자.

  • 오류는 어떤 때 발생하는가?
  • 오류 예외 처리 기법
    • try, except문
    • try .. finally
    • 여러개의 오류처리하기
    • try ... else
  • 오류 회피하기
  • 오류 일부러 발생시키기
  • 예외 만들기

오류는 어떤 때 발생하는가?

오류를 처리하는 방법을 공부하기 전에 어떤 상황에서 오류가 발생하는지 한번 알아보자. 오타를 입력했을 때 발생하는 구문 오류 같은 것이 아닌 실제 프로그램에서 자주 발생하는 오류를 중심으로 살펴보자.

먼저 존재하지 않는 파일을 사용하려고 시도했을 때 발생하는 오류이다.

>>> f = open("나없는파일", 'r')
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
FileNotFoundError: [Errno 2] No such file or directory: '나없는파일'

위 예에서 볼 수 있듯이 없는 파일을 열려고 시도하면 FileNotFoundError 오류가 발생한다.

이번에는 0으로 다른 숫자를 나누는 경우를 생각해 보자. 이 역시 자주 발생하는 오류이다.

>>> 4 / 0
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ZeroDivisionError: division by zero

4를 0으로 나누려니까 ZeroDivisionError 오류가 발생한다.

마지막으로 한 가지 예를 더 들어 보자. 다음 오류는 정말 빈번하게 일어난다.

>>> a = [1,2,3]
>>> a[4]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
IndexError: list index out of range

a는 리스트

>>> a = [1,2,3]
>>> a[4]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
IndexError: list index out of range
9이므로 a[4]는 a 리스트에서 얻을 수 없는 값이다. 따라서 IndexError 오류가 발생한다. 파이썬은 이런 오류가 발생하면 프로그램을 중단하고 오류 메시지를 보여 준다.

오류 예외 처리 기법

자, 이제 유연한 프로그래밍을 위한 오류 처리 방법에 대해 알아보자.

try, except문

다음은 오류 처리를 위한 try, except문의 기본 구조이다.

try:
    ...
except [발생오류 [as 오류변수]]:
    ...

try 블록 수행 중 오류가 발생하면 except 블록이 수행된다. 하지만 try 블록에서 오류가 발생하지 않는다면 except 블록은 수행되지 않는다.

except 구문을 자세히 살펴보자.

except [발생오류 [as 오류변수]]:

위 구문을 보면 [ ] 기호를 사용하는데, 이 기호는 괄호 안의 내용을 생략할 수 있다는 관례적인 표기법이다. 즉 except 구문은 다음 3가지 방법으로 사용할 수 있다.

1. try, except만 쓰는 방법

try:
    ...
except:
    ...

이 경우는 오류 종류에 상관없이 오류가 발생하면 except 블록을 수행한다.

2. 발생 오류만 포함한 except문

try:
    ...
except 발생오류:
    ...

이 경우는 오류가 발생했을 때 except문에 미리 정해 놓은 오류와 동일한 오류일 경우에만 except 블록을 수행한다는 뜻이다.

3. 발생오류와 오류변수까지 포함한 except문

try:
    ...
except 발생오류 as 오류변수:
    ...

이 경우는 두 번째 경우에서 오류의 내용까지 알고 싶을 때 사용하는 방법이다.

이 방법의 예를 들어 보면 다음과 같다.

try:
    4 / 0
except ZeroDivisionError as e:
    print(e)

위처럼 4를 0으로 나누려고 하면 ZeroDivisionError가 발생하여 except 블록이 실행되고 오류변수 e에 담기는 오류 메시지를 출력할 수 있다. 출력한 오류메시지는 다음과 같다.

division by zero

try .. finally

try문에는 finally절을 사용할 수 있다. finally절은 try문 수행 도중 예외 발생 여부에 상관없이 항상 수행된다. 보통 finally절은 사용한 리소스를 close해야 할 때에 많이 사용한다.

다음 예를 보자.

try:
    f = open('foo.txt', 'w')
    # 무언가를 수행한다.

    (... 생략 ...)

finally:
    f.close()  # 중간에 오류가 발생하더라도 무조건 실행된다.

foo.txt 파일을 쓰기 모드로 연 후에 try문을 수행한 후 예외 발생 여부와 상관없이 finally절에서

try:
    ...
except [발생오류 [as 오류변수]]:
    ...
0로 열린 파일을 닫을 수 있다.

여러개의 오류처리하기

try문 안에서 여러 개의 오류를 처리하려면 다음과 같이 사용한다.

try:
    ...
except 발생오류1:
   ... 
except 발생오류2:
   ...

즉 0으로 나누는 오류와 인덱싱 오류를 다음과 같이 처리할 수 있다.

>>> 4 / 0
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ZeroDivisionError: division by zero
0

a는 2개의 요솟값을 가지고 있기 때문에

try:
    ...
except [발생오류 [as 오류변수]]:
    ...
1는
try:
    ...
except [발생오류 [as 오류변수]]:
    ...
2를 발생시키므로 "인덱싱할 수 없습니다."라는 문자열이 출력될 것이다. 인덱싱 오류가 먼저 발생했으므로
try:
    ...
except [발생오류 [as 오류변수]]:
    ...
3으로 발생되는
try:
    ...
except [발생오류 [as 오류변수]]:
    ...
4 오류는 발생하지 않는다.

앞에서 알아본 것과 마찬가지로 오류 메시지도 다음과 같이 확인할 수 있다.

>>> 4 / 0
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ZeroDivisionError: division by zero
1

프로그램을 실행하면 "list index out of range" 라는 오류 메시지가 출력될 것이다.

다음과 같이

try:
    ...
except [발생오류 [as 오류변수]]:
    ...
5와
try:
    ...
except [발생오류 [as 오류변수]]:
    ...
2를 함께 처리할 수도 있다.

>>> 4 / 0
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ZeroDivisionError: division by zero
2

2개 이상의 오류를 동일하게 처리하기 위해서는 위와 같이 괄호를 사용하여 함께 묶어 처리하면 된다.

try ... else

try문에는 다음처럼 else절을 사용할 수 있다.

>>> 4 / 0
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ZeroDivisionError: division by zero
3

try문 수행중 오류가 발생하면 except절이 수행되고 오류가 없으면 else절이 수행된다.

다음은 try문에 else절을 사용한 간단한 예제이다.

>>> 4 / 0
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ZeroDivisionError: division by zero
4

만약 '나이를 입력하세요:' 라는 질문에 숫자가 아닌 다른 값을 입력하면 오류가 발생하여 '입력이 정확하지 않습니다.'라는 문장을 출력한다. 오류가 없을 경우에만 else절이 수행된다.

오류 회피하기

코드를 작성하다 보면 특정 오류가 발생할 경우 그냥 통과시켜야 할 때가 있다. 다음 예를 보자.

>>> 4 / 0
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ZeroDivisionError: division by zero
5

try문 안에서 FileNotFoundError가 발생할 경우에 pass를 사용하여 오류를 그냥 회피하도록 작성한 예제이다.

오류 일부러 발생시키기

이상하게 들리겠지만 프로그래밍을 하다 보면 종종 오류를 일부러 발생시켜야 할 경우도 생긴다. 파이썬은 raise 명령어를 사용해 오류를 강제로 발생시킬 수 있다.

예를 들어 Bird 클래스를 상속받는 자식 클래스는 반드시 fly라는 함수를 구현하도록 만들고 싶은 경우(강제로 그렇게 하고 싶은 경우)가 있을 수 있다. 다음 예를 보자.

>>> 4 / 0
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ZeroDivisionError: division by zero
6

위 예제는 Bird 클래스를 상속받는 자식 클래스는 반드시 fly 함수를 구현해야 한다는 의지를 보여 준다. 만약 자식 클래스가 fly 함수를 구현하지 않은 상태로 fly 함수를 호출한다면 어떻게 될까?

NotImplementedError는 파이썬에 이미 정의되어 있는 오류로, 꼭 작성해야 하는 부분이 구현되지 않았을 경우 일부러 오류를 일으키기 위해 사용한다.

>>> 4 / 0
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ZeroDivisionError: division by zero
7
>>> 4 / 0
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ZeroDivisionError: division by zero
8

Eagle 클래스는 Bird 클래스를 상속받았다. 그런데 Eagle 클래스는 fly 메서드를 오버라이딩하여 구현하지 않았다. 따라서 eagle 객체의 fly 메서드를 수행하는 순간 Bird 클래스의 fly 메서드가 수행되어 NotImplementedError가 발생한다.

상속받는 클래스에서 메서드를 재구현하는 것을 메서드 오버라이딩이라고 한다.

NotImplementedError가 발생되지 않게 하려면 다음과 같이 Eagle 클래스에 fly 함수를 구현해야 한다.

>>> 4 / 0
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ZeroDivisionError: division by zero
9

위 예처럼 fly 함수를 구현한 후 프로그램을 실행하면 오류 없이 다음 문장이 출력된다.

>>> a = [1,2,3]
>>> a[4]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
IndexError: list index out of range
0

예외 만들기

프로그램 수행 도중 특수한 경우에만 예외 처리를 하기 위해서 종종 예외를 만들어서 사용한다. 이번에는 직접 예외를 만들어 보자. 예외는 다음과 같이 파이썬 내장 클래스인 Exception 클래스를 상속하여 만들 수 있다.

>>> a = [1,2,3]
>>> a[4]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
IndexError: list index out of range
1

그리고 별명을 출력하는 함수를 다음과 같이 작성해 보자.

>>> a = [1,2,3]
>>> a[4]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
IndexError: list index out of range
2

그리고 다음과 같이 say_nick 함수를 호출해 보자.

>>> a = [1,2,3]
>>> a[4]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
IndexError: list index out of range
3

저장한 뒤 프로그램을 실행해 보면 다음과 같이 "천사"가 한 번 출력된 후 MyError가 발생한다.

>>> a = [1,2,3]
>>> a[4]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
IndexError: list index out of range
4

이번에는 예외 처리 기법을 사용하여 MyError 발생을 예외 처리해 보자.

>>> a = [1,2,3]
>>> a[4]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
IndexError: list index out of range
5

프로그램을 실행하면 다음과 같이 출력된다.

>>> a = [1,2,3]
>>> a[4]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
IndexError: list index out of range
6

만약 오류 메시지를 사용하고 싶다면 다음처럼 예외 처리를 하면 된다.

>>> a = [1,2,3]
>>> a[4]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
IndexError: list index out of range
7

하지만 프로그램을 실행해 보면

try:
    ...
except [발생오류 [as 오류변수]]:
    ...
7로 오류 메시지가 출력되지 않는 것을 확인할 수 있다. 오류 메시지를 출력했을 때 오류 메시지가 보이게 하려면 오류 클래스에 다음과 같은
try:
    ...
except [발생오류 [as 오류변수]]:
    ...
8 메서드를 구현해야 한다.
try:
    ...
except [발생오류 [as 오류변수]]:
    ...
8 메서드는 print(e)처럼 오류 메시지를 print문으로 출력할 경우에 호출되는 메서드이다.