# # Copyright (c) 2025, RT-Thread Development Team # # SPDX-License-Identifier: Apache-2.0 # # Change Logs: # Date Author Notes # 2025-05-15 supperthomas add PR status show # #!/usr/bin/env python3 import subprocess import sys import os import re import argparse import locale from typing import List, Dict class FileDiff: def __init__(self, path: str, status: str, size_change: int = 0, old_size: int = 0, new_size: int = 0): self.path = path self.status = status # A (added), M (modified), D (deleted), R (renamed) self.size_change = size_change self.old_size = old_size self.new_size = new_size def __str__(self): if self.status == 'A': return f"Added {self.path}: {self.size_change} bytes" elif self.status == 'D': return f"Deleted {self.path}: was {self.old_size} bytes" elif self.status == 'M' or self.status == 'R': return f"Modified {self.path}: {self.size_change} bytes change" else: return f"{self.status} {self.path}" class GitDiffAnalyzer: def __init__(self, target_branch: str): self.target_branch = target_branch self.encoding = locale.getpreferredencoding() def get_diff_files(self) -> List[FileDiff]: """获取当前分支与目标分支之间的差异文件""" # 找到当前分支和目标分支的最近共同祖先 merge_base = self.get_merge_base() if not merge_base: print("No common ancestor found between current branch and target branch") sys.exit(1) # 获取差异文件列表 diff_cmd = f"git diff --name-status {merge_base} " print(diff_cmd) try: output = subprocess.check_output(diff_cmd.split(), stderr=subprocess.STDOUT) output = output.decode(self.encoding).strip() print(output) except subprocess.CalledProcessError as e: print(f"Error executing git diff: {e.output.decode(self.encoding)}") sys.exit(1) if not output: print("No differences between current branch and target branch") sys.exit(0) # 处理可能的换行符问题 output = output.replace('\r\n', '\n') lines = output.split('\n') file_diffs = [] for line in lines: line = line.strip() if not line: continue parts = line.split('\t') if len(parts) < 2: # 可能是重命名文件，格式为 "RXXX\told_path\tnew_path" match = re.match(r'R(\d+)\t(.+)\t(.+)', line) if match: status = 'R' old_path = match.group(2) new_path = match.group(3) # 计算重命名文件的修改大小 old_size = self.get_file_size(old_path, self.target_branch) new_size = self.get_file_size(new_path, 'HEAD') size_change = new_size - old_size if old_size > 0 else new_size file_diffs.append(FileDiff(new_path, status, size_change, old_size, new_size)) else: status = parts[0][0] # 取状态码的第一个字符 path = parts[1] if status == 'A': # 新增文件，计算大小 size = self.get_file_size(path, 'HEAD') file_diffs.append(FileDiff(path, status, size, 0, size)) elif status == 'D': # 删除文件，计算原大小 size = self.get_file_size(path, self.target_branch) file_diffs.append(FileDiff(path, status, 0, size, 0)) elif status == 'M': # 修改文件，计算大小变化 old_size = self.get_file_size(path, self.target_branch) new_size = self.get_file_size(path, 'HEAD') size_change = new_size - old_size file_diffs.append(FileDiff(path, status, size_change, old_size, new_size)) return file_diffs def get_merge_base(self) -> str: """获取当前分支和目标分支的最近共同祖先""" try: cmd = f"git merge-base {self.target_branch} HEAD" print(cmd) output = subprocess.check_output(cmd.split(), stderr=subprocess.STDOUT) return output.decode(self.encoding).strip() except subprocess.CalledProcessError as e: print(f"Error executing git merge-base: {e.output.decode(self.encoding)}") return None def get_file_size(self, path: str, ref: str) -> int: """获取指定分支上文件的大小""" try: # 使用 git cat-file 来获取文件内容，然后计算其大小 cmd = f"git cat-file blob {ref}:{path}" output = subprocess.check_output(cmd.split(), stderr=subprocess.STDOUT) return len(output) except subprocess.CalledProcessError: # 如果文件不存在或无法获取，返回0 return 0 def format_size(size: int) -> str: """将字节大小转换为人类可读的格式""" if size < 1024: return f"{size} bytes" elif size < 1024 * 1024: return f"{size / 1024:.1f} KB" elif size < 1024 * 1024 * 1024: return f"{size / (1024 * 1024):.1f} MB" else: return f"{size / (1024 * 1024 * 1024):.1f} GB" def main(): parser = argparse.ArgumentParser(description='Compare current branch with target branch and show file differences.') parser.add_argument('target_branch', help='Target branch to compare against (e.g., master)') args = parser.parse_args() analyzer = GitDiffAnalyzer(args.target_branch) file_diffs = analyzer.get_diff_files() # 生成报告 generate_report(file_diffs, args.target_branch) def generate_report(file_diffs: List[FileDiff], target_branch: str): """生成差异报告""" print(f"\n=== Comparison between {target_branch} and current branch ===\n") # 分类统计 added_files = [f for f in file_diffs if f.status == 'A'] removed_files = [f for f in file_diffs if f.status == 'D'] modified_files = [f for f in file_diffs if f.status == 'M'] renamed_files = [f for f in file_diffs if f.status == 'R'] # 计算总变化量 total_added = sum(f.size_change for f in added_files) total_removed = sum(f.old_size for f in removed_files) total_modified = sum(abs(f.size_change) for f in modified_files) # 计算总的大小变化 total_size_change = sum(f.size_change for f in file_diffs) print(f"Total changes: {len(file_diffs)} files") print(f"Added: {len(added_files)} files ({format_size(total_added)})") print(f"Removed: {len(removed_files)} files ({format_size(total_removed)})") print(f"Modified: {len(modified_files)} files ({format_size(total_modified)})") print(f"Renamed: {len(renamed_files)} files") print(f"\nTotal size change: {format_size(total_size_change)}") print("\n" + "="*60 + "\n") # 显示详细差异 for diff in file_diffs: print(diff) # 详细展示修改和新增的文件大小 if diff.status == 'A': print(f" Size: {format_size(diff.new_size)}") elif diff.status == 'M': print(f" Original size: {format_size(diff.old_size)}") print(f" New size: {format_size(diff.new_size)}") print(f" Size change: {format_size(abs(diff.size_change))}") elif diff.status == 'R': print(f" Original size: {format_size(diff.old_size)}") print(f" New size: {format_size(diff.new_size)}") print(f" Size change: {format_size(abs(diff.size_change))}") elif diff.status == 'D': print(f" Original size: {format_size(diff.old_size)}") print("-" * 50) if __name__ == "__main__": main()