使用python创建同步工作流

crdz • 2024年6月29日 pm5:18 • VPS

GitHub的仓库内容千千万万，好的项目就想fork，但是fork多了以后仓库就显得特别凌乱，现在就用python脚本自动创建工作流同步需要fork的仓库，只需输入仓库的地址就能创建一个工作流，然后把工作流上传到 .github/workflows目录里就可以了。

自动创建工作流脚本：

import os
import re
import random

def create_workflow_file(repo_url, repo_name, source_branch):
    """
    创建 GitHub Actions 工作流文件用于同步分支。
    
    参数:
        repo_url (str): 源仓库的 URL。
        repo_name (str): 要同步的仓库名称。
        source_branch (str): 源仓库的分支名称。
    """
    # 生成一个随机的时间用于 cron 调度
    random_minute = random.randint(0, 59)
    random_hour = random.randint(0, 23)

    # 定义工作流文件的内容
    workflow_content = f"""name: {repo_name}
on:
  schedule:
    - cron: '{random_minute} {random_hour} * * *'
  workflow_dispatch:
  repository_dispatch:
    types: sync-{repo_name}
jobs:
  repo-sync:
    env:
      PAT: ${{{{ secrets.PAT }}}}
    runs-on: ubuntu-latest
    if: github.event.repository.owner.id == github.event.sender.id
    steps:
      - name: Checkout the repository
        uses: actions/checkout@v2
        with:
          persist-credentials: false

      - name: Check if branch exists
        id: check_branch
        run: |
          if git show-ref --verify --quiet refs/heads/{repo_name}; then
            echo "branch_exists=true" >> $GITHUB_ENV
          else:
            echo "branch_exists=false" >> $GITHUB_ENV
          fi

      - name: Delete existing branch
        if: ${{{{ env.branch_exists == 'true' }}}}
        run: |
          git push origin --delete {repo_name}

      - name: Create new branch
        if: ${{{{ env.branch_exists == 'false' }}}}
        run: |
          git checkout -b {repo_name}
          git push --set-upstream origin {repo_name}

      - name: Sync branch
        uses: repo-sync/github-sync@v2
        if: env.PAT
        with:
          source_repo: "{repo_url}"
          source_branch: "{source_branch}"
          destination_branch: "{repo_name}"
          github_token: ${{{{ secrets.PAT }}}}
"""

    # 创建目录（如果不存在）并写入工作流文件
    workflow_file_path = f".github/workflows/{repo_name}.yml"
    os.makedirs(os.path.dirname(workflow_file_path), exist_ok=True)
    with open(workflow_file_path, 'w') as file:
        file.write(workflow_content)
    
    print(f"工作流文件创建在 {workflow_file_path}")

def get_source_branch(branch_choice):
    """
    根据用户输入获取源分支名称。
    
    参数:
        branch_choice (str): 用户输入的分支选择。
        
    返回:
        str: 源分支名称。
    """
    if branch_choice == '1':
        return 'main'
    elif branch_choice == '2':
        return 'master'
    return branch_choice

def main():
    repo_url = input("请输入要同步的仓库URL: ").strip()
    branch_choice = input("请输入上游仓库的分支名称 (1 表示 'main', 2 表示 'master', 或输入分支名称): ").strip()

    source_branch = get_source_branch(branch_choice)

    # 验证仓库 URL 是否有效
    match = re.search(r'github\.com/([^/]+/[^/]+)', repo_url)
    if not match:
        print("无效的仓库 URL，请提供有效的 GitHub 仓库 URL。")
        return

    repo_name = match.group(1).split('/')[-1]

    create_workflow_file(repo_url, repo_name, source_branch)

if __name__ == "__main__":
    main()

把脚本文件保存为create_workflow.py，然后再cmd或者powershell下python create_workflow.py就可以了。