[Python]예외 처리

우리가 python을 통해 코드를 작성하다 보면 수많은 Error를 만나게 된다.

그때 Error를 처리하는 방법에 대해 알아보도록 하자.

try:
	예외가 발생할 가능성이 있는 코드
except:
	예외가 발생했을 때 실행할 코드

def division():
    try:
        num1 = int(input("첫번째 정수를 입력해주세요 > "))
        num2 = int(input("두번째 정수를 입력해주세요 > "))
    
        return print(f"{num1}을 {num2}로 나눈 값은 {num1/num2}입니다.")
    except:
        print("오류가 발생했습니다.")

division()

이렇게 실행할 코드를 try문 안에 넣고 , 예외가 발생 시 처리할 코드를 except문안에 넣었다.

그리고 함수를 실행해 0을 넣어서 ZeroDivisionError나 ValueError를 발생시키면

오류가 발생했습니다.라는 멘트가 출력이 된다.

또한 Error를 지정해서 해당 Error가 발생 시 출력되는 문구를 각각 지정해줄 수도 있다.

def division():
    try:
        num1 = int(input("첫번째 정수를 입력해주세요 > "))
        num2 = int(input("두번째 정수를 입력해주세요 > "))
    
        return print(f"{num1}을 {num2}로 나눈 값은 {num1/num2}입니다.")
    except ValueError:
        print("숫자로 된 정수를 넣어주세요.")
    except ZeroDivisionError:
        print("0은 나눌수 없습니다.")
    except KeyboardInterrupt:
        print("다른 키를 입력하지마세요.")

 

이렇게 에러마다 각각 출력할 문구를 지정해줄수 있다.

또한 어떤 에러가 발생했는지 출력도 가능하다.

def division():
    try:
        num1 = int(input("첫번째 정수를 입력해주세요 > "))
        num2 = int(input("두번째 정수를 입력해주세요 > "))
    
        return print(f"{num1}을 {num2}로 나눈 값은 {num1/num2}입니다.")
    except Exception as e:
        print(e)

이렇게 해서 에러를 발생시키면 어떤 에러인지에 대한 것이 출력이 된다.

그리고, 예외 처리에서도 else문을 사용할 수 있다.

def division():
    try:
        num1 = int(input("첫번째 정수를 입력해주세요 > "))
        num2 = int(input("두번째 정수를 입력해주세요 > "))
    
        result = num1 / num2
    except Exception :
        print("오류가 발생했습니다.")
    else:
        print("정상적으로 나누기가 출력되었습니다.")
        return print(f"{num1}을 {num2}로 나눈 값은 {result}입니다.")

division()

이렇게 else문은 try문에서 예외가 발생하지 않고 처리되었을 경우

else문이 실행이 된다.

또한 finally구문도 있다. finally구문은 무조건 실행되는 코드이다.

 

raise

raise구문은 사용자가 직접 예외를 발생시키는 구문이다.

#화장실이 너무 급한남자

class Toilet:
    def __init__(self):
        self.using = False

    def in_use(self):
        if self.using == False:
            print("화장실을 사용합니다.")
            self.using = True
        else:
            raise Exception("안에 사람이 있습니다. 잠시 기다려주세요.")

    def not_in_use(self):
        self.using = False
        print("안에 사람이 없습니다.")

man = Toilet()

while True:
    use = input("화장실을 사용하시겠습니까?(y/n)")
    try:
        if use == "y":
            man.in_use()
        else:
            man.not_in_use()
    except Exception as e:
        print(e)

위 코드를 실행 시 화장실을 사용할 건지에 대한 코드가 실행된다.

여기서 y를 입력하면 화장실을 사용한다는 문구가 출력이 되고,

한 번 더 y를 입력하면 안에 사람이 있다는 문구가 출력이 된다.

이것은 y를 입력 시 bool타입의 using이 True로 변경이 되고,

한 번 더 y를 입력 시 조건문이 거짓이 되기 때문에, else문을 만나게 된다.

else문에서 안에 사람이 있는데도 불구하고 화장실을 사용할 수 있다면 그건 논리적인 오류가 발생하기 때문에,

raise구문을 통해 예외를 발생시켰다.

즉슨, 함수 자체에서 오류가 발생한 것이 아닌, 예외를 떠 넘긴 것과 같다.