파이선으로 계산기 만들어줘

제가 만들어드린 파이썬 계산기입니다.

# 계산기 함수 정의
def calculator():
    # 사용자로부터 숫자 입력 받기
    num1 = float(input("첫 번째 숫자를 입력하세요: "))
    num2 = float(input("두 번째 숫자를 입력하세요: "))

    # 사용자로부터 연산자 입력 받기
    operator = input("연산자를 입력하세요 (+, -, *, /): ")

    # 연산자에 따라 계산하기
    if operator == "+":
        result = num1 + num2
    elif operator == "-":
        result = num1 - num2
    elif operator == "*":
        result = num1 * num2
    elif operator == "/":
        if num2 == 0:
            print("0으로 나눌 수 없습니다.")
            return None
        else:
            result = num1 / num2
    else:
        print("잘못된 연산자입니다.")
        return None

    # 계산 결과 출력하기
    print("결과:", result)

# 계산기 실행
calculator()

위 코드를 실행하면, 사용자로부터 숫자와 연산자를 입력받아 계산 결과를 출력합니다. 0으로 나누는 경우에는 오류 메시지를 출력합니다.