[PYTHON] 추상 구문 트리(AST)를 활용한 코드 분석 및 3가지 자동 변형 방법과 해결책

추상 구문 트리 (AST, Abstract Syntax Tree)

 

파이썬은 단순한 스크립트 언어를 넘어, 코드가 코드를 분석하고 조작하는 '메타 프로그래밍'의 정수를 보여주는 언어입니다. 그 마법의 중심에는 추상 구문 트리(AST, Abstract Syntax Tree)가 있습니다. 우리가 작성한 텍스트 형태의 소스 코드는 실행 전 인터프리터에 의해 트리 구조의 데이터로 변환되는데, 이 트리를 직접 다룰 수 있다면 정적 분석 도구 제작, 코드 스타일 강제, 심지어는 특정 패턴의 코드를 자동으로 최적화된 코드로 변환하는 방법까지 구현할 수 있습니다. 본 가이드에서는 파이썬 ast 모듈의 심층 구조와 이를 활용한 실전 코드 변형 기법을 상세히 다룹니다.

1. AST(Abstract Syntax Tree)란 무엇인가?

소스 코드가 컴파일되거나 해석될 때, 구문 분석기(Parser)는 코드의 구조를 계층적인 트리 형태로 구성합니다. 일반적인 파싱 트리(Parse Tree)가 세세한 문법 기호까지 포함하는 것과 달리, AST는 프로그램의 논리적 구조에 집중합니다. 예를 들어 x = 1 + 2라는 코드는 '할당(Assign)'이라는 루트 노드 아래에 '변수(Name)'와 '이항 연산(BinOp)'이라는 자식 노드가 매달린 형태로 표현됩니다.

---

2. AST 분석과 변형의 핵심 도구: NodeVisitor와 NodeTransformer

파이썬의 ast 모듈은 트리를 탐색하고 수정하기 위한 두 가지 강력한 클래스를 제공합니다. 이들의 동작 방식 차이를 이해하는 것이 메타 프로그래밍의 시작입니다.

표: ast.NodeVisitor vs ast.NodeTransformer 상세 비교

비교 항목	ast.NodeVisitor	ast.NodeTransformer
주요 목적	코드 분석 및 정보 추출 (Read-only)	코드 구조 수정 및 변형 (Read-Write)
동작 원리	트리를 순회하며 특정 노드 정보 수집	노드를 방문하여 새로운 노드로 교체하거나 삭제
반환 값	없음 (내부 상태 업데이트)	변형된 노드 객체 (또는 None)
활용 사례	린터(Linter), 복잡도 측정 도구	자동 리팩토링, 코드 난독화, 최적화

3. AST를 활용한 3가지 코드 변형 전략

단순한 텍스트 치환(Regex)으로는 불가능한 정교한 코드 변형 방법을 소개합니다.

전략 1: 특정 함수 호출 자동 교체

프로젝트 전체에서 사용 중인 특정 라이브러리의 함수가 Deprecated 되었을 때, AST를 이용해 해당 호출 패턴을 찾아 새로운 API로 자동 변환할 수 있습니다. 이는 단순 치환보다 훨씬 안전합니다.

전략 2: 보안 취약점 패턴 탐지 및 해결

eval()이나 os.system()과 같이 위험한 함수에 신뢰할 수 없는 입력값이 들어가는지 트리를 추적하여 탐지하고, 이를 안전한 대안으로 교체하는 해결책을 자동화할 수 있습니다.

전략 3: 소스 코드 레벨의 성능 최적화 (Constant Folding)

24 * 60 * 60과 같은 상 수 연산을 바이트코드 단계가 아닌 소스 코드 단계에서 미리 계산된 값인 86400으로 치환하여 가독성과 성능을 동시에 챙길 수 있습니다.

4. Sample Example: 모든 정수 리터럴에 1을 더하는 변형기

아래 코드는 소스 코드 내의 모든 숫자를 찾아 값을 1씩 증가시키는 엉뚱하지만 강력한 AST 변형 예제입니다.

import ast

# 1. 변형할 소스 코드
source_code = "result = (10 + 20) * 5"

# 2. 소스 코드를 AST로 파싱
tree = ast.parse(source_code)

# 3. 노드 변형기 정의 (숫자 노드만 가로채기)
class IncrementTransformer(ast.NodeTransformer):
    def visit_Constant(self, node):
        if isinstance(node.value, int):
            # 숫자인 경우 값을 1 증가시킨 새로운 노드 반환
            return ast.Constant(value=node.value + 1)
        return node

# 4. 트리 변형 실행
transformer = IncrementTransformer()
new_tree = transformer.visit(tree)

# 5. 변형된 트리를 다시 코드로 변환 (Python 3.9+)
print(f"원본 코드: {source_code}")
print(f"변형 코드: {ast.unparse(new_tree)}")
# 결과: result = (11 + 21) * 6

5. AST 활용 시 주의사항과 해결책

트리 조작은 강력하지만 그만큼 위험합니다. 발생 가능한 문제와 해결 방법은 다음과 같습니다.

• 행 번호 정보(Lineno) 손실: 노드를 새로 생성할 때 행 번호 정보를 생략하면 컴파일 에러가 발생합니다. ast.fix_missing_locations() 함수를 사용하여 이를 해결하십시오.
• 버전별 노드 구조 차이: 파이썬 버전이 올라감에 따라 AST 노드 이름이나 구조가 변할 수 있습니다(예: Num 노드가 Constant로 통합됨). 범용 도구를 만든다면 버전별 분기 처리가 필요합니다.
• 가독성 유지: 트리를 코드로 다시 바꿀 때(ast.unparse), 원본 코드의 주석이나 공백 형식이 사라질 수 있습니다. 이를 보존하려면 libcst 같은 라이브러리를 대안으로 고려할 수 있습니다.

6. 결론: 코드 그 이상의 가치, AST

추상 구문 트리는 파이썬 코드가 기계에 전달되기 전의 '생각'을 읽는 도구입니다. AST를 자유자재로 다룰 수 있게 되면, 당신은 단순히 코드를 작성하는 개발자를 넘어 코딩 환경을 직접 개선하고 자동화하는 아키텍트로 거듭나게 될 것입니다. 코드 분석 자동화부터 커스텀 DSL(Domain Specific Language) 제작까지, AST가 제공하는 무궁무진한 메타 프로그래밍의 세계를 탐험해 보시기 바랍니다.

콘텐츠 참조 및 기술 출처

• Python Official Docs.
• "Green Tree Snakes" - The Missing Python AST Docs.
• Real Python: "Python's ast Module: Examine and Modify Your Code".