=i3(<UdZddlmZddlZddlZddlZddlZddlZddlZddl m Z ddZ ddZ ejdejdejd ejd gZd ed <dd ZddZddZdddZddZedk(reyy)u impact_analyzer.py - 변경된 파일 목록을 입력받아 영향받는 함수/endpoint/모듈을 추출하는 정적 분석 도구. 사용법: python3 impact_analyzer.py --files "file1.py,file2.py" --output impact.json [--workspace /path/to/search] 제약: - LLM 호출 없음 (AST + grep + 정규식만 사용) - 실행 시간 30초 이내 - Python 파일만 대상 (.py) - 존재하지 않는 파일은 에러 없이 무시 - 빈 파일 목록은 빈 결과 반환 ) annotationsN)Pathctjj|sgS t|j d}t j ||}g}t j|D]}t|t js|j|j|jD]W}t|t j t j"fs.|j|jd|jYt j$|D]H}t|t j t j"fs.|j|jJ|S#tttf$rgcYSwxYw)u# Python AST로 함수/클래스 정의를 파싱하여 이름 목록 반환. - ast.FunctionDef, ast.AsyncFunctionDef → 함수 이름 - ast.ClassDef → 클래스 이름 + 클래스명.메서드명 - 중첩 함수/메서드도 포함 (클래스명.메서드명 형태) utf-8encodingfilename.)ospathisfiler read_textastparse SyntaxErrorOSErrorUnicodeDecodeErrorwalk isinstanceClassDefappendnamebody FunctionDefAsyncFunctionDefiter_child_nodes) file_pathsourcetreeresultsnodeitems 6/home/jay/workspace/teams/shared/qc/impact_analyzer.pyextract_functionsr%s6 77>>) $ i**G*<yy)4G? dCLL ) NN499 %  ?dS__c6J6J$KLNNdii[$))#=> ? ?$$T*& dS__c.B.BC D NN499 %& N% "4 5 s2E//FFcjtjj|sgS t|j d}t j ||}g}t j|D]}t|t jr-|jD]}|j|jJt|t j se|j"sr|j|j"t%t&j)|S#tttf$rgcYSwxYw)u ast.Import, ast.ImportFrom 노드로 import 목록 추출. 모듈 이름만 반환 (from X import Y → X 반환). rrr )r r rrrrrrrrrrImportnamesrr ImportFrommodulelistdictfromkeys)rrr r!r"aliass r$extract_importsr/Fs 77>>) $ i**G*<yy)4G, dCJJ ' +uzz* +cnn -{{t{{+,  g& '' "4 5 s2DD21D2z%self\.path\s*==\s*["\']([^"\']+)["\']z-self\.path\.startswith\(["\']([^"\']+)["\']\)z @\w+\.route\(["\']([^"\']+)["\']zC@\w+\.(get|post|put|patch|delete|head|options)\(["\']([^"\']+)["\']zlist[re.Pattern[str]]_ENDPOINT_PATTERNSctjj|sgS t|j d}g}tD]l}|j|D]V}|jr!|jdk\r|jd}n|jd}||vsF|j|Xn|S#t t f$rgcYSwxYw)u 파일에서 HTTP endpoint 경로 문자열 추출. 지원 패턴: - self.path == "..." - self.path.startswith("...") - @app.route("...") - @router.get/post/put/patch/delete("...") rr) r r rrrrrr0finditer lastindexgroupr)rrr!patternmatchendpoints r$extract_endpointsr:ss 77>>) $ i**G*<G%)%%f- )E5??a#7 ;;q> ;;q>w&x( )) N ' ( sB77C  C c|rtjj|sgSdtj|dtj|d} t j dddd||gddd }|jd vrgS|jjDcgc]#}|js|j%}}|Scc}w#t jtf$rgcYSwxYw) u workspace 내 모든 .py 파일에서 module_name을 import하는 파일 목록 반환. 검색 패턴: - from {module_name} import ... - import {module_name} z(from\s+z\s+import|import\s+z (\s|$|,))grepz-rlz--include=*.pyz-ET )capture_outputtexttimeout)rr3) r r isdirreescape subprocessrun returncodestdout splitlinesstripTimeoutExpiredFileNotFoundError) module_name workspacer7proclinefiless r$find_reverse_importsrQs bggmmI6 "))K011DRYY{E[D\\efG ~~ U,dGY G   ??& (I*.++*@*@*BS$djjlSS T  % %'8 9 s0.CCC 5C C CC.-C.c i}|D]w}t|j}t||}|Dcgc]D}tjj |tjj |k7sC|F}}|||<y|Scc}w)u 각 변경 파일에 대해 역방향 import를 추적하여 딕셔너리 반환. 반환 형식: {파일경로: [이 파일을 import하는 파일들]} )rstemrQr r abspath) changed_filesrMchainrrL importersps r$build_dependency_chainrYs #%E"% 9o** (i@ )^1RWW__Q-?277??S\C]-]Q^ ^$i % L_s AB2Bc|sggggidS|Dcgc]6}|jdstjj|s5|8}}g}g}g}|D]z}t |}|D]}||vs|j |t |} | D]} | |vs|j | t|} | D]} | |vs|j | |t||} ||||| dScc}w)u$ 변경된 파일 목록을 받아 영향 분석 결과 딕셔너리 반환. 반환 형식: { "changed_files": [...], "affected_functions": [...], "affected_endpoints": [...], "affected_imports": [...], "dependency_chain": {...} } )rUaffected_functionsaffected_endpointsaffected_importsdependency_chainz.py) endswithr r rr%rr:r/rY)rPrMf valid_files all_functions all_endpoints all_importsrfuncs endpointseimportsir^s r$analyzerjs. "$"$ " "   $Oqzz%'8RWW^^A=N1OKO!M!MK & !), (A %$$Q' (&i0  (A %$$Q' ("), &A #""1% && .k9E++',  1PsC%C%C%ctjd}|jddd|jddd |jd d d |j}|jj }d|vr>|j dDcgc]#}|j s|j %}}n|r|gng}t||j}t|j}|jjddt|dd5}tj||dddddt!d|t!dt#|dt!dt#|dt!dt#|dt!dt#|dycc}w#1swYxYw) NuI정적 분석으로 변경된 파일의 영향 범위를 파악합니다.) descriptionz--filesTuC분석할 파일 목록 (쉼표 구분) 또는 단일 파일 경로)requiredhelpz--outputz impact.jsonu?결과를 저장할 JSON 파일 경로 (기본값: impact.json))defaultrnz --workspacer uO역방향 import 탐색할 루트 디렉토리 (기본값: 현재 디렉토리),)rM)parentsexist_okwrrFr2) ensure_asciiindentzImpact analysis written to: z Changed files : rUz Affected funcs : r[z Affected endpoints: r\z Affected imports : r])argparseArgumentParser add_argument parse_argsrPrIsplitrjrMroutputparentmkdiropenjsondumpprintlen)parserargsrawr`rPresult output_paths r$mainrs  $ $1| }F  R   N   ^    D **   C cz$'IIcN@qaggi@@" Udnn 5Ft{{#KTD9 k3 1;Q &!%:; ( 67 !#f_&=">!? @A !#f-A&B"C!D EF "3v.B'C#D"E FG !#f-?&@"A!B CDA;;sF6*F6F;;G__main__)rstrreturn list[str])rLrrMrrr)rUrrMrrzdict[str, list[str]])r )rPrrMrrr,)rNone)__doc__ __future__rrvrrr rBrDpathlibrr%r/compiler0__annotations__r:rQrYrjr__name__r$rs #  P(FBJJ78BJJ?@BJJ23BJJUV -) LD06|&ER zFr