Python 타입 힌팅과 mypy로 안전한 코드 작성하기

파이썬으로 꽤 큰 프로젝트를 진행하다 보니, 코드가 산으로 가는 경험을 여러 번 했어요. 특히 팀으로 작업할 때는요. 변수 타입을 일일이 확인하지 않아도 되는 파이썬의 동적 타이핑이 오히려 독이 되는 순간이었죠. 그래서 제가 찾은 해결책이 바로 '타입 힌팅'과 'mypy'입니다. 이 둘을 활용하면 훨씬 안전하고 효율적인 파이썬 코딩이 가능해져요! 한 번 같이 살펴볼까요?

타입 힌팅? mypy? 뭐가 뭔가요?

쉽게 말해 타입 힌팅은 파이썬 코드에 변수나 함수의 타입을 명시적으로 적어주는 거예요. 예를 들어 name: str = "철수" 라고 하면 name 변수는 문자열 타입이라는 걸 미리 알려주는 거죠. : 기호 뒤에 타입을 적어주면 됩니다. int, float, str, bool 같은 기본 타입은 물론이고, list, dict, tuple도 당연히 가능해요. 함수의 매개변수와 반환값에도 타입을 지정할 수 있는데, 이걸 타입 어노테이션이라고 합니다. def greet(name: str) -> str: 이렇게요. 함수 greet는 문자열을 입력받아 문자열을 반환한다는 뜻이죠.

그럼 mypy는 뭘까요? mypy는 이렇게 타입 힌팅을 해놓은 코드를 검사하는 도구입니다. 코드를 실행하기 전에 미리 타입 오류를 찾아주는 거죠. "어? 여기서 int 타입을 기대했는데 str 타입이 들어왔네?" 이런 식으로요. 마치 코드의 문법 검사기 같은 역할이라고 생각하면 돼요. mypy your_script.py 명령어로 간단하게 실행할 수 있습니다. 저는 처음에 이걸 몰라서 엄청 헤맸는데, 알고 나니 정말 신세계였어요.

실제로 어떻게 쓰는 거예요?

제가 예전에 만들었던 코드를 조금 수정해서 보여드릴게요. 두 점 사이의 거리를 계산하는 함수와, 유저 정보를 가져오는 함수, 그리고 데이터 리스트의 합을 계산하는 함수예요.

from typing import List, Tuple, Optional

Point = Tuple[int, int]  # Point라는 타입을 Tuple로 정의했어요.

def calculate_distance(point1: Point, point2: Point) -> float:
    """두 점 사이의 거리를 계산합니다."""
    x1, y1 = point1
    x2, y2 = point2
    distance = ((x2 - x1)**2 + (y2 - y1)**2)**0.5
    return distance

def get_user_data(user_id: int) -> Optional[dict]:
    """유저 데이터를 가져옵니다. 유저가 없으면 None을 반환합니다."""
    # ... (데이터베이스 조회 로직) ...
    user_data = {"name": "Alice", "age": 30}  # 예시 데이터
    return user_data

def process_data(data: List[int]) -> int:
  """데이터 리스트의 합을 계산합니다."""
  total = sum(data)
  return total

# ... (실행 코드) ...

Optional[dict]는 딕셔너리 타입이거나 None일 수 있다는 뜻이에요. 유저가 없을 수도 있으니까요. Tuple[int, int]는 (x, y) 좌표처럼 두 개의 정수로 이루어진 튜플을 의미합니다. 보시다시피, 타입 힌팅을 추가하는 것은 생각보다 어렵지 않아요. mypy를 돌려보면 오류가 있는 부분을 바로 알려주기 때문에 디버깅 시간을 엄청나게 줄일 수 있답니다.

주의할 점은 뭐가 있을까요?

물론 모든 코드에 타입 힌팅을 붙일 필요는 없어요. 핵심 로직이나 복잡한 부분에 집중하는 게 좋죠. 그리고 타입 힌팅은 코드 실행 속도에 영향을 주지 않아요. 런타임에는 타입 체크가 일어나지 않거든요. mypy는 설정 파일(mypy.ini)을 통해 더욱 세밀하게 설정할 수도 있습니다. 저는 처음에 모든 코드에 타입 힌팅을 추가하려고 했는데, 너무 힘들었어요. 차근차근, 새로운 기능을 추가하거나 버그를 수정할 때마다 조금씩 추가하는 게 현실적인 방법입니다.

결론적으로...

타입 힌팅과 mypy는 파이썬 개발에서 정말 유용한 도구입니다. 처음에는 조금 귀찮을 수 있지만, 장기적으로는 코드의 안정성과 유지보수성을 높여주고, 버그를 미리 잡아 시간과 노력을 절약해 줍니다. 저처럼 삽질하지 마시고, 지금부터라도 사용해 보세요! 후회하지 않으실 거예요. 😊