Files
skills/scripts/git_auto_commit_with_skills_organize.py
2026-03-20 17:56:59 +08:00

252 lines
8.2 KiB
Python

#!/usr/bin/env python3
# git_auto_commit_with_skills_organize.py - 先整理 skills 文件夹,再自动提交到 Gitea
# 用法: python git_auto_commit_with_skills_organize.py [目录路径]
import os
import sys
import shutil
import subprocess
import tempfile
from datetime import datetime
from pathlib import Path
from urllib.parse import urlparse
# 配置
REPO_URL = "https://gittea.dev/popiskill/skills.git"
BRANCH = "master"
DEFAULT_SOURCE_DIR = "C:/ai/openclaw" # Windows 路径格式
SKILLS_BASE_DIR = "C:/ai/skills" # skills 文件夹根目录
# 分类映射
CATEGORY_MAPPING = {
"内容创作": "content_creation",
"图像制作": "image_generation",
"视频制作": "video_production",
"音频创作": "audio_creation",
"AI剪辑": "AI_video_trim",
"社媒运营": "social_media",
"电商工具": "E-commerce_tools",
"漫剧制作": "comic_drama"
}
# 排除的文件/目录
EXCLUDE_PATTERNS = ['.git', '.openclaw']
def load_config():
"""从配置文件读取凭证"""
script_dir = Path(__file__).parent
config_file = script_dir / "git-config.env"
if not config_file.exists():
print(f"错误: 配置文件不存在: {config_file}")
print("请创建配置文件 git-config.env 并设置 GIT_USERNAME 和 GIT_PASSWORD")
sys.exit(1)
config = {}
with open(config_file, 'r', encoding='utf-8') as f:
for line in f:
line = line.strip()
if line and not line.startswith('#') and '=' in line:
key, value = line.split('=', 1)
config[key.strip()] = value.strip().strip('"\'')
if not config.get('GIT_USERNAME') or not config.get('GIT_PASSWORD'):
print("错误: 请在配置文件中设置 GIT_USERNAME 和 GIT_PASSWORD")
sys.exit(1)
return config['GIT_USERNAME'], config['GIT_PASSWORD']
def build_auth_url(repo_url, username, token):
"""构建带凭证的 URL"""
parsed = urlparse(repo_url)
return f"{parsed.scheme}://{username}:{token}@{parsed.netloc}{parsed.path}"
def copy_directory(src, dst, exclude=None):
"""复制目录,排除指定文件"""
if exclude is None:
exclude = []
src_path = Path(src)
dst_path = Path(dst)
for item in src_path.rglob('*'):
# 检查是否在排除列表中
rel_path = item.relative_to(src_path)
if any(part in exclude for part in rel_path.parts):
continue
if item.is_file():
target = dst_path / rel_path
target.parent.mkdir(parents=True, exist_ok=True)
shutil.copy2(item, target)
def run_git_command(args, cwd=None, check=True):
"""运行 git 命令"""
result = subprocess.run(
['git'] + args,
cwd=cwd,
capture_output=True,
text=True,
encoding='utf-8'
)
if check and result.returncode != 0:
print(f"Git 错误: {result.stderr}")
raise subprocess.CalledProcessError(result.returncode, ['git'] + args)
return result
def organize_skills():
"""
整理 C:/ai/skills 中的各个分类文件夹
为每个.zip文件创建同名文件夹,将.zip放入对应创建文件夹,并配一个install.md
"""
print("=" * 60)
print("步骤 1: 整理 Skills 文件夹")
print("=" * 60)
base_path = Path(SKILLS_BASE_DIR)
if not base_path.exists():
print(f"警告: Skills 目录不存在: {SKILLS_BASE_DIR}")
return 0
total_processed = 0
for cat_name, cat_mapping in CATEGORY_MAPPING.items():
cat_path = base_path / cat_name
if not cat_path.exists():
print(f"分类不存在,跳过: {cat_name}")
continue
print(f"\n处理分类: {cat_name} -> {cat_mapping}")
# 只查找当前目录下的 .zip 文件(不递归)
zip_files = list(cat_path.glob("*.zip"))
for zip_file in zip_files:
zip_name = zip_file.stem # 不含 .zip 扩展名
folder_path = cat_path / zip_name
# 创建同名文件夹
if not folder_path.exists():
folder_path.mkdir(parents=True, exist_ok=True)
print(f" [创建] {zip_name}/")
# 移动 ZIP 文件到文件夹
dest_zip_path = folder_path / zip_file.name
if not dest_zip_path.exists():
shutil.move(str(zip_file), str(dest_zip_path))
print(f" [移动] {zip_file.name}")
# 创建 install.md
install_md_path = folder_path / "install.md"
zip_file_name = f"{zip_name}.zip"
install_content = f"skill名称={zip_name} 请执行以下流程:下载 https://gittea.dev/popiskill/skills/raw/branch/master/{cat_mapping}/{zip_name}/{zip_file_name} 文件,将该zip 解压至openclaw根目录(skill安装) ./skills/{zip_name} 目录,然后删除下载的临时文件和文档。"
with open(install_md_path, 'w', encoding='utf-8') as f:
f.write(install_content)
print(f" [创建] install.md")
total_processed += 1
print(f"\n[完成] Skills 整理完成,共处理 {total_processed} 个 ZIP 文件")
return total_processed
def git_auto_commit(source_dir):
"""执行 Git 自动提交"""
print("\n" + "=" * 60)
print("步骤 2: Git 自动提交")
print("=" * 60)
source_path = Path(source_dir)
if not source_path.exists():
print(f"错误: 目录不存在: {source_dir}")
sys.exit(1)
# 加载配置
username, token = load_config()
# 创建临时工作目录
work_dir = tempfile.mkdtemp(prefix='git_auto_commit_')
print(f"工作目录: {work_dir}")
print(f"源目录: {source_dir}")
try:
# 构建带凭证的 URL
auth_url = build_auth_url(REPO_URL, username, token)
# 尝试克隆现有仓库
try:
run_git_command(['clone', auth_url, '.'], cwd=work_dir)
print("已克隆现有仓库")
except subprocess.CalledProcessError:
print("初始化新仓库...")
run_git_command(['init'], cwd=work_dir)
run_git_command(['remote', 'add', 'origin', auth_url], cwd=work_dir)
# 复制文件到仓库
print("复制文件...")
copy_directory(source_dir, work_dir, exclude=EXCLUDE_PATTERNS)
# 配置 git
run_git_command(['config', 'user.email', 'auto@commit.local'], cwd=work_dir)
run_git_command(['config', 'user.name', 'Auto Commit'], cwd=work_dir)
# 添加所有更改
run_git_command(['add', '-A'], cwd=work_dir)
# 检查是否有更改要提交
status_result = run_git_command(['diff', '--cached', '--quiet'], cwd=work_dir, check=False)
if status_result.returncode == 0:
print("没有更改需要提交")
return False
# 提交并推送
commit_msg = f"Auto commit: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
run_git_command(['commit', '-m', commit_msg], cwd=work_dir)
run_git_command(['push', 'origin', BRANCH], cwd=work_dir)
print(f"[成功] 已提交到 {REPO_URL}")
return True
finally:
# 清理临时目录
shutil.rmtree(work_dir, ignore_errors=True)
def main():
# 获取源目录
if len(sys.argv) < 2:
source_dir = DEFAULT_SOURCE_DIR
print(f"未指定目录,使用默认: {source_dir}")
else:
source_dir = sys.argv[1]
print("=" * 60)
print("Git Auto Commit with Skills Organize")
print(f"开始时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
print("=" * 60)
# 步骤 1: 整理 skills 文件夹
skills_count = organize_skills()
# 步骤 2: Git 自动提交
committed = git_auto_commit(source_dir)
print("\n" + "=" * 60)
print("执行完成")
print(f"Skills 整理: {skills_count}")
print(f"Git 提交: {'成功' if committed else '无更改或失败'}")
print(f"结束时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
print("=" * 60)
if __name__ == '__main__':
main()