81 lines
3.9 KiB
Python
81 lines
3.9 KiB
Python
# 必要的依赖库
|
|
import requests
|
|
from datetime import datetime, timedelta, timezone
|
|
from concurrent.futures import ThreadPoolExecutor
|
|
import os
|
|
|
|
|
|
class GitHubActionsManager:
|
|
def __init__(self, repo_url, token):
|
|
self.repo_url = repo_url # GitHub仓库URL
|
|
self.token = token # GitHub个人访问令牌
|
|
self.runs_url = f"{self.repo_url}/actions/runs" # 获取运行记录的URL
|
|
self.current_time = datetime.now(timezone.utc) # 获取当前时间
|
|
|
|
def get_workflow_runs(self, url):
|
|
# 发送GET请求获取工作流运行记录
|
|
response = requests.get(url, headers={"Authorization": f"token {self.token}"})
|
|
if response.status_code == 200: # 如果请求成功
|
|
return response.json()["workflow_runs"] # 返回运行记录列表
|
|
else: # 如果请求失败
|
|
print(
|
|
f"Failed to fetch runs from {url}. Status code: {response.status_code}"
|
|
) # 打印错误信息
|
|
return [] # 返回空列表
|
|
|
|
def delete_run(self, run_id):
|
|
delete_url = f"{self.runs_url}/{run_id}" # 构建删除运行记录的URL
|
|
# 发送DELETE请求删除指定ID的运行记录
|
|
response = requests.delete(
|
|
delete_url, headers={"Authorization": f"token {self.token}"}
|
|
)
|
|
if response.status_code == 204: # 如果删除成功
|
|
print(f"Deleted run with ID {run_id}") # 打印删除成功的消息
|
|
else: # 如果删除失败
|
|
print(
|
|
f"Failed to delete run with ID {run_id}. Status code: {response.status_code}"
|
|
) # 打印错误信息
|
|
|
|
def delete_old_runs(self, max_workers=2):
|
|
next_page = self.runs_url # 初始化下一页的URL为第一页
|
|
while next_page: # 循环直到没有下一页
|
|
print(datetime.now().strftime("%Y-%m-%d %H:%M:%S:%f")[:-3]) # 打印时间
|
|
# 发送GET请求获取一页工作流运行记录
|
|
response = requests.get(
|
|
next_page, headers={"Authorization": f"token {self.token}"}
|
|
)
|
|
if response.status_code == 200: # 如果请求成功
|
|
data = response.json() # 将响应数据转换为JSON格式
|
|
runs = data["workflow_runs"] # 获取运行记录列表
|
|
next_page = response.links.get("next", {}).get("url") # 获取下一页的URL
|
|
|
|
with ThreadPoolExecutor(
|
|
max_workers=max_workers
|
|
) as executor: # 创建线程池
|
|
for run in runs: # 遍历每条运行记录
|
|
# 将运行记录的创建时间转换为UTC时间
|
|
run_time = datetime.strptime(
|
|
run["created_at"], "%Y-%m-%dT%H:%M:%SZ"
|
|
).replace(tzinfo=timezone.utc)
|
|
# 计算当前时间与运行记录创建时间的差值
|
|
time_difference = self.current_time - run_time
|
|
if time_difference > timedelta(hours=168): # 如果差值超过168小时
|
|
executor.submit(
|
|
self.delete_run, run["id"]
|
|
) # 提交删除运行记录的任务
|
|
else: # 如果请求失败
|
|
print(
|
|
f"Failed to fetch runs. Status code: {response.status_code}"
|
|
) # 打印错误信息
|
|
break # 退出循环
|
|
|
|
|
|
if __name__ == "__main__":
|
|
repository_name = os.environ.get("REPOSITORY_NAME")
|
|
github_token = os.environ.get("GITHUB_TOKEN")
|
|
repo_url = f"https://api.github.com/repos/{repository_name}" # GitHub仓库URL
|
|
token = f"{github_token}" # GitHub个人访问令牌
|
|
|
|
manager = GitHubActionsManager(repo_url, token) # 创建GitHubActionsManager实例
|
|
manager.delete_old_runs() # 删除168小时前的运行记录
|