Create process_renovate_prs.py
This commit is contained in:
166
.github/workflows/process_renovate_prs.py
vendored
Normal file
166
.github/workflows/process_renovate_prs.py
vendored
Normal file
@@ -0,0 +1,166 @@
|
||||
# process_renovate_prs.py
|
||||
import os
|
||||
import re
|
||||
import time
|
||||
import requests
|
||||
from typing import Optional, Dict, Any
|
||||
|
||||
# 从环境变量获取GitHub token和仓库信息
|
||||
GITHUB_TOKEN = os.getenv('GITHUB_TOKEN')
|
||||
REPOSITORY = os.getenv('GITHUB_REPOSITORY')
|
||||
HEADERS = {
|
||||
'Authorization': f'Bearer {GITHUB_TOKEN}',
|
||||
'Accept': 'application/vnd.github.v3+json',
|
||||
'X-GitHub-Api-Version': '2022-11-28'
|
||||
}
|
||||
|
||||
def get_prs() -> list:
|
||||
"""获取所有打开的PR列表"""
|
||||
url = f'https://api.github.com/repos/{REPOSITORY}/pulls?state=open'
|
||||
response = requests.get(url, headers=HEADERS)
|
||||
response.raise_for_status()
|
||||
return response.json()
|
||||
|
||||
def is_renovate_pr(pr: Dict[str, Any]) -> bool:
|
||||
"""检查PR是否来自Renovate"""
|
||||
author = pr.get('user', {}).get('login', '').lower()
|
||||
return 'renovate' in author or 'mend' in author
|
||||
|
||||
def parse_version_update(pr_body: str) -> Optional[str]:
|
||||
"""解析PR正文,确定更新类型"""
|
||||
# 匹配版本更新行
|
||||
pattern = r'^\s*([^\s]+)\s+([^\s]+)\s+([^\s]+)\s*->\s*([^\s]+)'
|
||||
lines = pr_body.split('\n')
|
||||
|
||||
for line in lines:
|
||||
match = re.search(pattern, line)
|
||||
if match:
|
||||
package, update_type, old_version, new_version = match.groups()
|
||||
|
||||
# 确定更新类型
|
||||
if update_type.lower() == 'major':
|
||||
return 'major'
|
||||
elif update_type.lower() in ['minor', 'patch']:
|
||||
return update_type.lower()
|
||||
|
||||
# 如果没有明确类型,通过版本号分析
|
||||
old_parts = old_version.split('.')
|
||||
new_parts = new_version.split('.')
|
||||
|
||||
if len(old_parts) > 0 and len(new_parts) > 0:
|
||||
if old_parts[0] != new_parts[0]:
|
||||
return 'major'
|
||||
elif len(old_parts) > 1 and len(new_parts) > 1 and old_parts[1] != new_parts[1]:
|
||||
return 'minor'
|
||||
else:
|
||||
return 'patch'
|
||||
|
||||
return None
|
||||
|
||||
def is_pr_mergeable(pr_number: int) -> bool:
|
||||
"""检查PR是否可以合并"""
|
||||
url = f'https://api.github.com/repos/{REPOSITORY}/pulls/{pr_number}'
|
||||
response = requests.get(url, headers=HEADERS)
|
||||
response.raise_for_status()
|
||||
pr_data = response.json()
|
||||
|
||||
# 检查合并状态
|
||||
return pr_data.get('mergeable', False) and pr_data.get('mergeable_state', '') == 'clean'
|
||||
|
||||
def wait_for_checks(pr_number: int, max_wait: int = 600) -> bool:
|
||||
"""等待检查通过,最多等待max_wait秒"""
|
||||
start_time = time.time()
|
||||
|
||||
while time.time() - start_time < max_wait:
|
||||
if is_pr_mergeable(pr_number):
|
||||
return True
|
||||
|
||||
# 等待30秒后再次检查
|
||||
time.sleep(30)
|
||||
|
||||
return False
|
||||
|
||||
def merge_pr(pr_number: int) -> bool:
|
||||
"""合并PR"""
|
||||
url = f'https://api.github.com/repos/{REPOSITORY}/pulls/{pr_number}/merge'
|
||||
data = {
|
||||
'merge_method': 'merge'
|
||||
}
|
||||
response = requests.put(url, headers=HEADERS, json=data)
|
||||
|
||||
if response.status_code == 200:
|
||||
print(f"成功合并PR #{pr_number}")
|
||||
return True
|
||||
else:
|
||||
print(f"合并PR #{pr_number} 失败: {response.json().get('message', '未知错误')}")
|
||||
return False
|
||||
|
||||
def add_comment(pr_number: int, comment: str):
|
||||
"""在PR上添加评论"""
|
||||
url = f'https://api.github.com/repos/{REPOSITORY}/issues/{pr_number}/comments'
|
||||
data = {'body': comment}
|
||||
response = requests.post(url, headers=HEADERS, json=data)
|
||||
response.raise_for_status()
|
||||
|
||||
def process_pr(pr: Dict[str, Any]):
|
||||
"""处理单个PR"""
|
||||
pr_number = pr['number']
|
||||
pr_author = pr['user']['login']
|
||||
pr_title = pr['title']
|
||||
pr_body = pr.get('body', '')
|
||||
|
||||
print(f"处理PR #{pr_number} 来自 {pr_author}: {pr_title}")
|
||||
|
||||
# 解析更新类型
|
||||
update_type = parse_version_update(pr_body)
|
||||
|
||||
if not update_type:
|
||||
print(f"无法确定PR #{pr_number} 的更新类型,跳过")
|
||||
add_comment(pr_number, "⚠️ 自动处理失败: 无法确定更新类型")
|
||||
return
|
||||
|
||||
print(f"PR #{pr_number} 更新类型: {update_type}")
|
||||
|
||||
# 如果是大版本更新,跳过
|
||||
if update_type == 'major':
|
||||
print(f"PR #{pr_number} 是大版本更新,跳过")
|
||||
add_comment(pr_number, "⏭️ 自动跳过: 大版本更新需要手动审核")
|
||||
return
|
||||
|
||||
# 检查PR是否可以合并
|
||||
if not wait_for_checks(pr_number):
|
||||
print(f"PR #{pr_number} 在等待时间内未通过检查,跳过")
|
||||
add_comment(pr_number, "⏰ 自动跳过: 在等待时间内未通过所有检查")
|
||||
return
|
||||
|
||||
# 尝试合并PR
|
||||
if merge_pr(pr_number):
|
||||
add_comment(pr_number, "✅ 自动合并: 小版本更新已自动合并")
|
||||
else:
|
||||
add_comment(pr_number, "❌ 自动合并失败: 请手动处理")
|
||||
|
||||
def main():
|
||||
"""主函数"""
|
||||
try:
|
||||
# 获取所有PR
|
||||
prs = get_prs()
|
||||
print(f"找到 {len(prs)} 个打开的PR")
|
||||
|
||||
# 筛选Renovate的PR
|
||||
renovate_prs = [pr for pr in prs if is_renovate_pr(pr)]
|
||||
print(f"找到 {len(renovate_prs)} 个Renovate/Mend的PR")
|
||||
|
||||
# 处理每个PR
|
||||
for pr in renovate_prs:
|
||||
try:
|
||||
process_pr(pr)
|
||||
except Exception as e:
|
||||
print(f"处理PR #{pr['number']} 时出错: {str(e)}")
|
||||
# 继续处理下一个PR
|
||||
|
||||
except Exception as e:
|
||||
print(f"处理PR时发生错误: {str(e)}")
|
||||
raise
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Reference in New Issue
Block a user