class Square:
def __init__(self, height, base):
self.height = height
self.base = base

def calculate_area(self):
area = self.height * self.base
return area

class Triangle:
def __init__(self, height, base):
self.height = height
self.base = base

def calculate_area(self):
area = self.height * self.base
return area

def startSqr():
try:
height = int(input(“높이를 입력하세요.”))
base = int(input(“밑변의 길이를 입력하세요.))

if height <= 0 or base <= 0:
print(“양수만을 입력하세요.”)
rerturn

sqr = Square(height, base)

print(“사각형의 넓이는”, area, “입니다.”)

except ValueError:
print(“숫자를 입력해야 합니다.)

def startTri():
try:
height = int(input(“높이를 입력하세요.”))
base = int(input(“밑변의 길이를 입력하세요.”))

if height <= 0 or base <= 0:
print(“양수만을 입력하세요.“)
return

tri = Triangle(height, base)

except ValueError:
print(“숫자를 입력해야 합니다.”)

def select():
try:
print(“사각형은 1번, 삼각형은 2번 입력”)
choice = int(input(“넓이는 구하고 싶은 도형의 번호를 입력하세요.))

if choice != 1 or choice != 2:
print(“1과 2 중 하나만 입력하세요.”)
return
elif choice == 1:
startSqr()
elif choice == 2:
startTri()
except ValueError:
print(“숫자를 입력해야 합니다.”)

if __name__ = “__main__”:
select()

이거 맞냐?