146 lines
4.7 KiB
Python
146 lines
4.7 KiB
Python
import os
|
|
import json
|
|
import requests
|
|
from datetime import datetime, timedelta
|
|
import time
|
|
|
|
class GitHubCleaner:
|
|
def __init__(self, repo, token, days_to_keep=7):
|
|
"""
|
|
初始化GitHub清理器
|
|
:param repo: 仓库名 (格式: 用户名/仓库名)
|
|
:param token: GitHub Token
|
|
:param days_to_keep: 保留最近几天的图片
|
|
"""
|
|
self.repo = repo
|
|
self.headers = {
|
|
"Authorization": f"token {token}",
|
|
"Accept": "application/vnd.github.v3+json"
|
|
}
|
|
self.days_to_keep = days_to_keep
|
|
|
|
def get_images(self):
|
|
"""获取仓库中的所有图片"""
|
|
try:
|
|
response = requests.get(
|
|
f"https://api.github.com/repos/{self.repo}/contents/images",
|
|
headers=self.headers
|
|
)
|
|
|
|
if response.status_code != 200:
|
|
print(f"❌ 获取文件列表失败：{response.json().get('message')}")
|
|
return []
|
|
|
|
return response.json()
|
|
|
|
except Exception as e:
|
|
print(f"❌ 获取文件列表出错：{str(e)}")
|
|
return []
|
|
|
|
def get_file_info(self, file_sha):
|
|
"""获取文件的详细信息"""
|
|
try:
|
|
response = requests.get(
|
|
f"https://api.github.com/repos/{self.repo}/git/blobs/{file_sha}",
|
|
headers=self.headers
|
|
)
|
|
|
|
if response.status_code == 200:
|
|
return response.json()
|
|
return None
|
|
|
|
except Exception:
|
|
return None
|
|
|
|
def delete_file(self, file_path, file_sha):
|
|
"""删除文件"""
|
|
try:
|
|
response = requests.delete(
|
|
f"https://api.github.com/repos/{self.repo}/contents/{file_path}",
|
|
headers=self.headers,
|
|
json={
|
|
"message": "Auto cleanup old images",
|
|
"sha": file_sha
|
|
}
|
|
)
|
|
|
|
return response.status_code == 200
|
|
|
|
except Exception as e:
|
|
print(f"❌ 删除文件出错：{str(e)}")
|
|
return False
|
|
|
|
def cleanup(self):
|
|
"""执行清理"""
|
|
print(f"\n=== 开始清理 {self.repo} 仓库 ===")
|
|
print(f"🔍 保留最近 {self.days_to_keep} 天的图片")
|
|
|
|
# 获取所有图片
|
|
images = self.get_images()
|
|
if not images:
|
|
print("ℹ️ 没有找到需要清理的图片")
|
|
return
|
|
|
|
cutoff_date = datetime.now() - timedelta(days=self.days_to_keep)
|
|
deleted_count = 0
|
|
kept_count = 0
|
|
|
|
print(f"📅 清理 {cutoff_date.strftime('%Y-%m-%d')} 之前的图片")
|
|
|
|
for image in images:
|
|
if not image['name'].endswith(('.jpg', '.jpeg', '.png')):
|
|
continue
|
|
|
|
# 从文件名中提取时间戳
|
|
try:
|
|
# 假设文件名格式为 wanx_1234567890.jpg
|
|
timestamp = int(image['name'].split('_')[1].split('.')[0])
|
|
file_date = datetime.fromtimestamp(timestamp)
|
|
|
|
if file_date < cutoff_date:
|
|
print(f"🗑️ 删除: {image['name']} ({file_date.strftime('%Y-%m-%d %H:%M:%S')})")
|
|
if self.delete_file(image['path'], image['sha']):
|
|
deleted_count += 1
|
|
# 添加延迟避免触发 GitHub API 限制
|
|
time.sleep(1)
|
|
else:
|
|
kept_count += 1
|
|
|
|
except (ValueError, IndexError):
|
|
print(f"⚠️ 跳过无效文件名: {image['name']}")
|
|
continue
|
|
|
|
print(f"""
|
|
=== 清理完成 ===
|
|
✅ 已删除: {deleted_count} 个文件
|
|
💾 已保留: {kept_count} 个文件
|
|
""")
|
|
|
|
def main():
|
|
"""主函数"""
|
|
try:
|
|
# 加载配置
|
|
config_path = os.path.join(os.path.dirname(__file__), 'config.json')
|
|
with open(config_path, 'r', encoding='utf-8') as f:
|
|
config = json.load(f)
|
|
|
|
repo = config.get('GITHUB_REPO')
|
|
token = config.get('GITHUB_TOKEN')
|
|
|
|
if not repo or not token:
|
|
print("❌ 配置文件中缺少必要的GitHub配置")
|
|
return
|
|
|
|
# 创建清理器并执行清理
|
|
cleaner = GitHubCleaner(
|
|
repo=repo,
|
|
token=token,
|
|
days_to_keep=7 # 默认保留7天
|
|
)
|
|
cleaner.cleanup()
|
|
|
|
except Exception as e:
|
|
print(f"❌ 清理失败：{str(e)}")
|
|
|
|
if __name__ == "__main__":
|
|
main() |